refactor: 重构代码架构并使用 Modern Go 特性

- 重命名模块 jwts -> schedule_converter
- 新增 service 包,统一业务逻辑层
- 新增 pkg/logger 包,使用 slog 结构化日志
- 新增 pkg/errors 包,统一错误处理
- 拆分 captcha 包,图片识别移至 service/recognizer
- 清理 models 包,只保留数据结构定义
- 消除 grpc/handler.go 中的重复代码
- 简化 main.go,使用 slog 和 context
- 配置支持单例模式和结构化配置
- 使用 any 替代 interface{}
- 使用 sync.Map 替代 map + RWMutex
This commit is contained in:
lafay
2026-03-16 13:40:53 +08:00
parent 294151cd40
commit 74438c1f7f
20 changed files with 1183 additions and 602 deletions

View File

@@ -1,36 +1,40 @@
package client
import (
"fmt"
"log"
"log/slog"
"net/http"
"net/http/cookiejar"
"net/url"
"strings"
"time"
"sync"
"jwts/config"
"schedule_converter/config"
)
var Client *http.Client
var RecognizerClient *http.Client
var (
Client *http.Client
RecognizerClient *http.Client
once sync.Once
)
func InitClient() {
jar, err := cookiejar.New(nil)
if err != nil {
log.Fatalf("Failed to create cookie jar: %v", err)
}
Client = &http.Client{
Jar: jar,
}
once.Do(func() {
jar, err := cookiejar.New(nil)
if err != nil {
slog.Error("创建 cookie jar 失败", "error", err)
panic(err)
}
Client = &http.Client{Jar: jar}
RecognizerClient = &http.Client{
Timeout: 120 * time.Second,
}
cfg := config.Get()
RecognizerClient = &http.Client{
Timeout: cfg.Recognizer.Timeout,
}
slog.Info("HTTP 客户端初始化完成")
})
}
// 设置请求头
func GetHeaders() http.Header {
return http.Header{
"User-Agent": []string{"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"},
@@ -39,56 +43,58 @@ func GetHeaders() http.Header {
}
}
// 获取带Content-Type的请求头
func GetHeadersWithContentType(contentType string) http.Header {
headers := GetHeaders()
headers["Content-Type"] = []string{contentType}
headers.Set("Content-Type", contentType)
return headers
}
// 步骤1: 获取初始Cookie
func GetInitialCookies() error {
fmt.Println("[1] 访问登录页面获取初始Cookie...")
return GetInitialCookiesWithClient(Client)
}
req, err := http.NewRequest("GET", config.BaseURL+"/loginNOCAS", nil)
func GetInitialCookiesWithClient(httpClient *http.Client) error {
slog.Info("访问登录页面获取初始Cookie")
cfg := config.Get()
req, err := http.NewRequest("GET", cfg.BaseURL+"/loginNOCAS", nil)
if err != nil {
return err
}
req.Header = GetHeaders()
resp, err := Client.Do(req)
resp, err := httpClient.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
fmt.Printf(" 登录页面状态码: %d\n", resp.StatusCode)
PrintCookies()
slog.Info("登录页面访问完成", "status_code", resp.StatusCode)
logCookies(httpClient)
return nil
}
// 打印Cookies
func PrintCookies() {
fmt.Printf(" 当前Cookies: ")
for _, cookie := range Client.Jar.Cookies(parseURL(config.BaseURL)) {
fmt.Printf("%s=%s; ", cookie.Name, cookie.Value)
func logCookies(httpClient *http.Client) {
cfg := config.Get()
cookies := httpClient.Jar.Cookies(parseURL(cfg.BaseURL))
var cookieStrs []string
for _, cookie := range cookies {
cookieStrs = append(cookieStrs, cookie.Name+"="+cookie.Value)
}
fmt.Println()
slog.Debug("当前Cookies", "cookies", strings.Join(cookieStrs, "; "))
}
// 解析URL
func parseURL(rawURL string) *url.URL {
u, _ := url.Parse(rawURL)
return u
}
// 检查登录是否成功
func CheckLoginSuccess(body string) bool {
return strings.Contains(body, "本科教学管理服务系统")
}
// 检查是否验证码错误
func IsCaptchaError(body string) bool {
return strings.Contains(body, "验证码")
}