基板

src/main/java/com/quinn/service/InviteService.java

@@ -13,4 +13,5 @@ import com.baomidou.mybatisplus.extension.service.IService;
  */
 public interface InviteService extends IService<Invite> {
 
+    String getCode();
 }

src/main/java/com/quinn/service/impl/InviteServiceImpl.java

@@ -1,9 +1,11 @@
 package com.quinn.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.quinn.pojo.Invite;
 import com.quinn.mapper.InviteMapper;
 import com.quinn.service.InviteService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.quinn.utils.QuinnUtils;
 import org.springframework.stereotype.Service;
 
 /**
@@ -17,4 +19,17 @@ import org.springframework.stereotype.Service;
 @Service
 public class InviteServiceImpl extends ServiceImpl<InviteMapper, Invite> implements InviteService {
 
+
+    @Override
+    public String getCode() {
+        String code = "";
+        while (true){
+            code = QuinnUtils.getCode();
+            int count = count(new QueryWrapper<Invite>().eq("apply_email", code).eq("status",'0'));
+            if (count <= 0){
+                break;
+            }
+        }
+        return code;
+    }
 }