From 31b131f1db2480354fd18c5afa9d265b5df71191 Mon Sep 17 00:00:00 2001 From: fuzhongyun <15339891972@163.com> Date: Sun, 4 Jan 2026 10:52:29 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E8=8E=B7=E5=8F=96?= =?UTF-8?q?=E7=BB=9F=E4=B8=80=E7=99=BB=E5=BD=95token=20=E7=9A=84=E6=96=B9?= =?UTF-8?q?=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/config_env.yaml | 7 ++ internal/config/config.go | 9 +++ internal/data/constants/lsxd.go | 8 +++ internal/pkg/lsxd/login.go | 120 ++++++++++++++++++++++++++++++++ internal/pkg/lsxd/login_test.go | 33 +++++++++ 5 files changed, 177 insertions(+) create mode 100644 internal/data/constants/lsxd.go create mode 100644 internal/pkg/lsxd/login.go create mode 100644 internal/pkg/lsxd/login_test.go diff --git a/config/config_env.yaml b/config/config_env.yaml index 4348d2d..c56d1d3 100644 --- a/config/config_env.yaml +++ b/config/config_env.yaml @@ -23,6 +23,13 @@ coze: base_url: "https://api.coze.cn" api_secret: "sat_AqvFcdNgesP8megy1ItTscWFXRcsHRzmM4NJ1KNavfcdT0EPwYuCPkDqGhItpx13" +lsxd: + # 统一登录 + login_url: "http://api.test.user.1688sup.com/v1/login/phone" + phone: "0zmINhJBwsDJYTmeaxXK4A==" + password: "LSSGoWhc63NpZc1rq3LT8g==" + check_token_url: "http://api.test.user.1688sup.com/v1/user/welcome" + sys: session_len: 6 diff --git a/internal/config/config.go b/internal/config/config.go index 132fb23..3955b44 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -14,6 +14,7 @@ type Config struct { Ollama OllamaConfig `mapstructure:"ollama"` Vllm VllmConfig `mapstructure:"vllm"` Coze CozeConfig `mapstructure:"coze"` + LSXD LSXDConfig `mapstructure:"lsxd"` Sys SysConfig `mapstructure:"sys"` Tools ToolsConfig `mapstructure:"tools"` EinoTools EinoToolsConfig `mapstructure:"eino_tools"` @@ -116,6 +117,14 @@ type CozeConfig struct { ApiSecret string `mapstructure:"api_secret"` } +// LSXDConfig 统一登录配置 +type LSXDConfig struct { + LoginURL string `mapstructure:"login_url"` + Phone string `mapstructure:"phone"` + Password string `mapstructure:"password"` + CheckTokenURL string `mapstructure:"check_token_url"` +} + type Redis struct { Host string `mapstructure:"host"` Type string `mapstructure:"type"` diff --git a/internal/data/constants/lsxd.go b/internal/data/constants/lsxd.go new file mode 100644 index 0000000..2cb92e6 --- /dev/null +++ b/internal/data/constants/lsxd.go @@ -0,0 +1,8 @@ +package constants + +import "time" + +const ( + CACHE_KEY_LSXD_TOKEN = "ai_scheduler:lsxd_token" + EXPIRE_LSXD_TOKEN = time.Hour * 2 // 2小时 +) diff --git a/internal/pkg/lsxd/login.go b/internal/pkg/lsxd/login.go new file mode 100644 index 0000000..e2c5509 --- /dev/null +++ b/internal/pkg/lsxd/login.go @@ -0,0 +1,120 @@ +package lsxd + +import ( + "ai_scheduler/internal/config" + "ai_scheduler/internal/data/constants" + "ai_scheduler/internal/pkg/l_request" + "ai_scheduler/utils" + "context" + "encoding/json" + + "github.com/go-kratos/kratos/v2/log" + "github.com/redis/go-redis/v9" +) + +type Login struct { + config *config.Config + redisCli *redis.Client +} + +func NewLogin(config *config.Config, rdb *utils.Rdb) *Login { + return &Login{ + config: config, + redisCli: rdb.Rdb, + } +} + +func (l *Login) GetToken() string { + ctx := context.Background() + // 1.取缓存 + token, err := l.redisCli.Get(ctx, constants.CACHE_KEY_LSXD_TOKEN).Result() + if err != nil { + log.Errorf("lsxd get token from redis failed, err: %v", err) + } + + // 2.缓存存在 + if token != "" { + // 3.缓存有效直接输出 + if l.checkTokenValid(token) { + return token + } + } + + // 4.缓存不存在或缓存无效，调用登录接口获取token + token, err = l.login() + if err != nil { + log.Errorf("lsxd login failed, err: %v", err) + return "" + } + // 5.缓存token + l.redisCli.Set(ctx, constants.CACHE_KEY_LSXD_TOKEN, token, constants.EXPIRE_LSXD_TOKEN) + + return token +} + +// 校验token是否有效 +func (l *Login) checkTokenValid(token string) bool { + // 欢迎页校验token有效 + checkTokenURL := l.config.LSXD.CheckTokenURL + // 调用欢迎页校验token有效 + r := l_request.Request{ + Url: checkTokenURL, + Headers: map[string]string{ + "Content-Type": "application/json", + "Authorization": token, + }, + Method: "GET", + } + res, err := r.Send() + if err != nil { + log.Errorf("lsxd check token valid failed, err: %v", err) + return false + } + if res.StatusCode != 200 { + log.Errorf("lsxd check token valid failed, status code: %d", res.StatusCode) + return false + } + + log.Info("lsxd check token valid success") + + return true +} + +// 调用登录接口获取token +func (l *Login) login() (string, error) { + // 1.获取配置 + loginURL := l.config.LSXD.LoginURL + phone := l.config.LSXD.Phone + password := l.config.LSXD.Password + + // 2.调用登录接口获取token + r := l_request.Request{ + Url: loginURL, + Headers: map[string]string{ + "Content-Type": "application/json", + }, + Method: "POST", + Json: map[string]any{ + "phone": phone, + "password": password, + "code": "123456", + }, + } + res, err := r.Send() + if err != nil { + return "", err + } + + // 3.解析token + var resp struct { + Token string `json:"accessToken"` + } + err = json.Unmarshal(res.Content, &resp) + if err != nil { + return "", err + } + token := resp.Token + + // 4.返回token + return token, nil +} diff --git a/internal/pkg/lsxd/login_test.go b/internal/pkg/lsxd/login_test.go new file mode 100644 index 0000000..6b15133 --- /dev/null +++ b/internal/pkg/lsxd/login_test.go @@ -0,0 +1,33 @@ +package lsxd + +import ( + "ai_scheduler/internal/config" + "ai_scheduler/utils" + "testing" +) + +func TestGetToken(t *testing.T) { + + rdb := utils.NewRdb(&config.Config{ + Redis: config.Redis{ + Host: "47.97.27.195:6379", + Pass: "lansexiongdi@666", + Key: "report-api-test", + }, + }) + + login := NewLogin(&config.Config{ + LSXD: config.LSXDConfig{ + LoginURL: "http://api.test.user.1688sup.com/v1/login/phone", + Phone: "0zmINhJBwsDJYTmeaxXK4A==", + Password: "LSSGoWhc63NpZc1rq3LT8g==", + CheckTokenURL: "http://api.test.user.1688sup.com/v1/user/welcome", + }, + }, rdb) + token := login.GetToken() + if token == "" { + t.Errorf("token is empty") + } + + t.Logf("token: %s", token) +}