视频055~(雪花算法)

beacon-api/src/main/java/com/mashibing/api/client/BeaconCacheClient.java

@@ -25,4 +25,5 @@ public interface BeaconCacheClient {
 
     @GetMapping("/cache/smember/{key}")
     Set<Map> smember(@PathVariable(value = "key")String key);
+
 }

beacon-api/src/main/java/com/mashibing/api/filter/impl/FeeCheckFilter.java

@@ -1,8 +1,14 @@
 package com.mashibing.api.filter.impl;
 
+import com.mashibing.api.client.BeaconCacheClient;
 import com.mashibing.api.filter.CheckFilter;
+import com.mashibing.common.constant.ApiConstant;
+import com.mashibing.common.constant.CacheConstant;
+import com.mashibing.common.enums.ExceptionEnums;
+import com.mashibing.common.exception.ApiException;
 import com.mashibing.common.model.StandardSubmit;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 /**
@@ -13,9 +19,49 @@ import org.springframework.stereotype.Service;
 @Slf4j
 public class FeeCheckFilter implements CheckFilter {
 
+    @Autowired
+    private BeaconCacheClient cacheClient;
+
+    /**
+     * 只要短信内容的文字长度小于等于70个字，按照一条计算
+     */
+    private final int MAX_LENGTH = 70;
+
+    /**
+     * 如果短信内容的文字长度超过70，67字/条计算
+     */
+    private final int LOOP_LENGTH = 67;
+
+    private final String BALANCE = "balance";
+
 
     @Override
     public void check(StandardSubmit submit) {
         log.info("【接口模块-校验客户余额】   校验ing…………");
+        //1、从submit中获取到短信内容
+        int length = submit.getText().length();
+
+        //2、判断短信内容的长度，如果小于等于70，算作一条，如果大于70字，按照67字/条，算出来当前短信的费用
+        if(length <= MAX_LENGTH){
+            // 当前短信内容是一条
+            submit.setFee(ApiConstant.SINGLE_FEE);
+        }else{
+            int strip = length % LOOP_LENGTH == 0 ? length / LOOP_LENGTH : length / LOOP_LENGTH + 1;
+            submit.setFee(ApiConstant.SINGLE_FEE * strip);
+        }
+
+        //3、从Redis中查询出客户剩余的金额
+        Long balance = ((Integer) cacheClient.hget(CacheConstant.CLIENT_BALANCE + submit.getClientId(), BALANCE)).longValue();
+
+        //4、判断金额是否满足当前短信费用\
+        if(balance >= submit.getFee()){
+            log.info("【接口模块-校验客户余额】   用户金额充足！！");
+            return;
+        }
+
+        //5、不满足就抛出异常
+        log.info("【接口模块-校验客户余额】   客户余额不足");
+        throw new ApiException(ExceptionEnums.BALANCE_NOT_ENOUGH);
     }
+
 }

beacon-common/pom.xml

@@ -16,6 +16,11 @@
             <groupId>org.projectlombok</groupId>
             <artifactId>lombok</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.springframework</groupId>
+            <artifactId>spring-context</artifactId>
+            <version>5.3.12</version>
+        </dependency>
     </dependencies>
 
 </project>

beacon-common/src/main/java/com/mashibing/common/constant/ApiConstant.java

@@ -15,4 +15,9 @@ public interface ApiConstant {
      */
     String SIGN_SUFFIX = "】";
 
+    /**
+     * 单条短信，默认每条50（厘）
+     */
+    Long SINGLE_FEE =  50L;
+
 }

beacon-common/src/main/java/com/mashibing/common/enums/ExceptionEnums.java

@@ -14,8 +14,9 @@ public enum ExceptionEnums {
     ERROR_SIGN(-3,"无可用签名"),
     ERROR_TEMPLATE(-4,"无可用模板"),
     ERROR_MOBILE(-5,"手机号格式不正确"),
-    BALANCE_NOT_ENOUGH(-6,"手客户余额不足"),
+    BALANCE_NOT_ENOUGH(-6,"客户余额不足"),
     PARAMETER_ERROR(-10,"参数不合法！"),
+    SNOWFLAKE_OUT_OF_RANGE(-11,"雪花算法的机器id或服务id超出最大范围！")
     ;
     private Integer code;
 

beacon-common/src/main/java/com/mashibing/common/model/StandardSubmit.java

@@ -58,7 +58,7 @@ public class StandardSubmit {
     private LocalDateTime sendTime;
 
     /**
-     * 当前短信的费用，计算短信内容的文字，70个字一条，超过部分，67个字一条
+     * 当前短信的费用，计算短信内容的文字，70个字一条，超过部分，67个字一条,单位（厘）
      */
     private Long fee;
 

beacon-common/src/main/java/com/mashibing/common/util/SnowFlakeUtil.java

@@ -0,0 +1,113 @@
+package com.mashibing.common.util;
+
+import com.mashibing.common.enums.ExceptionEnums;
+import com.mashibing.common.exception.ApiException;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import java.time.Instant;
+import java.time.LocalDate;
+import java.time.ZoneId;
+import java.util.Date;
+
+/**
+ * 雪花算法生成全局唯一ID
+ * 64个bit为的long类型的值
+ * 第一位：占1个bit位，就是0.
+ * 第二位：占41个bit位，代表时间戳
+ * 第三位：占5个bit位，代表机器id
+ * 第四位：占5个bit位，服务id
+ * 第五位：占12个bit位，序列，自增的数值
+ * @author zjw
+ * @description
+ */
+@Component
+@Slf4j
+public class SnowFlakeUtil {
+
+    /**
+     * 41个bit位存储时间戳，从0开始计算，最多可以存储69.7年。
+     * 那么如果默认使用，从1970年到现在，最多可以用到2039年左右。
+     * 按照从2022-11-11号开始计算，存储41个bit为，这样最多可以使用到2092年不到~~
+     */
+    private long timeStart = 1668096000000L;
+
+    /**
+     * 机器id
+     */
+    @Value("${snowflake.machineId:0}")
+    private long machineId;
+
+    /**
+     * 服务id
+     */
+    @Value("${snowflake.serviceId:0}")
+    private long serviceId;
+
+    /**
+     * 序列
+     */
+    private long sequence;
+
+
+    /**
+     * 机器id占用的bit位数
+     */
+    private long machineIdBits = 5L;
+
+    /**
+     * 服务id占用的bit位数
+     */
+    private long serviceIdBits = 5L;
+
+    /**
+     * 序列占用的bit位数
+     */
+    private long sequenceBits = 12L;
+
+    /**
+     * 计算出机器id的最大值
+     */
+    private long maxMachineId = -1 ^ (-1 << machineIdBits);
+
+    /**
+     * 计算出服务id的最大值
+     */
+    private long maxServiceId = -1 ^ (-1 << serviceIdBits);
+
+    @PostConstruct
+    public void init(){
+        if(machineId > maxMachineId || serviceId > maxServiceId){
+            System.out.println("机器ID或服务ID超过最大范围值！！");
+            throw new ApiException(ExceptionEnums.SNOWFLAKE_OUT_OF_RANGE);
+        }
+    }
+
+    /**
+     * 服务id需要位移的位数
+     */
+    private long serviceIdShift = sequenceBits;
+
+    /**
+     * 机器id需要位移的位数
+     */
+    private long machineIdShift = sequenceBits + serviceIdBits;
+
+    /**
+     * 时间戳需要位移的位数
+     */
+    private long timestampShift = sequenceBits + serviceIdBits + maxMachineId;
+
+
+
+
+
+
+
+
+
+
+
+}

beacon-test/src/main/java/com/mashibing/test/mapper/ClientBalanceMapper.java

@@ -1,5 +1,6 @@
 package com.mashibing.test.mapper;
 
+import com.mashibing.test.entity.ClientBalance;
 import org.apache.ibatis.annotations.Param;
 import org.apache.ibatis.annotations.Select;
 
@@ -9,7 +10,7 @@ import org.apache.ibatis.annotations.Select;
  */
 public interface ClientBalanceMapper {
 
-    @Select("select balance from client_balance where client_id = #{clientId}")
+    @Select("select * from client_balance where client_id = #{clientId}")
-    Long findByClientId(@Param("clientId")Long clientId);
+    ClientBalance findByClientId(@Param("clientId")Long clientId);
 
 }

beacon-test/src/test/java/com/mashibing/test/mapper/ClientBalanceMapperTest.java

@@ -1,13 +1,16 @@
 package com.mashibing.test.mapper;
 
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.mashibing.test.client.CacheClient;
+import com.mashibing.test.entity.ClientBalance;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.test.context.junit4.SpringRunner;
 
-import static org.junit.Assert.*;
+import java.util.Map;
 
 @SpringBootTest
 @RunWith(SpringRunner.class)
@@ -20,10 +23,11 @@ public class ClientBalanceMapperTest {
     private CacheClient cacheClient;
 
     @Test
-    public void findByClientId() {
+    public void findByClientId() throws JsonProcessingException {
-        Long balance = mapper.findByClientId(1L);
+        ClientBalance clientBalance = mapper.findByClientId(1L);
-        System.out.println(balance);
+        ObjectMapper objectMapper = new ObjectMapper();
+        Map map = objectMapper.readValue(objectMapper.writeValueAsString(clientBalance), Map.class);
 
-        cacheClient.set("client_balance:1",balance);
+        cacheClient.hmset("client_balance:1",map);
     }
 }