调整目录结构

internal/domain/token/token_service.go

@@ -0,0 +1,68 @@
+package token
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/golang-jwt/jwt/v5"
+)
+
+// Claims 定义了 JWT 的声明结构
+type Claims struct {
+	UserID uint `json:"user_id"`
+	jwt.RegisteredClaims
+}
+
+// TokenService 定义了 token 操作的接口
+type TokenService interface {
+	GenerateToken(userID uint) (string, error)
+	ParseToken(tokenString string) (*Claims, error)
+}
+
+// tokenService 是 TokenService 接口的实现
+type tokenService struct {
+	secret []byte
+}
+
+// NewTokenService 创建并返回一个新的 TokenService 实例
+func NewTokenService(secret []byte) TokenService {
+	return &tokenService{secret: secret}
+}
+
+// GenerateToken 生成一个新的 JWT token
+func (s *tokenService) GenerateToken(userID uint) (string, error) {
+	nowTime := time.Now()
+	expireTime := nowTime.Add(24 * time.Hour) // Token 有效期为 24 小时
+
+	claims := Claims{
+		UserID: userID,
+		RegisteredClaims: jwt.RegisteredClaims{
+			ExpiresAt: jwt.NewNumericDate(expireTime),
+			Issuer:    "pig-farm-controller",
+		},
+	}
+
+	tokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	token, err := tokenClaims.SignedString(s.secret)
+	return token, err
+}
+
+// ParseToken 解析并验证 JWT token
+func (s *tokenService) ParseToken(tokenString string) (*Claims, error) {
+	token, err := jwt.ParseWithClaims(tokenString, &Claims{}, func(token *jwt.Token) (interface{}, error) {
+		return s.secret, nil
+	})
+
+	// 优先检查解析过程中是否发生错误
+	if err != nil {
+		return nil, err
+	}
+
+	// 只有当 token 对象有效时，才尝试获取 Claims 并验证
+	if claims, ok := token.Claims.(*Claims); ok && token.Valid {
+		return claims, nil
+	}
+
+	// 如果 token 无效（例如，过期但没有返回错误，或者 Claims 类型不匹配），则返回一个通用错误
+	return nil, fmt.Errorf("token is invalid")
+}

internal/domain/token/token_service_test.go

@@ -0,0 +1,107 @@
+package token_test
+
+import (
+	"errors"
+	"testing"
+	"time"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/app/service/token"
+	"github.com/golang-jwt/jwt/v5"
+)
+
+func TestGenerateToken(t *testing.T) {
+	// 使用一个测试密钥初始化 TokenService
+	testSecret := []byte("test_secret_key")
+	service := token.NewTokenService(testSecret)
+
+	userID := uint(123)
+	tokenString, err := service.GenerateToken(userID)
+
+	if err != nil {
+		t.Fatalf("生成令牌失败: %v", err)
+	}
+
+	if tokenString == "" {
+		t.Fatal("生成的令牌字符串为空")
+	}
+
+	// 解析 token 以确保其有效性及声明
+	claims, err := service.ParseToken(tokenString)
+	if err != nil {
+		t.Fatalf("生成后解析令牌失败: %v", err)
+	}
+
+	if claims.UserID != userID {
+		t.Errorf("期望用户ID %d, 实际为 %d", userID, claims.UserID)
+	}
+
+	// 检查 token 是否未过期 (在合理范围内)
+	if claims.ExpiresAt == nil || claims.ExpiresAt.Time.Before(time.Now().Add(-time.Minute)) {
+		t.Errorf("令牌过期时间无效或已过期")
+	}
+
+	if claims.Issuer != "pig-farm-controller" {
+		t.Errorf("期望签发者 \"pig-farm-controller\", 实际为 \"%s\"", claims.Issuer)
+	}
+}
+
+func TestParseToken(t *testing.T) {
+	// 使用两个不同的测试密钥
+	correctSecret := []byte("the_correct_secret")
+	wrongSecret := []byte("a_very_wrong_secret")
+
+	serviceWithCorrectKey := token.NewTokenService(correctSecret)
+	serviceWithWrongKey := token.NewTokenService(wrongSecret)
+
+	userID := uint(456)
+
+	// 1. 生成一个有效的 token
+	validToken, err := serviceWithCorrectKey.GenerateToken(userID)
+	if err != nil {
+		t.Fatalf("为解析测试生成有效令牌失败: %v", err)
+	}
+
+	// 测试用例 1: 使用正确的密钥成功解析
+	claims, err := serviceWithCorrectKey.ParseToken(validToken)
+	if err != nil {
+		t.Errorf("使用正确密钥解析有效令牌失败: %v", err)
+	}
+	if claims.UserID != userID {
+		t.Errorf("解析有效令牌时期望用户ID %d, 实际为 %d", userID, claims.UserID)
+	}
+
+	// 测试用例 2: 无效 token (例如, 格式错误的字符串)
+	invalidTokenString := "this.is.not.a.valid.jwt"
+	_, err = serviceWithCorrectKey.ParseToken(invalidTokenString)
+	if err == nil {
+		t.Error("解析格式错误的令牌意外成功")
+	}
+
+	// 测试用C:\Users\divano\Desktop\work\AA-Pig\pig-farm-controller\internal\infra\repository\plan_repository_test.go例 3: 过期 token
+	expiredClaims := token.Claims{
+		UserID: userID,
+		RegisteredClaims: jwt.RegisteredClaims{
+			ExpiresAt: jwt.NewNumericDate(time.Now().Add(-time.Hour)), // 1 小时前
+			Issuer:    "pig-farm-controller",
+		},
+	}
+	expiredTokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, expiredClaims)
+	expiredTokenString, err := expiredTokenClaims.SignedString(correctSecret)
+	if err != nil {
+		t.Fatalf("生成过期令牌失败: %v", err)
+	}
+	_, err = serviceWithCorrectKey.ParseToken(expiredTokenString)
+	if err == nil {
+		t.Error("解析过期令牌意外成功")
+	}
+
+	// 新增测试用例 4: 使用错误的密钥解析
+	_, err = serviceWithWrongKey.ParseToken(validToken)
+	if err == nil {
+		t.Error("使用错误密钥解析令牌意外成功")
+	}
+	// 我们可以更精确地检查错误类型，以确保它是签名错误
+	if !errors.Is(err, jwt.ErrTokenSignatureInvalid) {
+		t.Errorf("期望得到签名无效错误 (ErrTokenSignatureInvalid)，但得到了: %v", err)
+	}
+}