content: /*
   * Copyright 1999-2018 Alibaba Group Holding Ltd.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package com.alibaba.csp.sentinel.dashboard.datasource.entity.gateway;
  
  import com.alibaba.csp.sentinel.adapter.gateway.common.rule.GatewayFlowRule;
  import com.alibaba.csp.sentinel.adapter.gateway.common.rule.GatewayParamFlowItem;
  import com.alibaba.csp.sentinel.dashboard.datasource.entity.rule.RuleEntity;
  import com.alibaba.csp.sentinel.slots.block.Rule;
  
  import java.util.Date;
  import java.util.Objects;
  
  /**
   * Entity for {@link GatewayFlowRule}.
   *
   * @author cdfive
   * @since 1.7.0
   */
  public class GatewayFlowRuleEntity implements RuleEntity {
  
      /**间隔单位*/
      /**0-秒*/
      public static final int INTERVAL_UNIT_SECOND = 0;
      /**1-分*/
      public static final int INTERVAL_UNIT_MINUTE = 1;
      /**2-时*/
      public static final int INTERVAL_UNIT_HOUR = 2;
      /**3-天*/
      public static final int INTERVAL_UNIT_DAY = 3;
  
      private Long id;
      private String app;
      private String ip;
      private Integer port;
  
      private Date gmtCreate;
      private Date gmtModified;
  
      private String resource;
      private Integer resourceMode;
  
      private Integer grade;
      private Double count;
      private Long interval;
      private Integer intervalUnit;
  
      private Integer controlBehavior;
      private Integer burst;
  
      private Integer maxQueueingTimeoutMs;
  
      private GatewayParamFlowItemEntity paramItem;
  
      public static Long calIntervalSec(Long interval, Integer intervalUnit) {
          switch (intervalUnit) {
              case INTERVAL_UNIT_SECOND:
                  return interval;
              case INTERVAL_UNIT_MINUTE:
                  return interval * 60;
              case INTERVAL_UNIT_HOUR:
                  return interval * 60 * 60;
              case INTERVAL_UNIT_DAY:
                  return interval * 60 * 60 * 24;
              default:
                  break;
          }
  
          throw new IllegalArgumentException("Invalid intervalUnit: " + intervalUnit);
      }
  
      public static Object[] parseIntervalSec(Long intervalSec) {
          if (intervalSec % (60 * 60 * 24) == 0) {
              return new Object[] {intervalSec / (60 * 60 * 24), INTERVAL_UNIT_DAY};
          }
  
          if (intervalSec % (60 * 60 ) == 0) {
              return new Object[] {intervalSec / (60 * 60), INTERVAL_UNIT_HOUR};
          }
  
          if (intervalSec % 60 == 0) {
              return new Object[] {intervalSec / 60, INTERVAL_UNIT_MINUTE};
          }
  
          return new Object[] {intervalSec, INTERVAL_UNIT_SECOND};
      }
  
      public GatewayFlowRule toGatewayFlowRule() {
          GatewayFlowRule rule = new GatewayFlowRule();
          rule.setResource(resource);
          rule.setResourceMode(resourceMode);
  
          rule.setGrade(grade);
          rule.setCount(count);
          rule.setIntervalSec(calIntervalSec(interval, intervalUnit));
  
          rule.setControlBehavior(controlBehavior);
  
          if (burst != null) {
              rule.setBurst(burst);
          }
  
          if (maxQueueingTimeoutMs != null) {
              rule.setMaxQueueingTimeoutMs(maxQueueingTimeoutMs);
          }
  
          if (paramItem != null) {
              GatewayParamFlowItem ruleItem = new GatewayParamFlowItem();
              rule.setParamItem(ruleItem);
              ruleItem.setParseStrategy(paramItem.getParseStrategy());
              ruleItem.setFieldName(paramItem.getFieldName());
              ruleItem.setPattern(paramItem.getPattern());
  
              if (paramItem.getMatchStrategy() != null) {
                  ruleItem.setMatchStrategy(paramItem.getMatchStrategy());
              }
          }
  
          return rule;
      }
  
      public static GatewayFlowRuleEntity fromGatewayFlowRule(String app, String ip, Integer port, GatewayFlowRule rule) {
          GatewayFlowRuleEntity entity = new GatewayFlowRuleEntity();
          entity.setApp(app);
          entity.setIp(ip);
          entity.setPort(port);
  
          entity.setResource(rule.getResource());
          entity.setResourceMode(rule.getResourceMode());
  
          entity.setGrade(rule.getGrade());
          entity.setCount(rule.getCount());
          Object[] intervalSecResult = parseIntervalSec(rule.getIntervalSec());
          entity.setInterval((Long) intervalSecResult[0]);
          entity.setIntervalUnit((Integer) intervalSecResult[1]);
  
          entity.setControlBehavior(rule.getControlBehavior());
          entity.setBurst(rule.getBurst());
          entity.setMaxQueueingTimeoutMs(rule.getMaxQueueingTimeoutMs());
  
          GatewayParamFlowItem paramItem = rule.getParamItem();
          if (paramItem != null) {
              GatewayParamFlowItemEntity itemEntity = new GatewayParamFlowItemEntity();
              entity.setParamItem(itemEntity);
              itemEntity.setParseStrategy(paramItem.getParseStrategy());
              itemEntity.setFieldName(paramItem.getFieldName());
              itemEntity.setPattern(paramItem.getPattern());
              itemEntity.setMatchStrategy(paramItem.getMatchStrategy());
          }
  
          return entity;
      }
  
      @Override
      public Long getId() {
          return id;
      }
  
      @Override
      public void setId(Long id) {
          this.id = id;
      }
  
      @Override
      public String getApp() {
          return app;
      }
  
      public void setApp(String app) {
          this.app = app;
      }
  
      @Override
      public String getIp() {
          return ip;
      }
  
      public void setIp(String ip) {
          this.ip = ip;
      }
  
      @Override
      public Integer getPort() {
          return port;
      }
  
      public void setPort(Integer port) {
          this.port = port;
      }
  
      @Override
      public Date getGmtCreate() {
          return gmtCreate;
      }
  
      public void setGmtCreate(Date gmtCreate) {
          this.gmtCreate = gmtCreate;
      }
  
      @Override
      public Rule toRule() {
          return null;
      }
  
      public Date getGmtModified() {
          return gmtModified;
      }
  
      public void setGmtModified(Date gmtModified) {
          this.gmtModified = gmtModified;
      }
  
      public GatewayParamFlowItemEntity getParamItem() {
          return paramItem;
      }
  
      public void setParamItem(GatewayParamFlowItemEntity paramItem) {
          this.paramItem = paramItem;
      }
  
      public String getResource() {
          return resource;
      }
  
      public void setResource(String resource) {
          this.resource = resource;
      }
  
      public Integer getResourceMode() {
          return resourceMode;
      }
  
      public void setResourceMode(Integer resourceMode) {
          this.resourceMode = resourceMode;
      }
  
      public Integer getGrade() {
          return grade;
      }
  
      public void setGrade(Integer grade) {
          this.grade = grade;
      }
  
      public Double getCount() {
          return count;
      }
  
      public void setCount(Double count) {
          this.count = count;
      }
  
      public Long getInterval() {
          return interval;
      }
  
      public void setInterval(Long interval) {
          this.interval = interval;
      }
  
      public Integer getIntervalUnit() {
          return intervalUnit;
      }
  
      public void setIntervalUnit(Integer intervalUnit) {
          this.intervalUnit = intervalUnit;
      }
  
      public Integer getControlBehavior() {
          return controlBehavior;
      }
  
      public void setControlBehavior(Integer controlBehavior) {
          this.controlBehavior = controlBehavior;
      }
  
      public Integer getBurst() {
          return burst;
      }
  
      public void setBurst(Integer burst) {
          this.burst = burst;
      }
  
      public Integer getMaxQueueingTimeoutMs() {
          return maxQueueingTimeoutMs;
      }
  
      public void setMaxQueueingTimeoutMs(Integer maxQueueingTimeoutMs) {
          this.maxQueueingTimeoutMs = maxQueueingTimeoutMs;
      }
  
      @Override
      public boolean equals(Object o) {
          if (this == o) { return true; }
          if (o == null || getClass() != o.getClass()) { return false; }
          GatewayFlowRuleEntity that = (GatewayFlowRuleEntity) o;
          return Objects.equals(id, that.id) &&
                  Objects.equals(app, that.app) &&
                  Objects.equals(ip, that.ip) &&
                  Objects.equals(port, that.port) &&
                  Objects.equals(gmtCreate, that.gmtCreate) &&
                  Objects.equals(gmtModified, that.gmtModified) &&
                  Objects.equals(resource, that.resource) &&
                  Objects.equals(resourceMode, that.resourceMode) &&
                  Objects.equals(grade, that.grade) &&
                  Objects.equals(count, that.count) &&
                  Objects.equals(interval, that.interval) &&
                  Objects.equals(intervalUnit, that.intervalUnit) &&
                  Objects.equals(controlBehavior, that.controlBehavior) &&
                  Objects.equals(burst, that.burst) &&
                  Objects.equals(maxQueueingTimeoutMs, that.maxQueueingTimeoutMs) &&
                  Objects.equals(paramItem, that.paramItem);
      }
  
      @Override
      public int hashCode() {
          return Objects.hash(id, app, ip, port, gmtCreate, gmtModified, resource, resourceMode, grade, count, interval, intervalUnit, controlBehavior, burst, maxQueueingTimeoutMs, paramItem);
      }
  
      @Override
      public String toString() {
          return "GatewayFlowRuleEntity{" +
                  "id=" + id +
                  ", app='" + app + '\'' +
                  ", ip='" + ip + '\'' +
                  ", port=" + port +
                  ", gmtCreate=" + gmtCreate +
                  ", gmtModified=" + gmtModified +
                  ", resource='" + resource + '\'' +
                  ", resourceMode=" + resourceMode +
                  ", grade=" + grade +
                  ", count=" + count +
                  ", interval=" + interval +
                  ", intervalUnit=" + intervalUnit +
                  ", controlBehavior=" + controlBehavior +
                  ", burst=" + burst +
                  ", maxQueueingTimeoutMs=" + maxQueueingTimeoutMs +
                  ", paramItem=" + paramItem +
                  '}';
      }
  }
  