作者 crossoverJie

:sparkles: Introducing new features.ai model

@@ -141,6 +141,6 @@ public class AsyncMsgLogger implements MsgLogger { @@ -141,6 +141,6 @@ public class AsyncMsgLogger implements MsgLogger {
141 LOGGER.info("IOException", e); 141 LOGGER.info("IOException", e);
142 } 142 }
143 143
144 - return sb.toString().replace(key,"\033[31;4m" + key+"\033[0m"); 144 + return sb.toString().replace(key, "\033[31;4m" + key + "\033[0m");
145 } 145 }
146 } 146 }
@@ -45,8 +45,22 @@ public class MsgHandler implements MsgHandle { @@ -45,8 +45,22 @@ public class MsgHandler implements MsgHandle {
45 @Autowired 45 @Autowired
46 private MsgLogger msgLogger ; 46 private MsgLogger msgLogger ;
47 47
  48 + private boolean aiModel = false ;
  49 +
48 @Override 50 @Override
49 public void sendMsg(String msg) { 51 public void sendMsg(String msg) {
  52 + if (aiModel){
  53 + aiChat(msg);
  54 + }else {
  55 + normalChat(msg);
  56 + }
  57 + }
  58 +
  59 + /**
  60 + * 正常聊天
  61 + * @param msg
  62 + */
  63 + private void normalChat(String msg) {
50 String[] totalMsg = msg.split(";;"); 64 String[] totalMsg = msg.split(";;");
51 if (totalMsg.length > 1) { 65 if (totalMsg.length > 1) {
52 //私聊 66 //私聊
@@ -71,6 +85,19 @@ public class MsgHandler implements MsgHandle { @@ -71,6 +85,19 @@ public class MsgHandler implements MsgHandle {
71 } 85 }
72 } 86 }
73 87
  88 + /**
  89 + * AI model
  90 + * @param msg
  91 + */
  92 + private void aiChat(String msg) {
  93 + msg = msg.replace("吗","") ;
  94 + msg = msg.replace("嘛","") ;
  95 + msg = msg.replace("?","!");
  96 + msg = msg.replace("?","!");
  97 + msg = msg.replace("你","我");
  98 + System.out.println("AI:\033[31;4m" + msg + "\033[0m");
  99 + }
  100 +
74 @Override 101 @Override
75 public void groupChat(GroupReqVO groupReqVO) throws Exception { 102 public void groupChat(GroupReqVO groupReqVO) throws Exception {
76 routeRequest.sendGroupMsg(groupReqVO); 103 routeRequest.sendGroupMsg(groupReqVO);
@@ -109,9 +136,16 @@ public class MsgHandler implements MsgHandle { @@ -109,9 +136,16 @@ public class MsgHandler implements MsgHandle {
109 printOnlineUsers(); 136 printOnlineUsers();
110 137
111 } else if (msg.startsWith(SystemCommandEnumType.QUERY.getCommandType().trim() + " ")){ 138 } else if (msg.startsWith(SystemCommandEnumType.QUERY.getCommandType().trim() + " ")){
112 - String[] split = msg.split(" ") ;  
113 - String res = msgLogger.query(split[1]);  
114 - System.out.println(res); 139 + //查询聊天记录
  140 + queryChatHistory(msg);
  141 + }else if (SystemCommandEnumType.AI.getCommandType().trim().equals(msg.toLowerCase())){
  142 + //开启 AI 模式
  143 + aiModel = true ;
  144 + System.out.println("\033[31;4m" + "Hello,我是估值两亿的 AI 机器人!" + "\033[0m");
  145 + }else if (SystemCommandEnumType.QAI.getCommandType().trim().equals(msg.toLowerCase())){
  146 + //关闭 AI 模式
  147 + aiModel = false ;
  148 + System.out.println("\033[31;4m" + "。゚(゚´ω`゚)゚。 AI 下线了!" + "\033[0m");
115 }else { 149 }else {
116 printAllCommand(allStatusCode); 150 printAllCommand(allStatusCode);
117 } 151 }
@@ -126,6 +160,16 @@ public class MsgHandler implements MsgHandle { @@ -126,6 +160,16 @@ public class MsgHandler implements MsgHandle {
126 } 160 }
127 161
128 /** 162 /**
  163 + * 查询聊天记录
  164 + * @param msg
  165 + */
  166 + private void queryChatHistory(String msg) {
  167 + String[] split = msg.split(" ") ;
  168 + String res = msgLogger.query(split[1]);
  169 + System.out.println(res);
  170 + }
  171 +
  172 + /**
129 * 打印在线用户 173 * 打印在线用户
130 */ 174 */
131 private void printOnlineUsers() { 175 private void printOnlineUsers() {
@@ -17,7 +17,9 @@ public enum SystemCommandEnumType { @@ -17,7 +17,9 @@ public enum SystemCommandEnumType {
17 ALL(":all ","获取所有命令"), 17 ALL(":all ","获取所有命令"),
18 ONLINE_USER(":olu ","获取所有在线用户"), 18 ONLINE_USER(":olu ","获取所有在线用户"),
19 QUIT(":q! ","退出程序"), 19 QUIT(":q! ","退出程序"),
20 - QUERY(":q ","查询聊天记录") 20 + QUERY(":q ","【:q 关键字】查询聊天记录"),
  21 + AI(":ai ","开启 AI 模式"),
  22 + QAI(":qai ","关闭 AI 模式")
21 23
22 ; 24 ;
23 25