oneao/workspace
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

u

Browse Source
This commit is contained in:
oneao committed 2025-08-12 21:47:37 +08:00
1 parent 9af3234672
commit ed8bee3d77
11 files changed
+171 -57
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

code/go-project/base-farmework/cmd/app/main.go
+4 -4
View File
@@ -1,8 +1,8 @@
package main
import (
appConfig "base-framework/configs/app"
"base-framework/internal/app"
commonConfig "base-framework/pkg/config"
"log"
"net/http"
"strconv"
@@ -11,12 +11,12 @@ import (
)
func main() {
_, err := appConfig.InitAppConfig("./configs/app/config.yaml")
_, err := commonConfig.InitApplicationConfig("./configs/app/application.yaml")
if err != nil {
log.Fatalf("加载配置失败: %v", err)
}
err = appConfig.InitDBConfig("D:\\db.yaml")
err = commonConfig.InitDBConfig("./configs/app/db.yaml")
if err != nil {
log.Fatalf("加载数据库配置失败： %v", err)
return
@@ -24,7 +24,7 @@ func main() {
r := app.InitAppRouter()
err = http.ListenAndServe(":"+strconv.Itoa(appConfig.Server.Port), r)
err = http.ListenAndServe(":"+strconv.Itoa(commonConfig.Server.Port), r)
if err != nil {
log.Fatalf("server start failed: %v", err)
}
code/go-project/base-farmework/configs/app/config.yaml → code/go-project/base-farmework/configs/app/application.yaml
View File
File renamed without changes.
code/go-project/base-farmework/internal/app/api/auth.go
+1
View File
@@ -0,0 +1 @@
package api
code/go-project/base-farmework/internal/app/handle/auth.go
+31
View File
@@ -0,0 +1,31 @@
package handle
import "base-framework/pkg/router"
func Login(c *router.Context) {
// 直接用 BindJSON 绑定请求体 JSON 到结构体
var req struct {
OrgID string `json:"orgID"`
UserID string `json:"userID"`
Password string `json:"password"`
}
if err := c.BindJSON(&req); err != nil {
c.JSON(400, map[string]string{"error": "请求体格式错误"})
return
}
// 模拟校验
if req.UserID == "" || req.Password == "" {
c.JSON(400, map[string]string{"error": "账号或密码不能为空"})
return
}
// 登录成功，返回 token 示例
c.JSON(200, map[string]interface{}{
"code": "0000",
"message": "登录成功",
"data": map[string]string{
"token": "这里是token字符串",
},
})
}
code/go-project/base-farmework/internal/app/router.go
+2 -1
View File
@@ -10,8 +10,9 @@ func InitAppRouter() *router.Router {
r := router.NewRouter()
r.Use(middleware.Recover())
//r.Use(middleware.Auth()).ExcludePaths("/login")
r.Use(middleware.Auth()).ExcludePaths("/login")
r.Use(middleware.Logger())
initApi(r)
return r
code/go-project/base-farmework/configs/app/config.go → code/go-project/base-farmework/pkg/config/application.go
+2 -2
View File
@@ -1,4 +1,4 @@
package app
package config
import (
"time"
@@ -20,7 +20,7 @@ var (
JWT JWTConfig
)
func InitAppConfig(configPath string) (error, error) {
func InitApplicationConfig(configPath string) (error, error) {
v := viper.New()
v.SetConfigFile(configPath)
v.SetConfigType("yaml")
code/go-project/base-farmework/configs/app/db.go → code/go-project/base-farmework/pkg/config/db.go
+1 -1
View File
@@ -1,4 +1,4 @@
package app
package config
import (
"fmt"
code/go-project/base-farmework/pkg/middleware/auth.go
+27 -17
View File
@@ -1,9 +1,12 @@
package middleware
import (
"base-framework/pkg/config"
"base-framework/pkg/router"
"base-framework/pkg/utils"
"net/http"
"base-framework/pkg/utils/response"
"errors"
"fmt"
"strings"
)
@@ -13,43 +16,50 @@ func Auth() router.HandlerFunc {
userIdHeader := c.Header("user_id")
orgIDHeader := c.Header("org_id")
if tokenHeader == "" {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "Authorization header missing"})
return
}
if userIdHeader == "" {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "user_id header missing"})
return
}
if orgIDHeader == "" {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "org_id header missing"})
// 缺少登录信息
if tokenHeader == "" || userIdHeader == "" || orgIDHeader == "" {
response.Error(c).Code(response.CodeNoLogin).Send()
return
}
// Bearer token 格式校验
parts := strings.Fields(tokenHeader)
if len(parts) != 2 || strings.ToLower(parts[0]) != "bearer" {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "Authorization header format must be Bearer {token}"})
if len(parts) != 2 || strings.ToLower(parts[0]) != "Bearer" {
response.Error(c).Code(response.CodeInvalidToken).Send()
return
}
// 解析 token
tokenStr := parts[1]
claims, err := utils.VerifyToken(tokenStr)
if err != nil {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "Invalid token: " + err.Error()})
switch {
case errors.Is(err, utils.ErrTokenExpired):
response.Error(c).Code(response.CodeLoginExpired).Send()
default:
response.Error(c).Code(response.CodeInvalidToken).Send()
}
return
}
// user_id 校验
if claims.UserID != userIdHeader {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "user_id does not match token"})
response.Error(c).Code(response.CodeInvalidToken).Send()
return
}
// org_id 校验
if claims.OrgID != orgIDHeader {
c.JSON(http.StatusUnauthorized, map[string]string{"error": "org_id does not match token"})
response.Error(c).Code(response.CodeInvalidToken).Send()
return
}
// 认证通过，继续执行后续中间件或处理器
configs := config.GetDBConfigs()
for k := range configs {
fmt.Println(k)
}
c.Next()
}
}
code/go-project/base-farmework/pkg/router/context.go
+63 -6
View File
@@ -1,23 +1,58 @@
package router
import (
"bytes"
"encoding/json"
"io"
"net/http"
"sync"
)
// BodyCache 专门负责缓存请求体，保证只读一次
type BodyCache struct {
once sync.Once
Data []byte
Err error
}
// Load 读取请求体并缓存，只执行一次
func (b *BodyCache) Load(r *http.Request) ([]byte, error) {
b.once.Do(func() {
if r.Body == nil {
b.Err = http.ErrBodyNotAllowed
return
}
var buf bytes.Buffer
b.Err = func() error {
_, err := io.Copy(&buf, r.Body)
return err
}()
if b.Err != nil {
return
}
b.Data = buf.Bytes()
// 重新设置请求体方便后续读取
r.Body = io.NopCloser(bytes.NewReader(b.Data))
})
return b.Data, b.Err
}
// Context 自定义请求上下文，组合 BodyCache
type Context struct {
Writer http.ResponseWriter
Request *http.Request
Params map[string]string
index int
handlers []HandlerFunc
Index int
Handlers []HandlerFunc
BodyCache BodyCache // 请求体缓存
}
// Next 执行下一个中间件或处理函数
func (c *Context) Next() {
c.index++
if c.index < len(c.handlers) {
c.handlers[c.index](c)
c.Index++
if c.Index < len(c.Handlers) {
c.Handlers[c.Index](c)
}
}
@@ -31,7 +66,7 @@ func (c *Context) Header(key string) string {
return c.Request.Header.Get(key)
}
// JSON 返回JSON格式响应
// JSON 返回 JSON 格式响应
func (c *Context) JSON(statusCode int, data interface{}) {
c.Writer.Header().Set("Content-Type", "application/json; charset=utf-8")
c.Writer.WriteHeader(statusCode)
@@ -39,3 +74,25 @@ func (c *Context) JSON(statusCode int, data interface{}) {
http.Error(c.Writer, err.Error(), http.StatusInternalServerError)
}
}
// Body 方便读取请求体，实际调用 BodyCache 的 Load 方法
func (c *Context) Body() ([]byte, error) {
return c.BodyCache.Load(c.Request)
}
// BindJSON 反序列化 JSON 请求体到 obj
func (c *Context) BindJSON(obj interface{}) error {
body, err := c.Body()
if err != nil {
return err
}
return json.Unmarshal(body, obj)
}
// PostForm 获取表单参数
func (c *Context) PostForm(key string) string {
if err := c.Request.ParseForm(); err != nil {
return ""
}
return c.Request.FormValue(key)
}
code/go-project/base-farmework/pkg/utils/jwt.go
+22 -10
View File
@@ -1,22 +1,28 @@
package utils
import (
appConfig "base-framework/configs/app"
commonConfig "base-framework/pkg/config"
"errors"
"github.com/golang-jwt/jwt"
"time"
"github.com/golang-jwt/jwt"
)
// CustomClaims 定义自己的 payload 结构，可以根据需要扩展
var (
ErrTokenExpired = errors.New("token expired")
ErrTokenInvalid = errors.New("token invalid")
)
// CustomClaims 定义自己的 payload 结构
type CustomClaims struct {
OrgID string `json:"org_id"`
UserID string `json:"user_id"`
jwt.StandardClaims
}
// CreateToken 创建一个JWT token
// CreateToken 创建一个 JWT token
func CreateToken(orgID, userID string) (string, error) {
expireTime := time.Now().Add(appConfig.JWT.Expiry).Unix()
expireTime := time.Now().Add(commonConfig.JWT.Expiry).Unix()
claims := CustomClaims{
OrgID: orgID,
@@ -28,29 +34,35 @@ func CreateToken(orgID, userID string) (string, error) {
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
signKey := []byte(appConfig.JWT.Secret)
signKey := []byte(commonConfig.JWT.Secret)
return token.SignedString(signKey)
}
// VerifyToken 验证并解析 JWT token
func VerifyToken(tokenString string) (*CustomClaims, error) {
signKey := []byte(appConfig.JWT.Secret)
signKey := []byte(commonConfig.JWT.Secret)
token, err := jwt.ParseWithClaims(tokenString, &CustomClaims{}, func(token *jwt.Token) (interface{}, error) {
// 校验签名算法
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
return nil, errors.New("unexpected signing method")
return nil, ErrTokenInvalid
}
return signKey, nil
})
if err != nil {
return nil, err
// 判断是否过期错误
if ve, ok := err.(*jwt.ValidationError); ok {
if ve.Errors&jwt.ValidationErrorExpired != 0 {
return nil, ErrTokenExpired
}
}
return nil, ErrTokenInvalid
}
if claims, ok := token.Claims.(*CustomClaims); ok && token.Valid {
return claims, nil
}
return nil, errors.New("invalid token")
return nil, ErrTokenInvalid
}
code/go-project/base-farmework/pkg/utils/response/response.go
+18 -16
View File
@@ -12,23 +12,27 @@ type Result struct {
Data interface{} `json:"data,omitempty"`
}
// 常量状态码
const (
CodeSuccess = "0000"
CodeFail = "9999"
// 通用
CodeSuccess = "0000" // 成功
CodeFail = "0001" // 失败
// 鉴权 / 登录
CodeNoLogin = "1001" // 未登录
CodeLoginExpired = "1002" // 登录过期
CodeInvalidToken = "1003" // Token 无效
CodeInvalidOrgCode = "1004" // 机构码有误
CodeInvalidAccount = "1005" // 账号或密码有误
)
// code 对应默认提示
var codeMessages = map[string]string{
CodeSuccess: "请求成功",
CodeFail: "请求失败",
"1001": "缺少用户ID",
"1002": "未授权",
"1003": "无权限访问",
"1004": "资源不存在",
"9000": "系统内部错误",
CodeSuccess: "请求成功",
CodeFail: "请求失败",
CodeNoLogin: "未登录",
CodeLoginExpired: "登录已过期",
CodeInvalidToken: "Token 无效",
CodeInvalidOrgCode: "机构码有误",
CodeInvalidAccount: "账号或密码有误",
}
// 兜底提示
@@ -65,9 +69,7 @@ func Error(c *router.Context) *Builder {
// Code 设置状态码（自动填充默认提示，除非后面手动改）
func (b *Builder) Code(code string) *Builder {
b.result.Code = code
if b.result.Message == "" || b.result.Message == getMessage(b.result.Code) {
b.result.Message = getMessage(code)
}
b.result.Message = getMessage(code)
return b
}