限流改造

sop-gateway/src/main/java/com/gitee/sop/gateway/entity/ConfigLimit.java

@@ -0,0 +1,67 @@
+package com.gitee.sop.gateway.entity;
+
+import lombok.Data;
+
+import javax.persistence.Column;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.Table;
+import java.util.Date;
+
+
+/**
+ * 表名：config_limit
+ * 备注：限流配置
+ *
+ * @author tanghc
+ */
+@Table(name = "config_limit")
+@Data
+public class ConfigLimit {
+    @Id
+    @Column(name = "id")
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    /**  数据库字段：id */
+    private Long id;
+
+    /** 路由id, 数据库字段：route_id */
+    private String routeId;
+
+    /**  数据库字段：app_key */
+    private String appKey;
+
+    /** 限流ip，多个用英文逗号隔开, 数据库字段：limit_ip */
+    private String limitIp;
+
+
+    /** 服务id, 数据库字段：service_id */
+    private String serviceId;
+
+    /** 限流策略，1：漏桶策略，2：令牌桶策略, 数据库字段：limit_type */
+    private Byte limitType;
+
+    /** 每秒可处理请求数, 数据库字段：exec_count_per_second */
+    private Integer execCountPerSecond;
+
+    /** 返回的错误码, 数据库字段：limit_code */
+    private String limitCode;
+
+    /** 返回的错误信息, 数据库字段：limit_msg */
+    private String limitMsg;
+
+    /** 令牌桶容量, 数据库字段：token_bucket_count */
+    private Integer tokenBucketCount;
+
+    /** 限流开启状态，1:开启，0关闭, 数据库字段：limit_status */
+    private Byte limitStatus;
+
+    /** 顺序，值小的优先执行, 数据库字段：order_index */
+    private Integer orderIndex;
+
+    /**  数据库字段：gmt_create */
+    private Date gmtCreate;
+
+    /**  数据库字段：gmt_modified */
+    private Date gmtModified;
+}

sop-gateway/src/main/java/com/gitee/sop/gateway/manager/DbLimitConfigManager.java

@@ -0,0 +1,67 @@
+package com.gitee.sop.gateway.manager;
+
+import com.alibaba.fastjson.JSON;
+import com.gitee.fastmybatis.core.query.Query;
+import com.gitee.sop.gateway.mapper.ConfigLimitMapper;
+import com.gitee.sop.gatewaycommon.bean.ChannelMsg;
+import com.gitee.sop.gatewaycommon.bean.ConfigLimitDto;
+import com.gitee.sop.gatewaycommon.manager.DefaultLimitConfigManager;
+import com.gitee.sop.gatewaycommon.manager.ZookeeperContext;
+import com.gitee.sop.gatewaycommon.util.MyBeanUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.env.Environment;
+
+import javax.annotation.PostConstruct;
+
+/**
+ * 限流配置管理
+ * @author tanghc
+ */
+@Slf4j
+public class DbLimitConfigManager extends DefaultLimitConfigManager {
+
+    @Autowired
+    ConfigLimitMapper configLimitMapper;
+
+    @Autowired
+    Environment environment;
+
+    @Override
+    public void load() {
+        Query query = new Query();
+        configLimitMapper.list(query)
+                .stream()
+                .forEach(configLimit -> putVal(configLimit));
+
+    }
+
+    protected void putVal(Object object) {
+        ConfigLimitDto configLimitDto = new ConfigLimitDto();
+        MyBeanUtil.copyPropertiesIgnoreNull(object, configLimitDto);
+        this.update(configLimitDto);
+    }
+
+
+    @PostConstruct
+    protected void after() throws Exception {
+        ZookeeperContext.setEnvironment(environment);
+        String path = ZookeeperContext.getLimitConfigChannelPath();
+        ZookeeperContext.listenPath(path, nodeCache -> {
+            String nodeData = new String(nodeCache.getCurrentData().getData());
+            ChannelMsg channelMsg = JSON.parseObject(nodeData, ChannelMsg.class);
+            final ConfigLimitDto configLimitDto = JSON.parseObject(channelMsg.getData(), ConfigLimitDto.class);
+            switch (channelMsg.getOperation()) {
+                case "reload":
+                    log.info("重新加载限流配置信息，configLimitDto:{}", configLimitDto);
+                    load();
+                    break;
+                case "update":
+                    log.info("更新限流配置信息，configLimitDto:{}", configLimitDto);
+                    update(configLimitDto);
+                    break;
+            }
+        });
+    }
+
+}

sop-gateway/src/main/java/com/gitee/sop/gateway/manager/ManagerInitializer.java

@@ -12,5 +12,6 @@ public class ManagerInitializer {
         apiConfig.setIsvManager(new DbIsvManager());
         apiConfig.setIsvRoutePermissionManager(new DbIsvRoutePermissionManager());
         apiConfig.setRouteConfigManager(new DbRouteConfigManager());
+        apiConfig.setLimitConfigManager(new DbLimitConfigManager());
     }
 }

sop-gateway/src/main/java/com/gitee/sop/gateway/mapper/ConfigLimitMapper.java

@@ -0,0 +1,11 @@
+package com.gitee.sop.gateway.mapper;
+
+import com.gitee.fastmybatis.core.mapper.CrudMapper;
+import com.gitee.sop.gateway.entity.ConfigLimit;
+
+
+/**
+ * @author tanghc
+ */
+public interface ConfigLimitMapper extends CrudMapper<ConfigLimit, Long> {
+}