initial commit

gateway/internal/config/config.go

@@ -0,0 +1,15 @@
+package config
+
+import (
+	"github.com/zeromicro/go-zero/rest"
+	"github.com/zeromicro/go-zero/zrpc"
+)
+
+type Config struct {
+	rest.RestConf
+	JwtAuth struct {
+		AccessSecret string
+		AccessExpire int64
+	}
+	UserRpc zrpc.RpcClientConf
+}

gateway/internal/handler/getuserinfohandler.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+	"godemo/gateway/internal/logic"
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+)
+
+func getUserInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.GetUserInfoReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := logic.NewGetUserInfoLogic(r.Context(), svcCtx)
+		resp, err := l.GetUserInfo(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

gateway/internal/handler/loginhandler.go

@@ -0,0 +1,51 @@
+package handler
+
+import (
+	"net/http"
+
+	"godemo/gateway/internal/logic"
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func loginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.LoginReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := logic.NewLoginLogic(r.Context(), svcCtx)
+		resp, err := l.Login(&req)
+
+		if err != nil {
+			if s, ok := status.FromError(err); ok {
+				// 判断 gRPC 错误码，并自定义 HTTP 返回
+				var code int
+				switch s.Code() {
+				case codes.NotFound:
+					code = http.StatusNotFound
+				case codes.InvalidArgument:
+					code = http.StatusBadRequest
+				default:
+					code = http.StatusInternalServerError
+				}
+
+				// 自定义 JSON 错误格式
+				httpx.WriteJson(w, code, map[string]interface{}{
+					"code":    code,
+					"message": s.Message(),
+				})
+			} else {
+				httpx.ErrorCtx(r.Context(), w, err)
+			}
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

gateway/internal/handler/logouthandler.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+	"godemo/gateway/internal/logic"
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+)
+
+func logoutHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.LogoutReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := logic.NewLogoutLogic(r.Context(), svcCtx)
+		resp, err := l.Logout(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

gateway/internal/handler/registerhandler.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+	"godemo/gateway/internal/logic"
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+)
+
+func registerHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.RegisterReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := logic.NewRegisterLogic(r.Context(), svcCtx)
+		resp, err := l.Register(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

gateway/internal/handler/routes.go

@@ -0,0 +1,45 @@
+// Code generated by goctl. DO NOT EDIT.
+// goctl 1.8.3
+
+package handler
+
+import (
+	"net/http"
+
+	"godemo/gateway/internal/svc"
+
+	"github.com/zeromicro/go-zero/rest"
+)
+
+func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
+	server.AddRoutes(
+		[]rest.Route{
+			{
+				Method:  http.MethodPost,
+				Path:    "/api/user/login",
+				Handler: loginHandler(serverCtx),
+			},
+			{
+				Method:  http.MethodPost,
+				Path:    "/api/user/register",
+				Handler: registerHandler(serverCtx),
+			},
+		},
+	)
+
+	server.AddRoutes(
+		[]rest.Route{
+			{
+				Method:  http.MethodGet,
+				Path:    "/api/user/:user_id",
+				Handler: getUserInfoHandler(serverCtx),
+			},
+			{
+				Method:  http.MethodPost,
+				Path:    "/api/user/logout",
+				Handler: logoutHandler(serverCtx),
+			},
+		},
+		rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret),
+	)
+}

gateway/internal/logic/getuserinfologic.go

@@ -0,0 +1,45 @@
+package logic
+
+import (
+	"context"
+
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+	"godemo/user/user"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetUserInfoLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserInfoLogic {
+	return &GetUserInfoLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetUserInfoLogic) GetUserInfo(req *types.GetUserInfoReq) (resp *types.GetUserInfoResp, err error) {
+	// todo: add your logic here and delete this line
+	rpcResp, err := l.svcCtx.UserRpc.GetUserInfo(l.ctx, &user.GetUserInfoRequest{
+		UserId: req.UserId,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	// 构造 API 返回值
+	resp = &types.GetUserInfoResp{
+		UserId:    rpcResp.UserId,
+		Username:  rpcResp.Username,
+		Email:     rpcResp.Email,
+		CreatedAt: rpcResp.CreatedAt,
+		Roles:     rpcResp.Roles,
+	}
+	return
+}

gateway/internal/logic/loginlogic.go

@@ -0,0 +1,43 @@
+package logic
+
+import (
+	"context"
+
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+	"godemo/user/user"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type LoginLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic {
+	return &LoginLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *LoginLogic) Login(req *types.LoginReq) (resp *types.LoginResp, err error) {
+	// 调用 user RPC 的 Login 方法
+	rpcResp, err := l.svcCtx.UserRpc.Login(l.ctx, &user.LoginRequest{
+		Username: req.Username,
+		Password: req.Password,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	// 构造 API 返回值
+	resp = &types.LoginResp{
+		Token:     rpcResp.Token,
+		ExpiresAt: rpcResp.ExpiresAt,
+	}
+	return
+}

gateway/internal/logic/logoutlogic.go

@@ -0,0 +1,30 @@
+package logic
+
+import (
+	"context"
+
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type LogoutLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewLogoutLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LogoutLogic {
+	return &LogoutLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *LogoutLogic) Logout(req *types.LogoutReq) (resp *types.LogoutResp, err error) {
+	// todo: add your logic here and delete this line
+
+	return
+}

gateway/internal/logic/registerlogic.go

@@ -0,0 +1,40 @@
+package logic
+
+import (
+	"context"
+
+	"godemo/gateway/internal/svc"
+	"godemo/gateway/internal/types"
+	"godemo/user/user"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type RegisterLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
+	return &RegisterLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *RegisterLogic) Register(req *types.RegisterReq) (resp *types.RegisterResp, err error) {
+	rpcResp, err := l.svcCtx.UserRpc.Register(l.ctx, &user.RegisterRequest{
+		Email:    req.Email,
+		Password: req.Password,
+		Username: req.Username,
+	})
+	if err != nil {
+		return nil, err
+	}
+	resp = &types.RegisterResp{
+		UserId: rpcResp.UserId,
+	}
+	return
+}

gateway/internal/svc/servicecontext.go

@@ -0,0 +1,21 @@
+package svc
+
+import (
+	"godemo/gateway/internal/config"
+	"godemo/user/user"
+
+	"github.com/zeromicro/go-zero/zrpc"
+)
+
+type ServiceContext struct {
+	Config  config.Config
+	UserRpc user.UserClient
+}
+
+func NewServiceContext(c config.Config) *ServiceContext {
+	conn := zrpc.MustNewClient(c.UserRpc).Conn()
+	return &ServiceContext{
+		Config:  c,
+		UserRpc: user.NewUserClient(conn),
+	}
+}

gateway/internal/types/types.go

@@ -0,0 +1,44 @@
+// Code generated by goctl. DO NOT EDIT.
+// goctl 1.8.3
+
+package types
+
+type GetUserInfoReq struct {
+	UserId string `path:"user_id"`
+}
+
+type GetUserInfoResp struct {
+	UserId    string   `json:"user_id"`
+	Username  string   `json:"username"`
+	Email     string   `json:"email"`
+	CreatedAt int64    `json:"created_at"`
+	Roles     []string `json:"roles"`
+}
+
+type LoginReq struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+type LoginResp struct {
+	Token     string `json:"token"`
+	ExpiresAt int64  `json:"expires_at"`
+}
+
+type LogoutReq struct {
+	Token string `json:"token"`
+}
+
+type LogoutResp struct {
+	Success bool `json:"success"`
+}
+
+type RegisterReq struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+	Email    string `json:"email"`
+}
+
+type RegisterResp struct {
+	UserId string `json:"user_id"`
+}