From a2cf97b039e7b34399daaff8b9a6218a2316666d Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Mon, 10 Apr 2023 06:44:27 +0800
Subject: [PATCH] logout function is ready

---
 server/chat_handler.go          |   4 ++++
 server/server.go                |  40 +++++++++++++++++++++++++++++---
 web/public/images/user-info.jpg | Bin 0 -> 8583 bytes
 web/src/views/ChatPlus.vue      |  14 +++++++++--
 4 files changed, 53 insertions(+), 5 deletions(-)
 create mode 100644 web/public/images/user-info.jpg

diff --git a/server/chat_handler.go b/server/chat_handler.go
index 47547551..19929d41 100644
--- a/server/chat_handler.go
+++ b/server/chat_handler.go
@@ -42,6 +42,9 @@ func (s *Server) ChatHandle(c *gin.Context) {
 		c.Abort()
 		return
 	}
+	// 保存会话连接
+	s.ChatClients[sessionId] = client
+
 	// 加载历史消息，如果历史消息为空则发送打招呼消息
 	_, err = GetChatHistory(session.Username, roleKey)
 	if err != nil {
@@ -53,6 +56,7 @@ func (s *Server) ChatHandle(c *gin.Context) {
 			if err != nil {
 				logger.Error(err)
 				client.Close()
+				delete(s.ChatClients, sessionId)
 				return
 			}
 
diff --git a/server/server.go b/server/server.go
index 716d1230..b0809c2e 100644
--- a/server/server.go
+++ b/server/server.go
@@ -39,9 +39,10 @@ type Server struct {
 
 	// 保存 Websocket 会话 Username, 每个 Username 只能连接一次
 	// 防止第三方直接连接 socket 调用 OpenAI API
-	ChatSession      map[string]types.ChatSession
-	ApiKeyAccessStat map[string]int64 // 记录每个 API Key 的最后访问之间，保持在 15/min 之内
-	DebugMode        bool             // 是否开启调试模式
+	ChatSession      map[string]types.ChatSession //map[sessionId]User
+	ApiKeyAccessStat map[string]int64             // 记录每个 API Key 的最后访问之间，保持在 15/min 之内
+	ChatClients      map[string]*WsClient         // Websocket 连接集合
+	DebugMode        bool                         // 是否开启调试模式
 }
 
 func NewServer(configPath string) (*Server, error) {
@@ -65,6 +66,7 @@ func NewServer(configPath string) (*Server, error) {
 		ConfigPath:       configPath,
 		ChatContexts:     make(map[string]types.ChatContext, 16),
 		ChatSession:      make(map[string]types.ChatSession),
+		ChatClients:      make(map[string]*WsClient),
 		ApiKeyAccessStat: make(map[string]int64),
 	}, nil
 }
@@ -84,6 +86,7 @@ func (s *Server) Run(webRoot embed.FS, path string, debug bool) {
 	engine.POST("/test", s.TestHandle)
 	engine.GET("/api/session/get", s.GetSessionHandle)
 	engine.POST("/api/login", s.LoginHandle)
+	engine.POST("/api/logout", s.LogoutHandle)
 	engine.Any("/api/chat", s.ChatHandle)
 	engine.POST("api/chat/history", s.GetChatHistoryHandle)
 	engine.POST("api/chat/history/clear", s.ClearHistoryHandle)
@@ -307,3 +310,34 @@ func (s *Server) LoginHandle(c *gin.Context) {
 		SessionId string     `json:"session_id"`
 	}{User: *user, SessionId: sessionId}})
 }
+
+// LogoutHandle 注销
+func (s *Server) LogoutHandle(c *gin.Context) {
+	var data struct {
+		Opt string `json:"opt"`
+	}
+	err := json.NewDecoder(c.Request.Body).Decode(&data)
+	if err != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: types.ErrorMsg})
+		return
+	}
+
+	if data.Opt == "logout" {
+		sessionId := c.GetHeader(types.TokenName)
+		session := sessions.Default(c)
+		session.Delete(sessionId)
+		err := session.Save()
+		if err != nil {
+			logger.Error("Error for save session: ", err)
+		}
+		// 删除 websocket 会话列表
+		delete(s.ChatSession, sessionId)
+		// 关闭 socket 连接
+		if client, ok := s.ChatClients[sessionId]; ok {
+			client.Close()
+		}
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Success})
+	} else {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Hack attempt!"})
+	}
+}
diff --git a/web/public/images/user-info.jpg b/web/public/images/user-info.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..580598a1d137527fd7a1e6078480210ea030518d
GIT binary patch
literal 8583
zcmeHqXIN8N8|?u?ks?L9C`eHRfzXSA459&K00E^#L^?<>2@uE#ibxSbiXcU$*GLD2
zM5&^KNS9tBMQTC~5XcSA-1+WvpYO-L_wSu`p6v4`=iTqwYwfd7)~0-?%m5rWVfrwD
zh6Vuis0Toq1-{-uy1M{?p&=jz000xfK*I&lQ6n1a0ni8n^uJ;NFsC{DPi#*k`F9R2
zHAfzxwg3<H<MP-Y0R5ehqMlDcy~|0Zzs|SZk<RzM90Gm(MQ-`~h$zZ}W#y^YK+izG
zYH(BI>Q^-Xx&JHLf9g5CqWe!w%>xv`3W^$FB@KD72v|i!UR6U`mCEN-w_9MKuZEnQ
zcYv&elg~Y8Sw|l)Ii!QHoV+Yp4uC+Bz7CF_&VeHLoL$|$HAOex<3vT=ois%)lnud#
zzB<ls?s}pA&Ss%Tw;e-09o3ygp;{skqz2N<*ULH3K?Lb_-#b79sVVxaxCS*o`B_d>
z<X1|dr>3Z-;VlszAAe^N6)HQhD7Cl0lZ%Gwb=cp1soyk3{~ja)fsjQg%KG@b%E_y%
ztIL5E<P;QSs2MT=A>M%wNEz<{vHw}&x^sY|zq@asyN|cXNkxZyK0$$+qSQP8H6$<J
zf6D$>?k9=98iv%-IynS7Tz3d`))YO_PzJ0dBM-iPqM@3G0$3WXrU3^3m8|3Md@@(J
zeSGd~ivG8``(M(W|7*pQS^2x7990Rqzoj?{|It)S<X`-&z`qLotH8es{Hwsf3jF`4
zz<*qsvp3a>A*h}UpsWE-x7>UJeFEHkd{3Mrpmt5)5OiWM0pN-djWs}bRzg>ZYCUNI
zS|M63fa%1{fR*IILbOnT1*{^k`pbHN!N2-R{OZTyn^O9NzZFNjY~}tI$=Nha`2cV-
z0ZxDu9gQeJ%Sl7WNkeG|gsGO5p5_;Rb)z;kv~(bP21ce+%q-M|Iu3xAhK`OFL`P45
zG9;RC>U99bNzZjbUYp^Ji36jkAGgATq}NPh*Q!49nD!CG74P{+o?_<Z;};N=xOhqO
zvXqjtimIBrhR*dHFkL<Un`XD~m|Iv{Svxv8ySTc!djtdq1tX9lp-~SXML&*t5}W)y
zB{ePmMMmbEy!?W~qT-U$>YCcR`i91)cdc#h9i3gDx<B_13=R#Cd><W~ox}f_UszoF
zxlG*L+TPjSBkdoY@S<Alzi3h0znJ}Ryf~@6Xh9%45aS6i8d}5&aZV8Z1$hQ8Z4*WZ
zzcZo=516>GCB3ftcuGvsl)!V(zmJ($TxnK<ctY(Lv;Q+<k^fK3{vq}sUgLlcK>G_I
zT3QeT2n1qaVxWSFndt<~tjxc_`ZrGh0>=rs{sDzrggT*g^dNdB5Qymv8w=Z+|7%bt
zsXG)#${4^(M?-y>=r{o=Kvw%!KDD`hJrCl7Z6T<(PP*P6I3D-nl#03(yFWc!_ojD_
z-a_EKCA1hb2LIzgB8hOn;2D>0Q5|X|j%;Odv4~#VV>KMkGVj-+viJ?^q^V-3vX4)>
zTk;Vz+#~q&#+%4cX7VaYxKP)%+)iH*zxQ~5<h%8B0_5sU3r`)6d%CS?45sDo>KnJ8
zQ#6K)uOTngDi+1HwPl+2e?dfa4Z0xVlh<+`E#`D2)*eI^V}u$SY6nF4((hMYFE6!f
z1ge;!3cq1w260~OqeWIN$K4G?uUF62zC7VMMFCKi)5DGM7+kHXTWDdf>x`EsMAyrY
zrMZA}KIYs9F@{8uZ8c5xIqdU?%^F|TqEw2adfbiKm@7?F7Ww$|pND<^y#85gBRJ%H
z`Sjscxzyjq!}Evxj4cpb;{h8zvd1=h*FF3?%oTN&2lUDXQfA-!Yp?VmE8~cwR#-ty
zT%8-X2bGpQp<Kn*?DZ9#?<4l$t8~{f==d7~$`MpmG~w>)l3Z4s&TRQOiJSdJRy=X1
zl9l}tvfPDHt>X(H_gk3)UJ4L9kUAl|>!E6Hw1zS3!DQYpvSH(Q$RC0~p#T@hj}CGz
zNUieWd74#AlO7F^%JDV*!+;uX*|K#(#ZXzTf^?~nyeShLQ()KcYUZA*4zgGX@*OI6
z7?k>$KYhDzn+R{3H!dM%QviCV0f#2yH>X>tjOyl~H9l=@(iTMe_sdK|T&H@%<qi>J
z84&yG2f4}3XJ$rX`P%q)_hfw)qz@gV<<*)95>osETHfZh6T~WR_wt^br31#Fj=Tqs
z`AINoNxY5TTW&N{8IDkalAr)uN$6?OW{)rU&S&+-0-b*5xLMTF^fV879P(0Q0TE9D
z&a88njO&Y0zLBFza6a;F0}5c9z8`y3LE>m6pmL3W!sDk|MWBRdm?J;<Pk3!E?VGNL
z;^(LPagqnx(8+;_$4j<+wNE!@kBb~;%h=4?t;_a<hYiFx8{WWJ<jIdBl#ahcYgJ>k
zl#xRh*ktNcy#!x)@gGgLLBWjlRf8kTKJmvQB=jt<&CY+%nnik|OI*Iy7a^eio2Bq9
zXlvhMh5{t=8k6)3GY)Q6Rp*+#W&ai_-Ag$0SaMF$dcRXDb)Lsff32U44c*uaQp7`k
ze{E<QwWE1yW~B8j1&FbZcNcOo{6T+9F^dA^kniEN<!~;z7$_S`DaeOy(%`;{zUUMT
zX8SZ^VXB2l0Y)kvcLj;()9*PXFEFX<q+HRS@K?LE(Df5uh7~1xrB&s%WVbXMH~B@L
zRzDCtUN7NySnB_1z$TcofRS~|UPpt8j69BSk;k@>{oA3QB_!c6;#oX>^lNd1mgJz!
z)X<*Jg7MYrda$p*U)KT~0!rW8B`h>)r|n!inTieH5=n~G!zX;DIh$CC-02C|CA#i<
zx>ee=tAfkrCl!)ksSJsq&JaKHTw)(R{5a;?rY6P*)>ZPqi0Xaenv1E2GB4PQCC@rS
zyaeS-_y}ETDu$k3X}xw+@AAg}Y`&iIHWq1hxM8q@gEx|oNNWVfIaoqI+S^m`(ZqY0
zwxS|TE9NwJxV*j2q5AO<1>lUJLt<GBVY^{R4Eom*X=duarA90rjoFng$eC?X+&#2g
zt`O1%|ItDv`l~w4HTkCWIy6K}$<rNW5+;!u;-{2fx){t@7&h{xqggS1F+=z5pT2(a
zmynI467>lZHFkodIq4uGuPRr8dGm&3%h2la*3R@UF0HLSdDQd3^GO9pX4ow&(8;gK
z_=o3Qi}dd~enK3CaL=Xm$UPGh<4ha+x!m`g<jLU27tp_3=o6-A5o48Jiz%)lbc|ED
zG$iA>2cLXRT3rHKr0z)YO?E1{j}M9MWp%Nb@tG^Wsq=%!LQ)}@=V%1iT*QDVV+!!l
zrro{*p2sJAYVEb;lRs3R4l$CYK853?_7>-gKsSwUz0_aJqX1&1>TI6UxOPe(xt?U_
zlgoLDx;z%7cYS71YJ{vP@;0Um_b3V7p_jYUfi8&ng5rqN;`M_khF`ekHemmFsade4
zTXFO1ZVqv(p%~>bgyOBK{$Bp<dS6uAeVXWq_*KvD+Ld$cohpW}mN!FG)8e~@%D%7)
z>+GfMh>1jf+Q_v#dfQA>iM)!>I+n`h^*m?e)y{QV+v$#l*^&<Xr`Y{_0|rChF3+u4
z@5!!Cu<k83ZgfO+ecZpV2+5T_?#R6|q*jYBTQIq4VHfOSv5NZYY#rzqG<;Fj$mr)T
zHkZI2Ot3w#L9&SOF2}2;HTAyqN}mbXx^!B8S$7d0K)!lZj=JE<ZMKp=<<;sh!qk>@
zH1~mq<bs2c1@JvLyKdP*&C^j1`mAA&*wB$|<Vw4F=15$nvYm1;%(uy2m4{!3H1g9{
zz0)eQ0+OiyDl-zN(dXl-v}3qZvxGu$5~6Jq3xsYv4R{atQvkH%4q^_m_R@LI?#j<7
z_jrG>wDd=KJ$vL!{o^*GP~p&XE!z$QNv(JWBX-+Ae8Vh$d>A_KFrPjBf&wsCosTPl
z_ttl_zQ?`q3C{{>^bfaB<$hCVzA>?n^yQ`R-v!4_FN7Bm(fA))l1a5aw2mm-XK}Px
zcAONn<iJ9cpt^g5^oSVN?&l$RRP)`vpgp-J{;I71Z|7<@?2Mxp`j5}J1q(W3vxUZv
zey~6aohg86yaJ1BR*tV4zkp`VE@E{L-z$iW>I(Hv_X<1L(;VM}n6I)+dZP&Nc|;bx
z)q;28`0!6v<{)^<&V`U0YJF|D9*&jC*vuPS-f(Nmto9E#Gq@drdFESam+J4ZY>??a
zwLyJPhwL=m#=2m-a!<QY@znRD6blN!gz;b3V|+7Xp4gT#r(0{hJ6v>B>^xrnoAGOm
z1PMK-cvR==!ERxFY~1a?yq|u$vtWJgvg%%a#7n8ZqL`a@rqJu_83rZLNgkWaCi)i%
z3AwX8yc?n9OV76V1t|UG8Ilbz<nL$t6Y9+GNautIZ8G@rv6K=8sFXgEUQJBQ0TVAh
zz`01j#IWAn-Ljw5@z{dtIVEiNL~3}CH!OK>ZV33tW_)#s{Or~ZyW`AXxNEVpsdX4#
z#lrTmEGGF~!st$KJ3AXnnphb#=Gq=yzj;AoSl`)lOcXg$uGh7xxw0jPw|IbdhO_@{
z?QNY>lnFRfwV9+fTm-=f+WaO$0ct}>7GJJtbo6}7UQE_E7M7RE+7;QUm?w-pse56H
z#P2iuf1q2Z4~H@#?c!&J?m%F|YQ@RJeJ1fieTUVHMvECQh~o0ZWMt6Rd-9!FtcXUy
z#A@Gs<~+?DOq<_fdcGT4fe_3-&M0>qjyO;7iGE$fDXC~@d?9lo<ZAM?l<0v}(YfwQ
zDGH!7JU8ZT6`}oy0=o5GFN1;Rs_FOf?3y)a`|sXE2NCX81Ci|kTaE80>r<z##QV|8
z&foSASjLQ#hjBXv>(<|}T%@tN8%>sTPgGCa-!M5CSx#?LB<Si9R(S;*9?eD`#rqov
zrpGmZE~N>&hmeG1=f&=SnGudwZh;=E2y<fhwWiJ~bi@`8JZN+}$dTR%8)vzem6W(w
zfYCYChF;0Jl_qyqvMa|t&?a`zF0Iree07`L%dS3`xPuT7<Sq_S<(^g?fhUbD=hmox
z#KcVT`|!GP6o(DUYfhLc^)qS~NW2seUMvcoAc+y^+JqS#)4a6I-@Qvk;p9723K1N6
zmrB$xL5TJjX0BrRe!h!tDI*2{ka?Q4IAz!V^}BBx`aQq59&5{0bJvd>ad3`iXRRmr
z?z&GK+Xlo-*$f>|X6^<n`Fq_Gc(#?46)(g(vf0J=1EQU-qd3bd8;)WXZ^D~=UN=w4
zV;FRH!Gl(RyxHD{8g_5-JGy`CG<v*ov9e-xV-Ft2dQ|3gxdhf$ow<*j`Q9`H3Gfmo
z6!q*|^ug=65{$Tq0~f8<(to(>WS56BoP+uL<<xB!wQihgo5gW=1zr9y9=nUuY@_}k
znw4FF$eAU(%V`>0$d~YBn3$y|&o>>smuW4&(R+i3Z!CQn>x3l@+^znjn*H6?At=c%
z7{2w*V#eS>jSFTcB)n@h+>xCEK>4Q7lmU|JY&<`a%Z~zFu{f|;Dr_kx1sIa}igo7<
z^nCIfX(cNbZ++U#x%CYF4z|&X<sw{+3ok@QinqjcT1$v&cJXX?b;+?I3-@M{NoB2W
zduDEH3!23OY-jI6UI@(ty#vbo=s51nb6Y6v^ODa6lA}{&+am>&sRzQZd=FBE&Uw2n
zV}g9Xk-rQ<SR4wcGp-Cqu0433MK^eSXPRWh@BCiSLb&Cu+0pE`2u-a<4C^3Pj1)WP
z)hnbIvc!(=e4E=jCF;V)J!JyoxOF#DLuh2R(e@=cOKhj);$cGop2(lcY#Psl)g27o
zh_t_bh59m$?s5&9WfyF15Xnb?IE;-NkAwQgkNa|zpZ0le$gHks!&s|fA74(^E0jPV
zg`i`boy$qRW`fwtrE1t~N4*gb(WorRgRNwJt-9q~-JV7~;GMDgWKZ%_i6<s09=d_O
z1Jl4V$9Q+4{pUkYZ^wxrGm;mg_1A{674L@wN8&;!2)cG_@T_SD2<qL)qfB+Cty#F#
z9TARzX~rs{ks2H0{QGCq<Vnn=93G;A(Anr5ofi%gm(8-W@#3|it0NN|rJ7n|gngn+
zXh`8%jyGARH5TK2x!R<F_TW(X1MEeQ&6|*@qOWBa#3OGs*`He3^4r0rqi0^i&t4(R
z;XQjFV#PB|2W;DiVLF=?{jA&qGXiE?QjDQ9I5*>1Og{N4JX%o9EB%h2#Mh}eAoV-Y
z4Y2SZdzcgy4}t2-2QMbxlGlsPR|@k#EA|@XFK>BQ4JPCs`L!w|UNa@&REcg`WTeT%
zcXpGi{VUx$kM1jNE;riZnD!_6^QRFb1ZYujEEd#HM$BA2;??W3ce66tJsMsvN`wK&
zJS6%GiQ)7TCF4}&i!%0Wr`f`a9`0Zgl70$14JxnwZuw>=KV9Li2%24{zxmys{voWe
z>XVvG50sg#*zSAdiV>s0(*&e`i3InEZ06JkqA<?Wr$k^a#31?epYvZEilO4q^^a=M
z8}Q$)k(cy5ExY4g-PX*}vy6QdVDSaHY<<cvt;kz<Pe__BK6Fc%GPwVx?On>0$cR@V
z^5H$T$eDnja9iYKyla#eh^%D@@5ukOC!i<u{OtT)#rQLOp|sRRn~INMCk2&!hj)o5
zaR77$5vYg}E%kRq;||XRF7>$4rDd79-;BO}KrQkME#jcI7M(;Wg99l~we7C>8jK~=
zJajbBxM2LSGJFy;RnLCPjK{`0IoRU(y{KGFE>{FV>auDa-lhOdw%-kWJ4WwfN6+I2
zAA;WsZ1eMMrQ$8^ruD{bWkBM|F+DAR?A>QQzE-`@Q#VVB=qtOyxVH(_fn(<P?I6{O
zHLunEJKTHJ4Q?D??2T=Al!(m?60JNG$BC||&%V#TsVykvg2j*!hobf<7-4e#b%BT0
zoukrqODB_gY`}x&^QXG}IzqlixeW(AykBA#N!@_2dTu}7bsX}7)|nZnAwM+1<s8zp
zyOJJ_DHuurKz?~bM6z-d`2|ZA60)~LF9nw$R$SfHB*#g&jg)miNf`N)yh}uUfX6}U
z1Kr66#YXii?J=Wyjq!@lztOWLM`S3^$s-RtcsCjCuC0jdrci(fm~X};$*Bgd-0~$X
zYklg&YpRV&48L75;?4meVYzL%g>#`3=bom8r<?Wo4bDs&)DH4?kJ|68?FohaaQLGg
zBFpSQpO|ZRV0oZ(%*PiMn0xvi2>}ZUI*h~y=M`5^JY$5*SULoZixBOL2s-$NxVp1n
zvZc+^E(-w33ilo%zw-OP_<*Oy3Sn-L%urWQAw|;lgH6v@H?kbL$a(6Fc^sj^7z5SM
zmzyPQJXWVF-rKxct*tiDVLk6-W&UU~ZOnZud^+RcFdx|g(RWko@fcI6W@R+v;>^L{
zOB{(M;&_qPhU_--Z$Fng+YA|t48xw6xE<Ut+l|SW@QtwS7K-2H*-p`r$W3h4^a?7J
z+)7fcaffYoe~{1vZFNMT+cG90xHCe^R@T{uSEbjnTT5HN3lVQm>PdgTlxnjA?xMXI
z%KY-sTiAMaeD-|(X^{9acP^nsxp#jKEp>=^{3Qq57rsPns=|Hhr3@YFw7>I;PMs}V
zgMF3Y`sgDbqxWf|R1Y&B)y|hL7~K?t4n`O#ZC=fyNw!S(1=FMN<fu}BB)K~j0HcrF
zrh|VrFQEVzSP~DGrvDu2Myi_Lk-|HhOdh^mA$NWEd)}YCbHp6v@_EkFpmE)6+%u1s
z{WaO6P7Asdx5IN-6C1JeofNn_pqp<^wjfp{jr+(njt{7${fV|r2sb;FK{*kp>WE?b
z{mSkY3w0rMO4s@)>3Ej(pDKH6#B;(KbFmS1)P=K}NC70?>r#LStnd>BTTnImcERIE
zgBOqax8pW7CopI3%!y=8t(0k|A2aKCza<C6-f+3Pnm3d6ATl)6VzDJ62K`YV_oHE)
zx=&9=E5fsPBF>iiuG3C`idUondt+GKtGwstt2C;F(&VDh%_301;Q7wd({%{y#_n0X
z@J=841*Xa4XU=tmR%`#u_J)Tw$j6r2jR_@Tl61le5wXWzNmJToh7QGWyJdsmte2xd
zvI;-wjW*5(Oh5J9>TR23KkR|O$*tBU$0t#M#_5Aom>#uW`fE?`m={QS$bPw<99CI0
s=CtX<Sq9ZP9ak71v1E#NbjNLdGOt=Q#C*V3ne9(cg(DQ#(Uj5u0oFAr3jhEB

literal 0
HcmV?d00001

diff --git a/web/src/views/ChatPlus.vue b/web/src/views/ChatPlus.vue
index 5c63e902..55feb2b6 100644
--- a/web/src/views/ChatPlus.vue
+++ b/web/src/views/ChatPlus.vue
@@ -14,7 +14,7 @@
 
               <el-dropdown :hide-on-click="true" class="user-info" trigger="click">
                 <span class="el-dropdown-link">
-                  <el-image src="images/avatar/user.png"/>
+                  <el-image src="images/user-info.jpg"/>
                   <el-icon><ArrowDown/></el-icon>
                 </span>
                 <template #dropdown>
@@ -33,7 +33,7 @@
                       <span>删除记录</span>
                     </el-dropdown-item>
 
-                    <el-dropdown-item>
+                    <el-dropdown-item @click="logout">
                       <el-icon>
                         <Monitor/>
                       </el-icon>
@@ -431,6 +431,7 @@ export default defineComponent({
       }).then((res) => {
         setLoginUser(res.data)
         this.connect();
+        this.token = '';
       }).catch(() => {
         ElMessage.error("口令错误");
         this.token = '';
@@ -485,6 +486,15 @@ export default defineComponent({
       }
       this.chatRoles = roles;
     },
+
+    // 退出登录
+    logout: function () {
+      httpPost("/api/logout", {opt: "logout"}).then(() => {
+        this.checkSession();
+      }).catch(() => {
+        ElMessage.error("注销失败");
+      })
+    }
   },
 
 })