From e16769409f0c147463d6350ed70eadad7511e6d5 Mon Sep 17 00:00:00 2001
From: wangwei <1610949092@qq.com>
Date: Fri, 16 Aug 2024 11:50:02 +0800
Subject: [PATCH] 8/16 code

---
 .../basic/BasicCollectBusTypeController.java       | 13 +++++++++++++
 .../service/basic/BasicCollectBustypeService.java  | 14 ++++++++++++++
 2 files changed, 27 insertions(+)

diff --git a/src/main/java/com/glxp/api/controller/basic/BasicCollectBusTypeController.java b/src/main/java/com/glxp/api/controller/basic/BasicCollectBusTypeController.java
index c3f1fe88a..b08dcc116 100644
--- a/src/main/java/com/glxp/api/controller/basic/BasicCollectBusTypeController.java
+++ b/src/main/java/com/glxp/api/controller/basic/BasicCollectBusTypeController.java
@@ -1,5 +1,6 @@
 package com.glxp.api.controller.basic;
 
+import cn.hutool.core.text.split.SplitIter;
 import cn.hutool.core.util.IdUtil;
 import com.github.pagehelper.PageInfo;
 import com.glxp.api.common.res.BaseResponse;
@@ -11,6 +12,7 @@ import com.glxp.api.res.PageSimpleResponse;
 import com.glxp.api.res.basic.BasicCollectBusTypeResponse;
 import com.glxp.api.service.basic.BasicCollectBustypeService;
 import lombok.extern.slf4j.Slf4j;
+import org.junit.Test;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -19,6 +21,7 @@ import org.springframework.web.bind.annotation.RestController;
 import javax.annotation.Resource;
 import java.util.Date;
 import java.util.List;
+import java.util.Random;
 
 /**
  * @author : zhuhzu
@@ -74,4 +77,14 @@ public class BasicCollectBusTypeController {
         }
         return ResultVOUtils.success("成功!!!!!");
     }
+
+
+    @PostMapping("/udiwms/basic/collectBusType/genCode")
+    public BaseResponse genCode() {
+//        String code = "SP";
+        String  code = basicCollectBustypeService.genCode();
+        return ResultVOUtils.success(code);
+    }
+
+
 }
diff --git a/src/main/java/com/glxp/api/service/basic/BasicCollectBustypeService.java b/src/main/java/com/glxp/api/service/basic/BasicCollectBustypeService.java
index 70fbed378..6417041e8 100644
--- a/src/main/java/com/glxp/api/service/basic/BasicCollectBustypeService.java
+++ b/src/main/java/com/glxp/api/service/basic/BasicCollectBustypeService.java
@@ -1,10 +1,12 @@
 package com.glxp.api.service.basic;
 
+import cn.hutool.core.util.IdUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.github.pagehelper.PageHelper;
 import com.glxp.api.req.basic.BasicCollectBusTypeRequest;
 import com.glxp.api.res.basic.BasicCollectBusTypeResponse;
 import lombok.Data;
+import org.junit.Test;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.glxp.api.dao.basic.BasicCollectBustypeMapper;
@@ -13,6 +15,7 @@ import com.glxp.api.entity.basic.BasicCollectBustypeEntity;
 import javax.annotation.Resource;
 import java.util.Collections;
 import java.util.List;
+import java.util.Random;
 
 @Service
 public class BasicCollectBustypeService extends ServiceImpl<BasicCollectBustypeMapper, BasicCollectBustypeEntity> {
@@ -34,4 +37,15 @@ public class BasicCollectBustypeService extends ServiceImpl<BasicCollectBustypeM
         return basicCollectBustypeMapper.selectOne(new LambdaQueryWrapper<BasicCollectBustypeEntity>().eq(BasicCollectBustypeEntity::getCode, code).last("limit  1 "));
     }
 
+    public String genCode() {
+        long workerId = IdUtil.getSnowflakeNextId();
+        String s = String.valueOf(workerId);
+        Random random = new Random();
+        int startRange = Math.max(0, s.length() - 6);
+        int randomStart = random.nextInt(startRange + 1);
+        String randomNumber = "SP" + s.substring(randomStart, randomStart + 6);
+        return  randomNumber;
+    }
+
+
 }