将端口号 pc 开头的账号显示为本地版

internal/logic/Wx/get_wx_list_logic.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	reqv3 "github.com/imroc/req/v3"
 	"github.com/suyuan32/simple-admin-core/rpc/types/core"
+	"strings"
 	"wechat-api/ent"
 	"wechat-api/ent/predicate"
 	"wechat-api/ent/usagetotal"
@@ -105,7 +106,12 @@ func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, err
 	for _, v := range data.List {
 		// 创建 hookClient 客户端
 		serverInfo := serverSet[v.ServerID]
-
+		serverName := serverInfo.Name
+		l.Infof("----------------strings.ToUpper(v.Port)--------------: %+v", strings.ToUpper(v.Port))
+		l.Infof("----------------HasPrefix--------------: %+v", strings.HasPrefix(strings.ToUpper(v.Port), "PC-"))
+		if strings.HasPrefix(strings.ToUpper(v.Port), "PC-") {
+			serverName = "本地版"
+		}
 		var loginStatus uint8 = 0
 		hookClient := hook.NewHook(serverInfo.PrivateIP, serverInfo.AdminPort, v.Port)
 		if v.ServerID > 0 {
@@ -205,6 +211,7 @@ func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, err
 				Examples:   &v.Edges.Agent.Examples,
 			}
 		}
+
 		resp.Data.Data = append(resp.Data.Data,
 			types.WxInfo{
 				BaseIDInfo: types.BaseIDInfo{
@@ -214,7 +221,7 @@ func (l *GetWxListLogic) GetWxList(req *types.WxListReq) (*types.WxListResp, err
 				},
 				Status:           &loginStatus,
 				ServerId:         &v.ServerID,
-				ServerName:       &serverInfo.Name,
+				ServerName:       &serverName,
 				Port:             &v.Port,
 				ProcessId:        &processID,
 				Callback:         &v.Callback,