loginUserInfos = loginUserInfoService.getBaseMapper().getAll();
40 | if(CollectionUtils.isNotEmpty(loginUserInfos)){
41 | for (LoginUserInfo loginUserInfo : loginUserInfos) {
42 | CacheUtil.put(loginUserInfo.getSessionId(),loginUserInfo.getUserInfo());
43 | }
44 | }
45 | log.info("reCacheLoginUser end cacheSize={} cost={}",loginUserInfos.size(),System.currentTimeMillis()-startTime);
46 | }
47 |
48 | }
49 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/mapper/LoginUserInfoMapper.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.mapper;
2 |
3 | import com.blue.cat.bean.entity.LoginUserInfo;
4 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
5 | import org.apache.ibatis.annotations.Delete;
6 | import org.apache.ibatis.annotations.Mapper;
7 | import org.apache.ibatis.annotations.Select;
8 |
9 | import java.util.List;
10 |
11 | /**
12 | *
13 | * 用户登录的id Mapper 接口
14 | *
15 | *
16 | * @author lixin
17 | * @since 2023-05-11
18 | */
19 | @Mapper
20 | public interface LoginUserInfoMapper extends BaseMapper {
21 |
22 | @Delete("delete from login_user_info ")
23 | int deleteAll();
24 |
25 | @Select("select * from login_user_info ")
26 | List getAll();
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/mapper/PopupInfoMapper.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.mapper;
2 |
3 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
4 | import com.blue.cat.bean.entity.PopupInfo;
5 | import org.apache.ibatis.annotations.Param;
6 | import org.apache.ibatis.annotations.Select;
7 |
8 | /**
9 | *
10 | * @author linyous
11 | * @since 2023-05-10
12 | */
13 | public interface PopupInfoMapper extends BaseMapper {
14 |
15 | @Select("select id,title,content, createTime, popupLocation,isShow from popup_info where popupLocation=#{source} order by createTime desc limit 1")
16 | PopupInfo getPopup(@Param("source")String source);
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/mapper/PromptModelMapper.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.mapper;
2 |
3 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
4 | import com.blue.cat.bean.entity.PromptModel;
5 |
6 | /**
7 | *
8 | * @author linyous
9 | * @since 2023-05-10
10 | */
11 | public interface PromptModelMapper extends BaseMapper {
12 |
13 |
14 | }
15 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/mapper/UserAccessRuleMapper.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.mapper;
2 |
3 | import com.blue.cat.bean.entity.UserAccessRule;
4 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
5 | import org.apache.ibatis.annotations.Mapper;
6 | import org.apache.ibatis.annotations.Param;
7 | import org.apache.ibatis.annotations.Select;
8 |
9 | import java.util.List;
10 | import java.util.Set;
11 |
12 | /**
13 | *
14 | * 用户访问规则 Mapper 接口
15 | *
16 | *
17 | * @author lixin
18 | * @since 2023-05-10
19 | */
20 | @Mapper
21 | public interface UserAccessRuleMapper extends BaseMapper {
22 |
23 | @Select("select * from user_access_rule where user_id = #{userId}")
24 | List queryByUserId(@Param("userId") Long userId);
25 |
26 | @Select({
27 | ""
34 | })
35 | List queryByIds(@Param("ids") Set ids);
36 | }
37 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/mapper/UserInfoMapper.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.mapper;
2 |
3 | import com.baomidou.mybatisplus.core.mapper.BaseMapper;
4 | import com.blue.cat.bean.entity.UserInfo;
5 | import org.apache.ibatis.annotations.Param;
6 | import org.apache.ibatis.annotations.Select;
7 |
8 | /**
9 | *
10 | * Mapper 接口
11 | *
12 | *
13 | * @author lixin
14 | * @since 2023-05-07
15 | */
16 | public interface UserInfoMapper extends BaseMapper {
17 |
18 | @Select("select id,username,account, avatar ,status ,user_level as userLevel from user_info where account=#{account} and password=#{password} and status = 0 limit 1")
19 | UserInfo queryAccountAndPassword(@Param("account") String account, @Param("password") String password);
20 |
21 | @Select("select id from user_info where account = #{account} limit 1")
22 | UserInfo queryByAccount(@Param("account")String account);
23 |
24 | }
25 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/ILoginUserInfoService.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service;
2 |
3 | import com.blue.cat.bean.entity.LoginUserInfo;
4 | import com.baomidou.mybatisplus.extension.service.IService;
5 |
6 | /**
7 | *
8 | * 用户登录的id 服务类
9 | *
10 | *
11 | * @author lixin
12 | * @since 2023-05-11
13 | */
14 | public interface ILoginUserInfoService extends IService {
15 |
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/IPopupInfoService.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service;
2 |
3 | import com.baomidou.mybatisplus.extension.service.IService;
4 | import com.blue.cat.bean.entity.PopupInfo;
5 | import com.blue.cat.bean.entity.UserInfo;
6 |
7 | /**
8 | *
9 | * @author linyous
10 | * @since 2023-05-10
11 | */
12 | public interface IPopupInfoService extends IService {
13 |
14 | }
15 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/IUserAccessRuleService.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service;
2 |
3 | import com.blue.cat.bean.entity.UserAccessRule;
4 | import com.baomidou.mybatisplus.extension.service.IService;
5 |
6 | /**
7 | *
8 | * 用户访问规则 服务类
9 | *
10 | *
11 | * @author lixin
12 | * @since 2023-05-10
13 | */
14 | public interface IUserAccessRuleService extends IService {
15 |
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/IUserInfoService.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service;
2 |
3 | import com.baomidou.mybatisplus.extension.service.IService;
4 | import com.blue.cat.bean.entity.UserInfo;
5 |
6 | /**
7 | *
8 | * 服务类
9 | *
10 | *
11 | * @author lixin
12 | * @since 2023-05-07
13 | */
14 | public interface IUserInfoService extends IService {
15 |
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/impl/LoginUserInfoServiceImpl.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service.impl;
2 |
3 | import com.blue.cat.bean.entity.LoginUserInfo;
4 | import com.blue.cat.mapper.LoginUserInfoMapper;
5 | import com.blue.cat.service.ILoginUserInfoService;
6 | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
7 | import com.blue.cat.utils.CacheUtil;
8 | import lombok.extern.slf4j.Slf4j;
9 | import org.springframework.stereotype.Service;
10 | import org.springframework.transaction.annotation.Transactional;
11 |
12 | import java.util.ArrayList;
13 | import java.util.List;
14 | import java.util.Map;
15 | import java.util.concurrent.ConcurrentMap;
16 |
17 | /**
18 | *
19 | * 用户登录的id 服务实现类
20 | *
21 | *
22 | * @author lixin
23 | * @since 2023-05-11
24 | */
25 | @Slf4j
26 | @Service
27 | public class LoginUserInfoServiceImpl extends ServiceImpl implements ILoginUserInfoService {
28 |
29 | @Transactional
30 | public void buildLogInfoUser(){
31 | long startTime = System.currentTimeMillis();
32 | Boolean result = null;
33 | this.getBaseMapper().deleteAll();
34 | ConcurrentMap cacheUserInfo = CacheUtil.getAllCacheUserInfo();
35 | if(cacheUserInfo.size()>0){
36 | List loginUserInfos = new ArrayList<>();
37 | for (Map.Entry entry : cacheUserInfo.entrySet()) {
38 | LoginUserInfo loginUserInfo = new LoginUserInfo();
39 | loginUserInfo.setSessionId(entry.getKey());
40 | loginUserInfo.setUserInfo(entry.getValue());
41 | loginUserInfos.add(loginUserInfo);
42 | }
43 | result = this.saveBatch(loginUserInfos);
44 | }
45 | log.info("buildLoginUserInfo end cacheSize={} ,result={} cost={}",cacheUserInfo.size(),result,System.currentTimeMillis()-startTime);
46 | }
47 |
48 | }
49 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/impl/PopupInfoServiceImpl.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service.impl;
2 |
3 | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
4 | import com.blue.cat.bean.entity.PopupInfo;
5 | import com.blue.cat.mapper.PopupInfoMapper;
6 | import com.blue.cat.service.IPopupInfoService;
7 | import lombok.extern.slf4j.Slf4j;
8 | import org.springframework.stereotype.Service;
9 |
10 | import java.util.Optional;
11 |
12 | /**
13 | *
14 | * @author linyous
15 | * @since 2023-05-10
16 | */
17 | @Slf4j
18 | @Service
19 | public class PopupInfoServiceImpl extends ServiceImpl implements IPopupInfoService {
20 |
21 | /**
22 | * 查询最新公告的内容
23 | * @return
24 | */
25 | public Optional queryPopupInfo(String source) {
26 | PopupInfo popupInfo = super.baseMapper.getPopup(source);
27 | log.info("queryPopupInfo PopupInfo={}",popupInfo);
28 | if(popupInfo==null){
29 | return Optional.empty();
30 | }
31 | return Optional.of(popupInfo);
32 | }
33 |
34 | }
35 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/impl/UserAccessRuleServiceImpl.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service.impl;
2 |
3 | import com.blue.cat.bean.entity.UserAccessRule;
4 | import com.blue.cat.bean.vo.UserLevelAccessVo;
5 | import com.blue.cat.mapper.UserAccessRuleMapper;
6 | import com.blue.cat.service.IUserAccessRuleService;
7 | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
8 | import lombok.extern.slf4j.Slf4j;
9 | import org.apache.commons.collections.CollectionUtils;
10 | import org.springframework.stereotype.Service;
11 |
12 | import java.util.ArrayList;
13 | import java.util.List;
14 | import java.util.Map;
15 | import java.util.Set;
16 | import java.util.concurrent.atomic.AtomicInteger;
17 | import java.util.function.Function;
18 | import java.util.stream.Collectors;
19 |
20 | /**
21 | *
22 | * 用户访问规则 服务实现类
23 | *
24 | *
25 | * @author lixin
26 | * @since 2023-05-10
27 | */
28 | @Slf4j
29 | @Service
30 | public class UserAccessRuleServiceImpl extends ServiceImpl implements IUserAccessRuleService {
31 |
32 |
33 | public List getAllAccessRuleByUserId(Long userId){
34 |
35 | List userAccessRules = super.baseMapper.queryByUserId(userId);
36 | return userAccessRules.stream().map(this::entityToVo).collect(Collectors.toList());
37 | }
38 |
39 | /**
40 | * 默认添加默认访问规则配置
41 | * @return
42 | */
43 | public int addAccess(UserAccessRule accessRule){
44 | return super.baseMapper.insert(accessRule);
45 | }
46 |
47 |
48 | /**
49 | * 根据ids查询
50 | * @param ids
51 | * @return
52 | */
53 | public List getByIds(Set ids){
54 | return this.baseMapper.queryByIds(ids);
55 | }
56 |
57 | /**
58 | * 实体转化为vo
59 | * @param accessRule
60 | * @return
61 | */
62 | private UserLevelAccessVo entityToVo(UserAccessRule accessRule){
63 | UserLevelAccessVo vo = new UserLevelAccessVo();
64 | vo.setId(accessRule.getId());
65 | vo.setUseNumber(new AtomicInteger(accessRule.getUseNumber()));
66 | vo.setServiceType(accessRule.getServiceType());
67 | vo.setStartEffectiveTime(accessRule.getStartEffectiveTime());
68 | vo.setEndEffectiveTime(accessRule.getEndEffectiveTime());
69 | return vo;
70 | }
71 |
72 | /**
73 | * 批量更新 用户访问配置
74 | * @param accessRules
75 | * @return
76 | */
77 | public boolean batchUpdate(List accessRules){
78 | if(CollectionUtils.isEmpty(accessRules)){
79 | log.info("batchUpdate accessVos is empty");
80 | return false;
81 | }
82 | return this.updateBatchById(accessRules);
83 | }
84 |
85 |
86 |
87 | /**
88 | * vo转实体
89 | * @param vo
90 | * @return
91 | */
92 | private UserAccessRule voToEntity(UserLevelAccessVo vo){
93 | UserAccessRule entity = new UserAccessRule();
94 | entity.setId(vo.getId());
95 | entity.setStartEffectiveTime(vo.getStartEffectiveTime());
96 | entity.setEndEffectiveTime(vo.getEndEffectiveTime());
97 | entity.setUpdateUser("system_auto");
98 | entity.setUseNumber(vo.getUseNumber().get());
99 | return entity;
100 | }
101 |
102 | }
103 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/service/impl/UserInfoServiceImpl.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.service.impl;
2 |
3 | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
4 | import com.blue.cat.bean.entity.UserInfo;
5 | import com.blue.cat.mapper.UserInfoMapper;
6 | import com.blue.cat.service.IUserInfoService;
7 | import lombok.extern.slf4j.Slf4j;
8 | import org.springframework.stereotype.Service;
9 |
10 | import java.util.Optional;
11 |
12 | /**
13 | *
14 | * 服务实现类
15 | *
16 | *
17 | * @author lixin
18 | * @since 2023-05-07
19 | */
20 | @Slf4j
21 | @Service
22 | public class UserInfoServiceImpl extends ServiceImpl implements IUserInfoService {
23 |
24 | /**
25 | * 根据账号和密码查询用户信息
26 | * @param account
27 | * @param password
28 | * @return
29 | */
30 | public Optional queryUserByAccountAndPassword(String account, String password) {
31 | UserInfo userInfo = super.baseMapper.queryAccountAndPassword(account, password);
32 | log.info("login account={},userInfo={}",account,userInfo);
33 | if(userInfo==null){
34 | return Optional.empty();
35 | }
36 | return Optional.of(userInfo);
37 | }
38 |
39 | /**
40 | * 检测账号是否已经存在了
41 | * @param account
42 | * @return
43 | */
44 | public boolean checkExistAccount(String account){
45 | UserInfo userInfo = super.baseMapper.queryByAccount(account);
46 | log.info("checkExistAccount userInfo={}",userInfo);
47 | return userInfo!=null;
48 | }
49 |
50 | /**
51 | * 添加用户
52 | * @param userInfo
53 | * @return
54 | */
55 | public int addUser(UserInfo userInfo){
56 | return this.baseMapper.insert(userInfo);
57 | }
58 | }
59 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/CacheUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.blue.cat.bean.constants.CommonConstant;
4 | import com.google.common.cache.Cache;
5 | import com.google.common.cache.CacheBuilder;
6 |
7 | import java.util.Collection;
8 | import java.util.concurrent.ConcurrentMap;
9 | import java.util.concurrent.TimeUnit;
10 |
11 | public class CacheUtil {
12 |
13 | private static Cache cache = CacheBuilder.newBuilder().expireAfterWrite(CommonConstant.CACHE_TIME_OUT, TimeUnit.SECONDS).build();
14 |
15 | public static void put(String key, String value) {
16 | cache.put(key, value);
17 | }
18 |
19 | public static String getIfPresent(String key){
20 | return cache.getIfPresent(key);
21 | }
22 |
23 |
24 | /**
25 | * 获取所有的登录的用户信息
26 | * @return
27 | */
28 | public static ConcurrentMap getAllCacheUserInfo(){
29 | return cache.asMap();
30 | }
31 |
32 | /**
33 | * 获取所有的缓存的用户信息
34 | * @return
35 | */
36 | public static Collection getUserInfo(){
37 | return cache.asMap().values();
38 | }
39 |
40 |
41 | public static void invalidate(String key){
42 | cache.invalidate(key);
43 | }
44 |
45 | /**
46 | * 删除登录缓存
47 | * @param key
48 | */
49 | public static void removeCache(String key){
50 | cache.asMap().remove(key);
51 | }
52 |
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/ChatGptUtils.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.alibaba.fastjson.JSONObject;
4 | import com.blue.cat.bean.gpt.ChoiceMessage;
5 | import com.blue.cat.bean.gpt.GptTurbo;
6 | import com.blue.cat.bean.gpt.OpenReq;
7 | import org.apache.commons.lang3.StringUtils;
8 |
9 | import java.util.ArrayList;
10 | import java.util.HashMap;
11 | import java.util.List;
12 | import java.util.Map;
13 |
14 | /**
15 | * @author huyd
16 | * @date 2023/5/5 11:42 PM
17 | */
18 | public class ChatGptUtils {
19 |
20 | public static void main(String[] args) throws Exception{
21 | String chat = chatWithSangiu(null, "西红柿炒鸡蛋怎么做");
22 | System.out.println(chat);
23 | }
24 |
25 | public static String chatWithSangiu(String systemDesc, String message) throws Exception {
26 | //String req = "http://amd2.watilion.top:8088/api/open/req";
27 | String req = "http://localhost:8088/api/open/req";
28 | String method = "POST";
29 | Map map = new HashMap<>();
30 | map.put("Content-Type", "application/json");
31 | map.put("Authorization", "Bearer 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");
32 |
33 | String url = "https://api.openai.com/v1/chat/completions";
34 | List list = new ArrayList<>();
35 | if (StringUtils.isNotBlank(systemDesc)) {
36 | list.add(new ChoiceMessage("system", systemDesc));
37 | }
38 | list.add(new ChoiceMessage("user", "user question: " + message));
39 |
40 | GptTurbo gptTurbo = new GptTurbo();
41 | gptTurbo.setModel("gpt-3.5-turbo");
42 | gptTurbo.setMessages(list);
43 | String body = JSONObject.toJSONString(gptTurbo);
44 |
45 | OpenReq reqGpt = OpenReq.builder().method(method).url(url).body(body).headers(map).build();
46 |
47 | Map headers = new HashMap<>(2);
48 | headers.put("Content-Type", "application/json");
49 | headers.put("gpt-token", "OldSanGiuNBPlus666");
50 |
51 | String result = OkHttpUtils.postReq(req, JSONObject.toJSONString(reqGpt), headers);
52 | if(StringUtils.isBlank(result)){
53 | return null;
54 | }
55 |
56 | JSONObject jsonObject = JSONObject.parseObject(result);
57 | int code = jsonObject.getIntValue("code");
58 | if(code != 0){
59 | return null;
60 | }
61 | String msg = jsonObject.getString("msg");
62 | JSONObject msgJb = JSONObject.parseObject(msg);
63 | Object choices = msgJb.getJSONArray("choices").get(0);
64 | Object messageContent = JSONObject.parseObject(choices.toString()).get("message");
65 | return JSONObject.parseObject(messageContent.toString()).getString("content");
66 | }
67 |
68 | }
69 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/HttpUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import eu.bitwalker.useragentutils.Browser;
4 | import eu.bitwalker.useragentutils.OperatingSystem;
5 | import eu.bitwalker.useragentutils.UserAgent;
6 |
7 | import javax.servlet.http.HttpServletRequest;
8 | import java.io.BufferedReader;
9 | import java.io.IOException;
10 | import java.io.InputStreamReader;
11 | import java.util.regex.Matcher;
12 | import java.util.regex.Pattern;
13 |
14 | public class HttpUtil {
15 |
16 | public static String browserName(){
17 | HttpServletRequest request = WebUtils.getRequest();
18 | String userAgent = request.getHeader("User-Agent");
19 | UserAgent ua = UserAgent.parseUserAgentString(userAgent);
20 | Browser browser = ua.getBrowser();
21 | return browser.getName() + "-" + browser.getVersion(userAgent);
22 | }
23 |
24 | /**
25 | * 根据IP获取MAC地址
26 | * @return
27 | */
28 | public static String getMacAddrByIp(String ip) {
29 | String macAddr = null;
30 | try {
31 | Process process = Runtime.getRuntime().exec("nbtstat -a " + ip);
32 | BufferedReader br = new BufferedReader(
33 | new InputStreamReader(process.getInputStream()));
34 | Pattern pattern = Pattern.compile("([A-F0-9]{2}-){5}[A-F0-9]{2}");
35 | Matcher matcher;
36 | for (String strLine = br.readLine(); strLine != null;
37 | strLine = br.readLine()) {
38 | matcher = pattern.matcher(strLine);
39 | if (matcher.find()) {
40 | macAddr = matcher.group();
41 | break;
42 | }
43 | }
44 | } catch (IOException e) {
45 | e.printStackTrace();
46 | }
47 | return macAddr;
48 | }
49 |
50 | public static String osName(HttpServletRequest request){
51 | String userAgent = request.getHeader("User-Agent");
52 | UserAgent ua = UserAgent.parseUserAgentString(userAgent);
53 | OperatingSystem os = ua.getOperatingSystem();
54 | return os.getName();
55 | }
56 |
57 | /**
58 | * 用户的ip
59 | * @param request
60 | * @return
61 | */
62 | public static String getIpAddress() {
63 | HttpServletRequest request = WebUtils.getRequest();
64 | String ip = request.getHeader("X-Forwarded-For");
65 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
66 | ip = request.getHeader("Proxy-Client-IP");
67 | }
68 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
69 | ip = request.getHeader("WL-Proxy-Client-IP");
70 | }
71 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
72 | ip = request.getHeader("HTTP_X_FORWARDED_FOR");
73 | }
74 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
75 | ip = request.getHeader("HTTP_X_FORWARDED");
76 | }
77 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
78 | ip = request.getHeader("HTTP_X_CLUSTER_CLIENT_IP");
79 | }
80 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
81 | ip = request.getHeader("HTTP_CLIENT_IP");
82 | }
83 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
84 | ip = request.getHeader("HTTP_FORWARDED_FOR");
85 | }
86 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
87 | ip = request.getHeader("HTTP_FORWARDED");
88 | }
89 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
90 | ip = request.getHeader("HTTP_VIA");
91 | }
92 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
93 | ip = request.getHeader("REMOTE_ADDR");
94 | }
95 | if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
96 | ip = request.getRemoteAddr();
97 | }
98 | if (ip.contains(",")) {
99 | return ip.split(",")[0];
100 | } else {
101 | return ip;
102 | }
103 | }
104 | }
105 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/IpCacheUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.blue.cat.bean.constants.LimitEnum;
4 | import com.google.common.cache.Cache;
5 | import com.google.common.cache.CacheBuilder;
6 |
7 | import java.util.concurrent.TimeUnit;
8 | import java.util.concurrent.atomic.AtomicInteger;
9 |
10 | public class IpCacheUtil {
11 |
12 | private static Cache cache = CacheBuilder.newBuilder().expireAfterWrite(LimitEnum.IP.getTime(), TimeUnit.SECONDS).build();
13 |
14 | public static void put(String key, AtomicInteger value) {
15 | cache.put(key, value);
16 | }
17 |
18 | public static AtomicInteger getIfPresent(String key){
19 | return cache.getIfPresent(key);
20 | }
21 |
22 | /**
23 | * 删除缓存
24 | * @param key
25 | */
26 | public static void removeCache(String key){
27 | cache.asMap().remove(key);
28 | }
29 |
30 | }
31 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/MybatisPlusGeneratorUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.baomidou.mybatisplus.annotation.DbType;
4 | import com.baomidou.mybatisplus.generator.AutoGenerator;
5 | import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
6 | import com.baomidou.mybatisplus.generator.config.GlobalConfig;
7 | import com.baomidou.mybatisplus.generator.config.PackageConfig;
8 | import com.baomidou.mybatisplus.generator.config.StrategyConfig;
9 | import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
10 |
11 | /**
12 | * @author melo
13 | * @date 2020/5/31
14 | */
15 | public class MybatisPlusGeneratorUtil {
16 | public static void main(String[] args) {
17 | String author = "lixin";
18 | String database = "bluecat";
19 | String tableName = "discern_record";
20 | generateByTables(author, database,tableName);
21 | }
22 |
23 | private static void generateByTables(String author, String database, String... tableNames) {
24 | GlobalConfig config = new GlobalConfig();
25 | String dbUrl = "jdbc:mysql://127.0.0.1:3306/" + database + "?useUnicode=true&characterEncoding=utf8&serverTimezone=UTC";
26 | config.setActiveRecord(false)
27 | .setAuthor(author)
28 | .setOutputDir(System.getProperty("user.dir") +"/src/main/java/")
29 | .setFileOverride(true)
30 | .setEnableCache(false)
31 | .setEntityName("%s")
32 | .setMapperName("%sMapper")
33 | // .setServiceName("%sService")
34 | //.setServiceImplName("%sServiceImpl")
35 | .setOpen(false);
36 |
37 | DataSourceConfig dataSourceConfig = new DataSourceConfig();
38 | dataSourceConfig.setDbType(DbType.MYSQL)
39 | .setUrl(dbUrl)
40 | .setUsername("root")
41 | // .setPassword("root")
42 | .setDriverName("com.mysql.cj.jdbc.Driver");
43 |
44 | StrategyConfig strategyConfig = new StrategyConfig();
45 | strategyConfig
46 | .setCapitalMode(true)
47 | .setEntityLombokModel(true)
48 | .setNaming(NamingStrategy.underline_to_camel)
49 | // .setSuperMapperClass("")
50 | .setInclude(tableNames);//修改替换成你需要的表名，多个表名传数组
51 |
52 | new AutoGenerator().setGlobalConfig(config)
53 | .setDataSource(dataSourceConfig)
54 | .setStrategy(strategyConfig)
55 | .setPackageInfo(
56 | new PackageConfig()
57 | .setParent("com.blue.cat")
58 | .setEntity("bean.entity.DiscernRecord")
59 | .setMapper("mapper")
60 | //.setService("service")
61 | //.setServiceImpl("service.impl")
62 | //.setXml("mybatis.mappers")
63 | ).execute();
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/OkHttpUtils.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import lombok.extern.slf4j.Slf4j;
4 | import okhttp3.*;
5 |
6 | import java.util.Map;
7 | import java.util.concurrent.TimeUnit;
8 |
9 | /**
10 | * 请求微信接口获取相关Json信息
11 | */
12 | @Slf4j
13 | public class OkHttpUtils {
14 |
15 | /**
16 | * OkHttpClient使用了连接池，应保持一个实例存在
17 | * 可以定制配置连接池
18 | */
19 | private static final OkHttpClient okHttpClient = new OkHttpClient().newBuilder()
20 | .connectTimeout(180,TimeUnit.SECONDS)
21 | .readTimeout(180,TimeUnit.SECONDS)
22 | .writeTimeout(180,TimeUnit.SECONDS)
23 | .build();
24 |
25 | public static String getReq(String url, Map headers) throws Exception {
26 | Request.Builder request = new Request.Builder().url(url);
27 | if(!headers.isEmpty()){
28 | for (Map.Entry entry : headers.entrySet()) {
29 | request.addHeader(entry.getKey(), entry.getValue());
30 | }
31 | }
32 |
33 | Response response = okHttpClient.newCall(request.build()).execute();
34 | if (response.body() == null) {
35 | return null;
36 | }
37 | String result = response.body().string();
38 | response.close();
39 | return result;
40 | }
41 |
42 | public static String postReq(String url, String body, Map headers) throws Exception {
43 | Request.Builder req = new Request.Builder().url(url);
44 | if(!headers.isEmpty()){
45 | for (Map.Entry entry : headers.entrySet()) {
46 | req.addHeader(entry.getKey(), entry.getValue());
47 | }
48 | }
49 |
50 | if(body != null){
51 | RequestBody requestBody = RequestBody.create(MediaType.parse("application/json"), body);
52 | req.post(requestBody);
53 | }
54 |
55 | Response response = okHttpClient.newCall(req.build()).execute();
56 | String result = response.body() == null ? null : response.body().string();
57 | response.close();
58 | return result;
59 | }
60 | }
61 |
62 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/OpenGptUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | /**
4 | * @author huyd
5 | * @date 2023/5/2 7:43 PM
6 | */
7 |
8 | import com.alibaba.fastjson.JSONObject;
9 | import com.blue.cat.bean.gpt.ChatReq;
10 | import com.blue.cat.bean.gpt.ImageReq;
11 | import com.blue.cat.bean.gpt.drawImageRes;
12 | import com.google.common.collect.Lists;
13 | import com.theokanning.openai.completion.chat.ChatCompletionRequest;
14 | import com.theokanning.openai.completion.chat.ChatMessage;
15 | import lombok.extern.slf4j.Slf4j;
16 |
17 | import javax.servlet.http.HttpServletResponse;
18 | import java.io.*;
19 | import java.net.HttpURLConnection;
20 | import java.net.URL;
21 | import java.net.URLEncoder;
22 | import java.nio.charset.Charset;
23 | import java.nio.charset.StandardCharsets;
24 |
25 | @Slf4j
26 | public class OpenGptUtil {
27 |
28 | public static void main(String[] args) {
29 | }
30 |
31 | public static void stramChatV2() {
32 | try {
33 | ChatMessage message = new ChatMessage();
34 | message.setContent("香辣嗦螺怎么做好吃");
35 | message.setRole("system");
36 |
37 | ChatCompletionRequest request = new ChatCompletionRequest();
38 | request.setMessages(Lists.newArrayList(message));
39 | request.setUser("123");
40 | request.setModel("gpt-3.5-turbo");
41 |
42 | // 指定接口地址
43 | String req = "http://localhost:8088/api/stream?request=%s";
44 | URL url = new URL(String.format(req, URLEncoder.encode(JSONObject.toJSONString(request), "UTF-8")));
45 |
46 | HttpURLConnection connection = (HttpURLConnection) url.openConnection();
47 | connection.setRequestMethod("GET");
48 | connection.setRequestProperty("Accept", "text/event-stream");
49 |
50 | InputStream inputStream = connection.getInputStream();
51 | BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
52 |
53 | String line;
54 | while ((line = reader.readLine()) != null) {
55 | System.out.println(line);
56 | }
57 | reader.close();
58 | connection.disconnect();
59 | } catch (Exception e) {
60 | e.printStackTrace();
61 | }
62 | }
63 |
64 | public static void setBodyParameter(String str, HttpURLConnection conn) throws IOException {
65 | try (DataOutputStream out = new DataOutputStream(conn.getOutputStream())) {
66 | out.write(str.getBytes(StandardCharsets.UTF_8));
67 | out.flush();
68 | } catch (Exception e) {
69 | System.out.println("setBodyParameter error:" + e);
70 | }
71 | }
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/RegexUtil.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import java.util.regex.Matcher;
4 | import java.util.regex.Pattern;
5 |
6 | public class RegexUtil {
7 |
8 |
9 | /**
10 | * 验证手机号
11 | * @param hex
12 | * @return
13 | */
14 | public static boolean validatePhone(final String hex) {
15 | Pattern pattern;
16 | Matcher matcher;
17 | String PHONE_PATTERN = "^\\d{11}$";
18 | pattern = Pattern.compile(PHONE_PATTERN);
19 | matcher = pattern.matcher(hex);
20 | return matcher.matches();
21 | }
22 |
23 |
24 |
25 | /**
26 | * 验证手机号
27 | * @param hex
28 | * @return
29 | */
30 | public static boolean validateEmail(final String hex) {
31 | Pattern pattern;
32 | Matcher matcher;
33 | String EMAIL_PATTERN =
34 | "^[_A-Za-z0-9-\\+]+(\\.[_A-Za-z0-9-]+)*@"
35 | + "[A-Za-z0-9-]+(\\.[A-Za-z0-9]+)*(\\.[A-Za-z]{2,})$";
36 | pattern = Pattern.compile(EMAIL_PATTERN);
37 | matcher = pattern.matcher(hex);
38 | return matcher.matches();
39 | }
40 |
41 | }
42 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/ResultVO.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.blue.cat.bean.common.BaseCode;
4 | import com.blue.cat.bean.common.BaseCodeEnum;
5 |
6 | /**
7 | * web接口统一返回结果
8 | */
9 | public class ResultVO {
10 |
11 | /**
12 | * 响应状态码
13 | */
14 | private int code;
15 |
16 | /**
17 | * 响应消息
18 | */
19 | private String msg;
20 |
21 | /**
22 | * 响应数据
23 | */
24 | private T data;
25 |
26 | public static ResultVO build(BaseCode baseCode, T data) {
27 | return new ResultVO<>(baseCode, data);
28 | }
29 |
30 | public static ResultVO build(BaseCode baseCode) {
31 | return new ResultVO<>(baseCode);
32 | }
33 |
34 | public static ResultVO success() {
35 | return new ResultVO<>(BaseCodeEnum.SUCCESS);
36 | }
37 |
38 | public static ResultVO success(String msg) {
39 | return new ResultVO<>(BaseCodeEnum.SUCCESS.getCode(), msg);
40 | }
41 |
42 | public static ResultVO success(T data) {
43 | return new ResultVO<>(BaseCodeEnum.SUCCESS, data);
44 | }
45 |
46 | public static ResultVO error(BaseCode baseCode) {
47 | return new ResultVO<>(baseCode.getCode(), baseCode.getMsg());
48 | }
49 |
50 | public static ResultVO error(int code, String msg) {
51 | return new ResultVO<>(code, msg);
52 | }
53 |
54 | public static ResultVO fail(int code, String msg) {
55 | return new ResultVO<>(code, msg);
56 | }
57 |
58 | public static ResultVO fail(BaseCode baseCode) {
59 | return new ResultVO<>(baseCode.getCode(), baseCode.getMsg());
60 | }
61 |
62 | public static ResultVO fail(String msg) {
63 | return new ResultVO<>(BaseCodeEnum.FAIL.getCode(), msg);
64 | }
65 |
66 | public ResultVO() {
67 | }
68 |
69 | public ResultVO(int code, String msg) {
70 | this.code = code;
71 | this.msg = msg;
72 | }
73 |
74 | public ResultVO(BaseCode baseCode) {
75 | this.code = baseCode.getCode();
76 | this.msg = baseCode.getMsg();
77 | }
78 |
79 | public ResultVO(BaseCode baseCode, T data) {
80 | this.code = baseCode.getCode();
81 | this.msg = baseCode.getMsg();
82 | this.data = data;
83 | }
84 |
85 | public int getCode() {
86 | return code;
87 | }
88 |
89 | public void setCode(int code) {
90 | this.code = code;
91 | }
92 |
93 | public String getMsg() {
94 | return msg;
95 | }
96 |
97 | public void setMsg(String msg) {
98 | this.msg = msg;
99 | }
100 |
101 | public T getData() {
102 | return data;
103 | }
104 |
105 | public void setData(T data) {
106 | this.data = data;
107 | }
108 | }
109 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/SessionUser.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import com.blue.cat.bean.vo.CacheUserInfoVo;
4 | import org.apache.commons.lang3.StringUtils;
5 |
6 | import java.util.HashSet;
7 | import java.util.Optional;
8 | import java.util.Set;
9 |
10 | public class SessionUser {
11 |
12 | private static ThreadLocal sessionUser = new ThreadLocal<>();
13 |
14 | public static void setSessionUserInfo(CacheUserInfoVo adminUserVO) {
15 | sessionUser.set(adminUserVO);
16 | }
17 |
18 | public static CacheUserInfoVo get(){return sessionUser.get();}
19 |
20 | public static Long getUserId(){
21 | CacheUserInfoVo cacheUserInfoVo = sessionUser.get();
22 | if(cacheUserInfoVo == null){
23 | return null;
24 | }
25 | return cacheUserInfoVo.getId();
26 | }
27 |
28 | /**
29 | * 用户登录的信息
30 | * @return
31 | */
32 | public static Optional getUserInfoVO(){
33 | CacheUserInfoVo cacheUserInfoVo = sessionUser.get();
34 | if(cacheUserInfoVo == null){
35 | return Optional.empty();
36 | }
37 | return Optional.of(cacheUserInfoVo);
38 | }
39 |
40 | public static Optional getUserName(){
41 | CacheUserInfoVo cacheUserInfoVo = sessionUser.get();
42 | if(cacheUserInfoVo == null){
43 | return Optional.empty();
44 | }
45 | return Optional.of(cacheUserInfoVo.getUsername());
46 | }
47 |
48 | public static String getAccount(){
49 | CacheUserInfoVo cacheUserInfoVo = sessionUser.get();
50 | if(cacheUserInfoVo == null){
51 | return null;
52 | }
53 | return cacheUserInfoVo.getAccount();
54 | }
55 |
56 | public static void remove(){
57 | sessionUser.remove();
58 | }
59 |
60 |
61 | /**
62 | * 管理账号集合，目前直接写死到项目中
63 | */
64 | private static Set adminSet = new HashSet<>();
65 |
66 | static {
67 | adminSet.add("xxxxx");// laoban
68 | }
69 |
70 | /**
71 | * 判断是否是管理账号
72 | * @param account
73 | * @return
74 | */
75 | public static boolean isAdmin(String account){
76 | if(StringUtils.isEmpty(account)){
77 | return false;
78 | }
79 | return adminSet.contains(account);
80 | }
81 |
82 | public static boolean isAdmin(){
83 | String account = getAccount();
84 | return isAdmin(account);
85 | }
86 |
87 | }
88 |
--------------------------------------------------------------------------------
/src/main/java/com/blue/cat/utils/WebUtils.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.utils;
2 |
3 | import org.springframework.web.context.request.RequestAttributes;
4 | import org.springframework.web.context.request.RequestContextHolder;
5 | import org.springframework.web.context.request.ServletRequestAttributes;
6 |
7 | import javax.servlet.http.HttpServletRequest;
8 | import javax.servlet.http.HttpServletResponse;
9 | import javax.servlet.http.HttpSession;
10 |
11 | /**
12 | * @author Ths Sun
13 | * @create 2020/7/30.
14 | */
15 | public class WebUtils {
16 |
17 | /**
18 | * 获取request
19 | */
20 | public static HttpServletRequest getRequest()
21 | {
22 | return getRequestAttributes().getRequest();
23 | }
24 |
25 | /**
26 | * 获取response
27 | */
28 | public static HttpServletResponse getResponse()
29 | {
30 | return getRequestAttributes().getResponse();
31 | }
32 |
33 | /**
34 | * 获取session
35 | */
36 | public static HttpSession getSession()
37 | {
38 | return getRequest().getSession();
39 | }
40 |
41 | public static ServletRequestAttributes getRequestAttributes(){
42 | RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
43 | return (ServletRequestAttributes) attributes;
44 | }
45 |
46 | }
47 |
--------------------------------------------------------------------------------
/src/main/resources/application-dev.yml:
--------------------------------------------------------------------------------
1 | server:
2 | port: 8080
3 | servlet:
4 | context-path: /api
5 |
6 | # 填写自己的open-api key
7 | chatgpt:
8 | token:
9 | session-expiration-time: 2
10 |
11 | spring:
12 | http:
13 | encoding:
14 | charset: UTF-8
15 |
16 | servlet:
17 | multipart:
18 | enabled: true
19 | max-file-size: 500MB
20 | max-request-size: 500MB
21 |
22 | datasource:
23 | driver-class-name: com.mysql.cj.jdbc.Driver
24 | url: jdbc:mysql://127.0.0.1:3306/bluecat?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
25 | username: root
26 | # password: root
27 |
28 |
29 | mybatis-plus:
30 | configuration:
31 | map-underscore-to-camel-case: true
32 | global-config:
33 | db-config:
34 | column-underline: true
35 |
--------------------------------------------------------------------------------
/src/main/resources/application-prod.yml:
--------------------------------------------------------------------------------
1 | server:
2 | port: 8080
3 | servlet:
4 | context-path: /api
5 |
6 | #填写自己的open api的token
7 | chatgpt:
8 | token:
9 | session-expiration-time: 2
10 |
11 | spring:
12 | http:
13 | encoding:
14 | charset: UTF-8
15 |
16 | servlet:
17 | multipart:
18 | enabled: true
19 | max-file-size: 500MB
20 | max-request-size: 500MB
21 |
22 | datasource:
23 | driver-class-name: com.mysql.cj.jdbc.Driver
24 | url: jdbc:mysql://127.0.0.1:3306/bluecat?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai
25 | username: root
26 | password: root
27 |
28 | mybatis-plus:
29 | configuration:
30 | map-underscore-to-camel-case: true
31 | global-config:
32 | db-config:
33 | column-underline: true
34 |
--------------------------------------------------------------------------------
/src/test/java/com/blue/cat/BlueCatApplicationTests.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat;
2 |
3 | import io.github.asleepyfish.util.OpenAiUtils;
4 | import org.junit.jupiter.api.Test;
5 |
6 |
7 | class BlueCatApplicationTests extends RunnerTest {
8 |
9 | @Test
10 | public void testGenerateImg() {
11 | OpenAiUtils.createImage("英短").forEach(System.out::println);
12 | }
13 |
14 | @Test
15 | public void createStreamChatCompletion() {
16 | System.out.println("tets");
17 | }
18 |
19 |
20 | }
21 |
--------------------------------------------------------------------------------
/src/test/java/com/blue/cat/RunnerTest.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat;
2 |
3 | import org.springframework.boot.test.context.SpringBootTest;
4 | import org.springframework.test.context.ActiveProfiles;
5 |
6 |
7 | /**
8 | * 所有测试类都需要继承此类
9 | */
10 | @ActiveProfiles(value = "dev")
11 | @SpringBootTest
12 | public class RunnerTest {
13 |
14 |
15 |
16 | }
17 |
--------------------------------------------------------------------------------
/src/test/java/com/blue/cat/access/AccessTest.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.access;
2 |
3 | import com.blue.cat.RunnerTest;
4 | import com.blue.cat.bean.entity.UserAccessRule;
5 | import com.blue.cat.bean.util.UserAccessRuleUtil;
6 | import com.blue.cat.mapper.UserAccessRuleMapper;
7 | import com.blue.cat.service.impl.UserAccessRuleServiceImpl;
8 | import org.junit.jupiter.api.Test;
9 | import org.springframework.beans.factory.annotation.Autowired;
10 |
11 | import javax.annotation.Resource;
12 | import java.util.Arrays;
13 | import java.util.HashSet;
14 | import java.util.List;
15 |
16 | public class AccessTest extends RunnerTest {
17 |
18 | @Autowired
19 | private UserAccessRuleServiceImpl accessRuleService;
20 |
21 | @Resource
22 | private UserAccessRuleMapper ruleMapper;
23 |
24 | @Test
25 | public void addAccess(){
26 | int i = accessRuleService.addAccess(UserAccessRuleUtil.getAccessRuleGpt3(1234L));
27 | System.out.println(i);
28 | }
29 |
30 |
31 | @Test
32 | public void queryIds(){
33 | List userAccessRules = ruleMapper.queryByIds(new HashSet<>(Arrays.asList(1656217949066317826L)));
34 | System.out.println(userAccessRules);
35 | }
36 |
37 | }
38 |
--------------------------------------------------------------------------------
/src/test/java/com/blue/cat/user/UserInfoTest.java:
--------------------------------------------------------------------------------
1 | package com.blue.cat.user;
2 |
3 |
4 | import com.blue.cat.RunnerTest;
5 | import com.blue.cat.bean.req.UserInfoLoginReq;
6 | import com.blue.cat.bean.req.UserInfoRegisterReq;
7 | import com.blue.cat.controller.UserInfoController;
8 | import com.blue.cat.mapper.UserInfoMapper;
9 | import com.blue.cat.utils.ResultVO;
10 | import org.junit.jupiter.api.Test;
11 | import org.springframework.beans.factory.annotation.Autowired;
12 |
13 | public class UserInfoTest extends RunnerTest {
14 |
15 | @Autowired
16 | UserInfoMapper userInfoMapper;
17 |
18 | @Autowired
19 | UserInfoController userInfoController;
20 |
21 | @Test
22 | public void login() {
23 | UserInfoLoginReq loginReq = new UserInfoLoginReq();
24 | loginReq.setAccount("admin4");
25 | loginReq.setPassword("admin");
26 | ResultVO login = userInfoController.login(loginReq,null);
27 | assert login !=null;
28 | }
29 |
30 | @Test
31 | public void register(){
32 | UserInfoRegisterReq registerReq = new UserInfoRegisterReq();
33 | registerReq.setAccount("admin");
34 | registerReq.setPassword("admin");
35 | registerReq.setUsername("ai助手-test");
36 | ResultVO register = userInfoController.register(registerReq);
37 | assert register!=null;
38 | }
39 | }
40 |
--------------------------------------------------------------------------------
/执行SQL/blueCat.sql:
--------------------------------------------------------------------------------
1 | /*
2 | Navicat MySQL Data Transfer
3 |
4 | Source Server : localhost_3306
5 | Source Server Version : 50712
6 | Source Host : localhost:3306
7 | Source Database : driver
8 |
9 | Target Server Type : MYSQL
10 | Target Server Version : 50712
11 | File Encoding : 65001
12 |
13 | Date: 2023-06-02 08:26:13
14 | */
15 |
16 | SET FOREIGN_KEY_CHECKS=0;
17 |
18 | -- ----------------------------
19 | -- Table structure for `login_user_info`
20 | -- ----------------------------
21 | DROP TABLE IF EXISTS `login_user_info`;
22 | CREATE TABLE `login_user_info` (
23 | `id` bigint(22) NOT NULL,
24 | `session_id` varchar(64) NOT NULL COMMENT 'session_key',
25 | `user_info` varchar(512) NOT NULL COMMENT '用户登录信息的json串',
26 | PRIMARY KEY (`id`)
27 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用户登录的id';
28 |
29 | -- ----------------------------
30 | -- Records of login_user_info
31 | -- ----------------------------
32 |
33 | -- ----------------------------
34 | -- Table structure for `popup_info`
35 | -- ----------------------------
36 | DROP TABLE IF EXISTS `popup_info`;
37 | CREATE TABLE `popup_info` (
38 | `id` int(11) NOT NULL AUTO_INCREMENT,
39 | `title` varchar(255) NOT NULL,
40 | `content` text NOT NULL,
41 | `createTime` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
42 | `popupLocation` varchar(255) DEFAULT NULL,
43 | `isShow` int(11) NOT NULL DEFAULT '0',
44 | PRIMARY KEY (`id`)
45 | ) ENGINE=InnoDB AUTO_INCREMENT=41 DEFAULT CHARSET=utf8mb4;
46 |
47 | -- ----------------------------
48 | -- Records of popup_info
49 | -- ----------------------------
50 | INSERT INTO `popup_info` VALUES ('11', '蓝猫AI', '蓝猫AI目前已开源，源码地址 ，
ChatGpt3.5功能会一直免费下去，不限速，不限制次数。', '2023-05-30 23:57:42', 'login', '1');
51 | INSERT INTO `popup_info` VALUES ('40', '蓝猫AI第1版公告', '蓝猫AI目前已开源，源码地址：点击联系我们', '2023-05-30 23:55:16', 'index', '1');
52 |
53 | -- ----------------------------
54 | -- Table structure for `prompt_model`
55 | -- ----------------------------
56 | DROP TABLE IF EXISTS `prompt_model`;
57 | CREATE TABLE `prompt_model` (
58 | `id` bigint(20) NOT NULL AUTO_INCREMENT,
59 | `type` varchar(100) NOT NULL COMMENT '分类',
60 | `title` varchar(255) CHARACTER SET utf8mb4 NOT NULL COMMENT '标题',
61 | `introduce` varchar(255) CHARACTER SET utf8mb4 DEFAULT NULL COMMENT '简介',
62 | `demo` varchar(255) CHARACTER SET utf8mb4 DEFAULT NULL COMMENT '示例',
63 | `content` varchar(1000) CHARACTER SET utf8mb4 NOT NULL COMMENT '提示内容',
64 | `state` tinyint(4) NOT NULL COMMENT '0，无效；1，有效',
65 | `sort` tinyint(4) NOT NULL DEFAULT '0' COMMENT '排序值',
66 | `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
67 | `update_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
68 | PRIMARY KEY (`id`)
69 | ) ENGINE=InnoDB AUTO_INCREMENT=9 DEFAULT CHARSET=utf8;
70 |
71 | -- ----------------------------
72 | -- Records of prompt_model
73 | -- ----------------------------
74 | INSERT INTO `prompt_model` VALUES ('1', 'Java', 'Java代码优化', '只需要将你想要的优化的代码复制到输入框，AI将会自动帮你优化它', '请输入/粘贴你想要优化的代码', '假如你是高级Java开发工程师，有着丰富的代码编写能力，那么请使用JDK8新语法将下面的代码进行优化并且将优化后的代码能够被markdowm渲染的代码出来：', '1', '0', '2023-05-21 22:42:46', '2023-05-21 22:56:34');
75 | INSERT INTO `prompt_model` VALUES ('7', 'Java', '使用JDK8新特性优化代码', '只需要将你想要的优化的代码复制到输入框，AI将会自动使用JDK8新语法帮你优化它', '请输入/粘贴你想要优化的代码', '假如你是高级Java开发工程师，非常擅长使用JDK8新语法，那么请使用JDK8新语法将下面的代码进行优化并且将优化后的代码能够被markdowm渲染的代码出来：', '1', '0', '2023-05-21 22:47:14', '2023-05-21 22:55:46');
76 | INSERT INTO `prompt_model` VALUES ('8', 'Java', 'Java面试打分工具', '只需要将你的问题和回答复制到输入框，AI将会自动根据你的问题和回答进行分析和打分', '问题：在synchronized代码块中调用wait方法进入等待的线程和因为拿不到锁而等待线程是否同一种状态？blocking？waiting？\r\n回答：1. 因为拿不到锁会处于 blocking状态；\r\n2. 调用wait方法，会处于 waiting状态，不会释放锁。\r\n', '假如你是Java面试官，有着非常丰富的面试经验，那么请基于下面的问题和回答进行系统的分析，给出相应的分数，并且以markdown的格式分点输出回答中不足的地方并且加以补充：', '1', '0', '2023-05-21 22:58:02', '2023-05-21 23:15:45');
77 |
78 | -- ----------------------------
79 | -- Table structure for `user_access_rule`
80 | -- ----------------------------
81 | DROP TABLE IF EXISTS `user_access_rule`;
82 | CREATE TABLE `user_access_rule` (
83 | `id` bigint(22) NOT NULL COMMENT '用户id',
84 | `user_id` bigint(22) NOT NULL,
85 | `service_type` varchar(32) COLLATE utf8mb4_bin NOT NULL COMMENT '模型标识',
86 | `use_number` tinyint(4) NOT NULL DEFAULT '10' COMMENT '使用次数, 如果 = -2 代表不限制次数',
87 | `start_effective_time` datetime NOT NULL COMMENT '开始生效时间',
88 | `end_effective_time` datetime NOT NULL COMMENT '有效结束时间',
89 | `create_time` datetime DEFAULT NULL COMMENT '创建时间',
90 | `update_time` datetime DEFAULT NULL COMMENT '更新时间',
91 | `update_user` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
92 | PRIMARY KEY (`id`)
93 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin COMMENT='用户访问规则';
94 |
95 | -- ----------------------------
96 | -- Records of user_access_rule
97 | -- ----------------------------
98 | INSERT INTO `user_access_rule` VALUES ('1664427904947597313', '1664427904784019457', 'chat_gpt_model3.5', '30', '2023-06-02 08:25:11', '2023-06-03 08:25:11', '2023-06-02 08:25:11', null, null);
99 |
100 | -- ----------------------------
101 | -- Table structure for `user_info`
102 | -- ----------------------------
103 | DROP TABLE IF EXISTS `user_info`;
104 | CREATE TABLE `user_info` (
105 | `id` bigint(22) NOT NULL,
106 | `username` varchar(64) DEFAULT '小可爱！' COMMENT '用户昵称',
107 | `open_id` varchar(128) DEFAULT NULL COMMENT '微信授权的openId',
108 | `avatar` varchar(128) DEFAULT '/src/assets/avatar.jpg' COMMENT '用户头像',
109 | `phone` varchar(11) DEFAULT NULL COMMENT '电话信息',
110 | `account` varchar(32) NOT NULL COMMENT '用户账号',
111 | `user_level` varchar(16) DEFAULT 'common_user' COMMENT '用户等级',
112 | `status` tinyint(4) NOT NULL DEFAULT '0' COMMENT '账号状态，0正常，1禁用',
113 | `password` varchar(64) NOT NULL COMMENT '登录密码',
114 | `create_time` datetime DEFAULT NULL COMMENT '创建时间',
115 | `update_time` datetime DEFAULT NULL COMMENT '更新时间',
116 | PRIMARY KEY (`id`),
117 | UNIQUE KEY `account_index` (`account`) USING HASH
118 | ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
119 |
120 | -- ----------------------------
121 | -- Records of user_info
122 | -- ----------------------------
123 | INSERT INTO `user_info` VALUES ('1664427904784019457', '小可爱！', null, '/src/assets/avatar.jpg', null, '18230675983', 'common_user', '0', 'acaaf694427dcb42423626d4aa74c4ae', '2023-06-02 08:25:11', null);
124 |
--------------------------------------------------------------------------------