自动建立表-toekn

config/loader.go

@@ -67,16 +67,40 @@ func convertMap(input map[interface{}]interface{}) map[string]interface{} {
 	return output
 }
 
-// findConfigFile 查找配置文件（内部函数保持不变）
+// findConfigFile 查找配置文件
 func findConfigFile() (string, error) {
 	exePath, _ := os.Executable()
 	exeDir := filepath.Dir(exePath)
 
+	// 获取当前工作目录
+	currentDir, err := os.Getwd()
+	if err != nil {
+		currentDir = "."
+	}
+
+	appName := filepath.Base(currentDir)
+	// 构建可能的配置文件路径
 	possiblePaths := []string{
+
+		// 1. 当前目录的 appName.yaml
+		filepath.Join(currentDir, appName+".yaml"),
+
+		// 2. 当前目录的上级目录的 appName.yaml
+		filepath.Join(filepath.Dir(currentDir), appName+".yaml"),
+
+		// 3. 可执行文件所在目录的 appName.yaml
+		filepath.Join(exeDir, appName+".yaml"),
+
+		// 4. 可执行文件所在目录的上级目录的 appName.yaml
+		filepath.Join(filepath.Dir(exeDir), appName+".yaml"),
+
+		// 5. 原始逻辑中的 db.yaml 路径
 		filepath.Join(exeDir, "db.yaml"),
 		filepath.Join(exeDir, "config", "db.yaml"),
 		"db.yaml",
 		"config/db.yaml",
+
+		// 6. 环境变量指定的路径
 		os.Getenv("DB_CONFIG_PATH"),
 	}
 
@@ -84,24 +108,35 @@ func findConfigFile() (string, error) {
 		if path == "" {
 			continue
 		}
-		if _, err := os.Stat(path); err == nil {
-			log.Printf("✅ Using config file: %s\n", path)
-			return path, nil
+		log.Printf("find config file: %s", path)
+		// 清理路径
+		cleanPath := filepath.Clean(path)
+		if _, err := os.Stat(cleanPath); err == nil {
+			log.Printf("✅ Using config file: %s\n", cleanPath)
+			return cleanPath, nil
 		}
 	}
 
 	return "", fmt.Errorf(`no configuration file found
 
 Searched locations:
-1. %s
-2. %s
-3. ./db.yaml
-4. ./config/db.yaml
-5. DB_CONFIG_PATH环境变量指定的路径
+1. %s (当前目录)
+2. %s (上级目录)
+3. %s (可执行文件目录)
+4. %s (可执行文件上级目录)
+5. %s
+6. %s
+7. ./db.yaml
+8. ./config/db.yaml
+9. DB_CONFIG_PATH环境变量指定的路径
 
 请确保配置文件存在，或通过环境变量指定：
 export DB_CONFIG_PATH=/your/config/path/db.yaml
 set DB_CONFIG_PATH=C:\your\config\path\db.yaml (Windows)`,
+		filepath.Join(currentDir, appName+".yaml"),
+		filepath.Join(filepath.Dir(currentDir), appName+".yaml"),
+		filepath.Join(exeDir, appName+".yaml"),
+		filepath.Join(filepath.Dir(exeDir), appName+".yaml"),
 		filepath.Join(exeDir, "db.yaml"),
 		filepath.Join(exeDir, "config", "db.yaml"))
 }

go.mod

@@ -44,6 +44,7 @@ require (
 
 require (
 	github.com/elastic/go-elasticsearch/v8 v8.19.0
+	github.com/golang-jwt/jwt/v5 v5.3.0
 	github.com/iancoleman/strcase v0.3.0
 	github.com/miekg/dns v1.1.68 // indirect
 	github.com/mitchellh/go-homedir v1.1.0 // indirect

go.sum

@@ -24,6 +24,8 @@ github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
 github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
 github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
 github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
+github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
+github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
 github.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
 github.com/google/btree v1.0.0 h1:0udJVsspx3VBr5FwtLhQQtuAsVc79tTq0ocGIPAU6qo=
 github.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=

sdk/opencode/opencode_api_client.gen.go

@@ -13,6 +13,8 @@ import (
 	"net/url"
 	"strings"
 
+	"github.com/oapi-codegen/runtime"
+)
 
 // RequestEditorFn  is the function signature for the RequestEditor callback function
 type RequestEditorFn func(ctx context.Context, req *http.Request) error

util/jwt_util.go

@@ -0,0 +1,111 @@
+package util
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/golang-jwt/jwt/v5"
+)
+
+// Claims 结构定义
+type Claims struct {
+	UserID    string                 `json:"user_id"`
+	Username  string                 `json:"username"`
+	TenantID  string                 `json:"tenant_id"`
+	ProjectID string                 `json:"project_id"`
+	Extra     map[string]interface{} `json:"extra,omitempty"`
+	jwt.RegisteredClaims
+}
+
+// CreateToken 创建Token
+// claims: Claims对象
+// expiresHours: 过期小时数（0表示默认7天）
+// secretKey: 密钥
+func CreateToken(claims *Claims, expiresHours int, secretKey string) (string, error) {
+	return createTokenInternal(claims, 0, expiresHours, secretKey)
+}
+
+// CreateTokenDays 创建Token（按天）
+// claims: Claims对象
+// expiresDays: 过期天数（优先使用）
+// secretKey: 密钥
+func CreateTokenDays(claims *Claims, expiresDays int, secretKey string) (string, error) {
+	return createTokenInternal(claims, expiresDays, 0, secretKey)
+}
+
+// 内部实现
+func createTokenInternal(claims *Claims, expiresDays, expiresHours int, secretKey string) (string, error) {
+	if secretKey == "" {
+		return "", fmt.Errorf("secret key is required")
+	}
+	if claims == nil {
+		return "", fmt.Errorf("claims cannot be nil")
+	}
+
+	// 计算过期时间
+	var expiresDuration time.Duration
+	if expiresDays > 0 {
+		expiresDuration = time.Duration(expiresDays) * 24 * time.Hour
+	} else if expiresHours > 0 {
+		expiresDuration = time.Duration(expiresHours) * time.Hour
+	} else {
+		// 默认7天
+		expiresDuration = 7 * 24 * time.Hour
+	}
+
+	// 设置时间字段（内部处理）
+	now := time.Now()
+	claims.RegisteredClaims = jwt.RegisteredClaims{
+		ExpiresAt: jwt.NewNumericDate(now.Add(expiresDuration)),
+		IssuedAt:  jwt.NewNumericDate(now),
+		NotBefore: jwt.NewNumericDate(now),
+		Issuer:    "jwt-app",
+		Subject:   claims.UserID,
+	}
+
+	// 生成Token
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	return token.SignedString([]byte(secretKey))
+}
+
+// ParseToken 解析Token
+// tokenString: Token字符串
+// secretKey: 密钥
+func ParseToken(tokenString, secretKey string) (*Claims, error) {
+	if secretKey == "" {
+		return nil, fmt.Errorf("secret key is required")
+	}
+
+	claims := &Claims{}
+	token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
+		return []byte(secretKey), nil
+	})
+
+	if err != nil {
+		return nil, fmt.Errorf("parse token failed: %w", err)
+	}
+
+	if !token.Valid {
+		return nil, fmt.Errorf("invalid token")
+	}
+
+	return claims, nil
+}
+
+// ValidateToken 验证Token是否有效
+func ValidateToken(tokenString, secretKey string) bool {
+	_, err := ParseToken(tokenString, secretKey)
+	return err == nil
+}
+
+// RefreshToken 刷新Token
+func RefreshToken(tokenString string, expiresHours int, secretKey string) (string, error) {
+	// 解析原Token
+	claims, err := ParseToken(tokenString, secretKey)
+	if err != nil {
+		return "", err
+	}
+
+	// 重新创建Token
+	return CreateToken(claims, expiresHours, secretKey)
+}

webx/router/router_service.go

@@ -96,6 +96,13 @@ type RouteBuilder struct {
 	description string
 }
 
+// WrapMiddleware 通用包装函数
+func WrapMiddleware[T any](middleware func(http.Handler, T) http.Handler, dep T) func(http.Handler) http.Handler {
+	return func(next http.Handler) http.Handler {
+		return middleware(next, dep)
+	}
+}
+
 // Use 添加路由级中间件
 func (rb *RouteBuilder) Use(middleware ...func(http.Handler) http.Handler) *RouteBuilder {
 	rb.middlewares = append(rb.middlewares, middleware...)