预生成gpt用户token，使用时激活token

netflix-common/src/main/java/com/cyksj/common/constant/Constant.java

@@ -17,6 +17,8 @@ public interface Constant {
 
 	String DEFAULT_NAME = "银河用户";
 
+	String YH_GPT_USER_NAME = "银河录像局";
+
 	List<String> noOrderStatus = List.of("close", "noPayment");
 
 	List<String> noOrderAllStatus = List.of("noPayment", "close", "refund");

netflix-dao/src/main/java/com/cyksj/mapper/ChatgptUserTokenPrepareMapper.java

@@ -0,0 +1,17 @@
+package com.cyksj.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.cyksj.model.entity.ChatgptUserTokenPrepare;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
+
+/**
+ * 项目名: yhlxj2
+ * 文件名: ChatgptUserTokenPrepareMapper
+ * 创建者: JavaZou
+ * 创建时间:2024/4/18 17:58
+ */
+public interface ChatgptUserTokenPrepareMapper extends BaseMapper<ChatgptUserTokenPrepare> {
+	void batchInsert(@Param("list") List<ChatgptUserTokenPrepare> list);
+}

netflix-dao/src/main/java/com/cyksj/model/entity/ChatgptUserTokenPrepare.java

@@ -0,0 +1,36 @@
+package com.cyksj.model.entity;
+
+import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
+import com.alibaba.excel.annotation.ExcelProperty;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ * 项目名: yhlxj2
+ * 文件名: chatgpt_user_token_papre
+ * 创建者: JavaZou
+ * 创建时间:2024/4/18 17:56
+ */
+@TableName("chatgpt_user_token_prepare")
+@Getter
+@Setter
+@ExcelIgnoreUnannotated
+public class ChatgptUserTokenPrepare extends BaseEntity{
+	private Long skuId;
+
+	private String name;
+
+	@TableField(exist = false)
+	@ExcelProperty("规格")
+	private String skuName;
+
+	private String img;
+
+	@TableField("userToken")
+	@ExcelProperty("用户token")
+	private String userToken;
+
+	private Boolean status;
+}

netflix-dao/src/main/java/com/cyksj/model/request/ChatGptUserTokenGenerateReq.java

@@ -0,0 +1,34 @@
+package com.cyksj.model.request;
+
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ * 项目名: yhlxj2
+ * 文件名: ChatGptUserTokenGenerateReq
+ * 创建者: JavaZou
+ * 创建时间:2024/4/18 17:49
+ */
+@Getter
+@Setter
+public class ChatGptUserTokenGenerateReq {
+	/**
+	 * 生成数量
+	 */
+	private Integer num;
+
+	/**
+	 * 名称
+	 */
+	private String name;
+
+	/**
+	 * logo
+	 */
+	private String img;
+
+	/**
+	 * 生成镜像规格
+	 */
+	private Long skuId;
+}

netflix-dao/src/main/resources/mapper/ChatgptUserTokenPrepareMapper.xml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+
+<mapper namespace="com.cyksj.mapper.ChatgptUserTokenPrepareMapper">
+
+    <insert id="batchInsert">
+        insert into
+        chatgpt_user_token_prepare(`name`,`img`,`userToken`,`sku_id`,`created_time`,`update_time`)
+        values
+        <foreach collection="list" item="item" separator=",">
+            (#{item.name},#{item.img},#{item.userToken},#{item.skuId},now(),now())
+        </foreach>
+    </insert>
+</mapper>

netflix-service/src/main/java/com/cyksj/service/chatgpt/impl/ChatGptAccountServiceImpl.java

@@ -1,6 +1,7 @@
 package com.cyksj.service.chatgpt.impl;
 
 import cn.hutool.core.date.DateTime;
+import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.lang.UUID;
 import cn.hutool.http.HttpRequest;
 import cn.hutool.http.HttpResponse;
@@ -89,6 +90,8 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
 
     private final BeanSearcher beanSearcher;
 
+    private final ChatgptUserTokenPrepareMapper chatgptUserTokenPrepareMapper;
+
     @Override
     public String addAccount(Account account) {
         if (isAccountExists(account)) {
@@ -349,6 +352,36 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
         return chatgptUser;
     }
 
+    /**
+     * 创建chatgptUser 无需车队
+     */
+    public ChatgptUser generateChatGptUserUnderPrepare(ChatgptUserTokenPrepare prepare) {
+        ChatgptUser chatgptUser = new ChatgptUser();
+        GoodsDonSku goodsDonSku = skuMapper.selectById(prepare.getSkuId());
+        if (goodsDonSku == null) {
+            throw BusinessRuntimeException.getInstance("车队规格不存在,请联系客服..");
+        }
+        if (goodsDonSku.getIsCar()) {
+            chatgptUser.setIsCar(true);
+            chatgptUser.setLimitNum(goodsDonSku.getGptLimitNum());
+            chatgptUser.setLimitTime(goodsDonSku.getGptLimitTime());
+        }
+        try {
+            chatgptUser.setExpireTime(DateUtil.offsetMonth(DateTime.now(), goodsDonSku.getMonths()));
+            chatgptUser.setIsPlus(1);
+            chatgptUser.setName(prepare.getName());
+            chatgptUser.setImg(prepare.getImg());
+            chatgptUser.setUserToken(prepare.getUserToken());
+            chatgptUserMapper.insert(chatgptUser);
+
+            prepare.setStatus(true);
+            chatgptUserTokenPrepareMapper.updateById(prepare);
+            log.info("用户激活userToken:{}成功", prepare.getUserToken());
+        } catch (DuplicateKeyException e) {
+        }
+        return chatgptUser;
+    }
+
     private void updateChatgptUser(GroupsRelation groupsRelation, User user, ChatgptUser chatgptUser) {
         chatgptUser.setName(user.getNickname());
         chatgptUser.setImg(getWxImg(user.getHeadimgurl(), user));
@@ -567,6 +600,14 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
     @Override
     public ChatgptUser getCarChatGptUserWithToken(String userToken) {
         ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class).eq(ChatgptUser::getUserToken, userToken).last(" limit 1"));
+        if (chatgptUser == null) {
+            //是否是用户token预备账号
+            ChatgptUserTokenPrepare chatgptUserTokenPrepare = chatgptUserTokenPrepareMapper.selectOne(Wrappers.lambdaQuery(ChatgptUserTokenPrepare.class)
+                    .eq(ChatgptUserTokenPrepare::getUserToken, userToken).last("limit 1"));
+            if (chatgptUserTokenPrepare != null) {
+                chatgptUser = generateChatGptUserUnderPrepare(chatgptUserTokenPrepare);
+            }
+        }
         if (chatgptUser != null) {
             if(!chatgptUser.getIsCar()){
                 throw BusinessRuntimeException.getInstance("您非车队用户，请购买车队套餐");

netflix-web/src/main/java/com/cyksj/web/controller/manage/mirror/CmsMirrorController.java

@@ -0,0 +1,78 @@
+package com.cyksj.web.controller.manage.mirror;
+
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.lang.UUID;
+import com.alibaba.excel.support.ExcelTypeEnum;
+import com.cyksj.common.constant.Constant;
+import com.cyksj.mapper.ChatgptUserTokenPrepareMapper;
+import com.cyksj.mapper.GoodsDonSkuMapper;
+import com.cyksj.model.entity.ChatgptUserTokenPrepare;
+import com.cyksj.model.entity.GoodsDonSku;
+import com.cyksj.model.request.ChatGptUserTokenGenerateReq;
+import com.cyksj.web.util.EasyExcelUtils;
+import com.google.common.collect.Lists;
+import lombok.RequiredArgsConstructor;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.servlet.http.HttpServletResponse;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Optional;
+
+/**
+ * 项目名: yhlxj2
+ * 文件名: CmsMirrorController
+ * 创建者: JavaZou
+ * 创建时间:2024/4/18 17:45
+ */
+@RestController
+@RequestMapping("/manage/mirror/gpt")
+@RequiredArgsConstructor
+public class CmsMirrorController {
+
+	private final GoodsDonSkuMapper goodsDonSkuMapper;
+
+	private final ChatgptUserTokenPrepareMapper chatgptUserTokenPrepareMapper;
+
+	/**
+	 * 生成并导出gpt会话
+	 */
+	@PostMapping("/generateAndExport/gpt/session")
+	@Transactional(rollbackFor = Exception.class)
+	public void generateAndExportGptSession(@RequestBody ChatGptUserTokenGenerateReq gptUserTokenGenerateReq, HttpServletResponse response) {
+		Integer num = gptUserTokenGenerateReq.getNum();
+		if (num == null || num <= 0) {
+			throw new IllegalArgumentException("生成数量必须大于0");
+		}
+		Long skuId = gptUserTokenGenerateReq.getSkuId();
+		GoodsDonSku sku = goodsDonSkuMapper.selectById(skuId);
+		if (sku == null) {
+			throw new IllegalArgumentException("镜像规格不存在");
+		}
+		if (!sku.getIsCar()) {
+			throw new IllegalArgumentException("不是gpt车队规格");
+		}
+		String name = Optional.ofNullable(gptUserTokenGenerateReq.getName()).orElse(Constant.YH_GPT_USER_NAME);
+		String img = Optional.ofNullable(gptUserTokenGenerateReq.getName()).orElse(Constant.DEFAULT_HEAD_IMG);
+		List<ChatgptUserTokenPrepare> prepares = new ArrayList<>(num);
+		ChatgptUserTokenPrepare chatgptUserTokenPrepare = null;
+
+		for (int i = 0; i < num; i++) {
+			chatgptUserTokenPrepare = new ChatgptUserTokenPrepare();
+			chatgptUserTokenPrepare.setUserToken(UUID.randomUUID().toString());
+			chatgptUserTokenPrepare.setName(name);
+			chatgptUserTokenPrepare.setImg(img);
+			chatgptUserTokenPrepare.setSkuId(skuId);
+			chatgptUserTokenPrepare.setSkuName(sku.getSpecVal());
+			prepares.add(chatgptUserTokenPrepare);
+		}
+		if (CollUtil.isNotEmpty(prepares)) {
+			Lists.partition(prepares, 500).forEach(e -> chatgptUserTokenPrepareMapper.batchInsert(e));
+		}
+		EasyExcelUtils.createExcelStreamMutilByEasyExcel(response, ChatgptUserTokenPrepare.class, prepares, "用户token","用户token", ExcelTypeEnum.XLSX, null);
+	}
+}