Solon Java Framework Java 新的生态型应用开发框架，更小、更快、更简单

package demo.dso.db; import demo.Config; import demo.dso.CacheUtil; import demo.dso.ConfigType; import demo.model.TagCountsVo; import demo.model.water_cfg.ConfigDo; import demo.model.water_cfg.LoggerDo; import demo.model.water_cfg.WhitelistDo; import org.noear.snack.ONode; import org.noear.water.WaterClient; import org.noear.water.dso.WhitelistApi; import org.noear.water.utils.TextUtils; import org.noear.wood.DbContext; import org.noear.wood.DbTableQuery; import java.sql.SQLException; import java.util.Arrays; import java.util.List; import java.util.stream.Collectors; public class DbWaterCfgApi { private static DbContext db() { return Config.water; } public static List<LoggerDo> getLoggerByTag(String tag) throws Exception { return db().table("water_cfg_logger") .where("is_enabled=1") .build(tb -> { if (!TextUtils.isEmpty(tag)) { tb.and("tag = ?",tag); } }) .orderBy("logger asc") .selectList("*",LoggerDo.class); } // public static LoggerDo getLogger(String logger) { try { return db().table("water_cfg_logger") .where("logger = ?", logger) .limit(1) .selectItem("*", LoggerDo.class); }catch (Exception ex){ throw new RuntimeException(ex); } } //获取白名单表tag public static List<TagCountsVo> getWhitelistTags() throws Exception { return db().table("water_cfg_whitelist") .groupBy("tag") .orderByAsc("tag") .selectList("tag,count(*) counts", TagCountsVo.class); } //获取ip白名单列表 public static List<WhitelistDo> getWhitelistByTag(String tag_name, String key, int state) throws SQLException { return db().table("water_cfg_whitelist") .whereEq("is_enabled",state==1) .build(tb -> { if(tag_name != null){ tb.andEq("tag",tag_name); } if (TextUtils.isEmpty(key) == false) { tb.andLk("value", key + "%"); } }) .selectList("*", WhitelistDo.class); } //新增ip白名单 public static boolean setWhitelist(Integer row_id, String tag, String type, String value, String note) throws SQLException { if (row_id == null) { row_id = 0; } if(value == null){ return false; } DbTableQuery qr = db().table("water_cfg_whitelist") .set("tag", tag.trim()) .set("type", type.trim()) .set("value", value.trim()) .set("note", note); if (row_id > 0) { return qr.whereEq("row_id", row_id).update() > 0; } else { return qr.insert() > 0; } } //批量导入 public static void impWhitelist(String tag, WhitelistDo wm) throws SQLException { if(TextUtils.isEmpty(tag) == false){ wm.tag = tag; } if(TextUtils.isEmpty(wm.tag) || TextUtils.isEmpty(wm.value)){ return; } db().table("water_cfg_whitelist") .set("tag", wm.tag) .set("type", wm.type) .set("value", wm.value) .set("note", wm.note) .insertBy("tag,type,value"); } //删除 public static boolean delWhitelist(int row_id) throws SQLException { return db().table("water_cfg_whitelist") .where("row_id = ?", row_id) .delete() > 0; } //批量删除 public static void delWhitelistByIds(int act, String ids) throws SQLException { List<Object> list = Arrays.asList(ids.split(",")).stream().map(s -> Integer.parseInt(s)).collect(Collectors.toList()); if(act == 9) { db().table("water_cfg_whitelist") .whereIn("row_id", list) .delete(); }else { db().table("water_cfg_whitelist") .set("is_enabled", (act == 1 ? 1 : 0)) .whereIn("row_id", list) .update(); } } public static WhitelistDo getWhitelist(int row_id) throws SQLException { return db().table("water_cfg_whitelist") .where("row_id = ?", row_id) .selectItem("*", WhitelistDo.class); } public static List<WhitelistDo> getWhitelistByIds(String ids) throws SQLException { List<Object> list = Arrays.asList(ids.split(",")) .stream() .map(s->Integer.parseInt(s)) .collect(Collectors.toList()); return db().table("water_cfg_whitelist") .whereIn("row_id", list) .selectList("*", WhitelistDo.class); } //加载IP白名单到静态缓存里 public static void reloadWhitelist() throws Exception { CacheUtil.data.clear("water_cfg_whitelist");//waterapi那儿，过一分钟就会自动刷新 } public static boolean isWhitelist(String val) throws SQLException { return db().table("water_cfg_whitelist") .whereIn("tag", Arrays.asList(WhitelistApi.tag_client, WhitelistApi.tag_server)) .andEq("type", "ip") .andEq("value", val) .caching(CacheUtil.data).usingCache(60) .cacheTag("whitelist") .selectExists(); } public static boolean hasGateway(String name) { try { return db().table("water_cfg_properties") .whereEq("tag", "_gateway") .andEq("key", name) .selectExists(); }catch (Exception ex){ return false; } } public static void addGateway(String tag, String key, String url, String policy, int is_enabled) throws SQLException { ONode n = new ONode() .set("url", url) .set("service",key.trim()) .set("policy", policy); db().table("water_cfg_properties") .set("tag", tag.trim()) .set("key", key.trim()) .set("is_enabled", is_enabled) .set("type", ConfigType.water_gateway) .set("value", n.toJson()) .insert(); } public static void updGateway(String tag, String ori_key, String key, String url, String policy, int is_enabled) throws SQLException { ONode n = new ONode() .set("url", url) .set("service", key.trim()) .set("policy", policy); //由 tag 决定，是否为gateway db().table("water_cfg_properties") .set("key", key.trim()) .set("value", n.toJson()) .set("is_enabled", is_enabled) .set("type", ConfigType.water_gateway) .whereEq("tag", tag.trim()) .andEq("key", ori_key.trim()) .update(); WaterClient.Notice.updateCache("upstream:" + ori_key); if (ori_key.equals(key) == false) { WaterClient.Notice.updateCache("upstream:" + key); } } public static List<ConfigDo> getGateways() throws SQLException { return db().table("water_cfg_properties") .whereEq("type", ConfigType.water_gateway) .selectList("tag,key,row_id,type,is_enabled", ConfigDo.class); } public static void delConfigByIds(int act, String ids) throws SQLException { List<Object> list = Arrays.asList(ids.split(",")).stream().map(s -> Integer.parseInt(s)).collect(Collectors.toList()); if (list.size() == 0) { return; } if (act == 9) { db().table("water_cfg_prope