zhuyujun/go-desk-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增用户服务连接

Browse Source
This commit is contained in:
zyj
2025-09-02 11:40:46 +08:00
parent 3aae35abb7
commit 7371d04216
11 changed files with 195 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -6,6 +6,7 @@ yarn-debug.log*
yarn-error.log*
/public/uploads/
config.yaml
*.pb.go
# Editor directories and files
.idea

53
api/user.go Normal file
View File

@@ -0,0 +1,53 @@
package api
import (
"context"
"fmt"
grpcClient "go-desk-service/grpc-client"
"go-desk-service/libs"
userpb "go-desk-service/proto/gen"
"log"
"net/http"
"time"
"github.com/gin-gonic/gin"
)
type User struct {
}
type LoginRequestParams struct {
Username string `form:"username" json:"username" uri:"username" xml:"username" binding:"required"`
Password string `form:"password" json:"password" uri:"password" xml:"password" binding:"required"`
}
func (*User) Login(ctx *gin.Context) {
var res *libs.ErrorInfo
var paramsJson LoginRequestParams
if err := ctx.ShouldBindJSON(&paramsJson); err != nil {
res = libs.ErrorCode["ParamsError"]
fmt.Println(err)
ctx.JSON(http.StatusBadRequest, gin.H{
"code": res.Code,
"msg": res.Msg,
"data": res.Data,
})
return
}
// 获取当前的grpc连接
client := grpcClient.GetUserClient()
ctx1, cancel := context.WithTimeout(context.Background(), time.Second*5)
defer cancel()
loginResp, err := client.Login(ctx1, &userpb.LoginRequest{
Username: paramsJson.Username,
Password: paramsJson.Password,
})
if err != nil {
log.Fatalf("登录失败: %v", err)
}
ctx.JSON(http.StatusBadRequest, gin.H{
"code": 200,
"msg": "登录成功",
"data": loginResp,
})
}

2
cmd/main.go
View File

@@ -3,6 +3,7 @@ package main
import (
"fmt"
"go-desk-service/config"
grpcClient "go-desk-service/grpc-client"
"go-desk-service/libs"
"go-desk-service/router"
"log"
@@ -23,6 +24,7 @@ func main() {
go handleShutdownSignals()
libs.Connect()
grpcClient.UserClientInit()
app := gin.Default()
router.Init(app)
app.Run(":" + strconv.Itoa(appConfig.Port))

1
config/index.go
View File

@@ -11,6 +11,7 @@ type AppConfig struct {
Database DatabaseConfig `yaml:"database"`
WorkerID int64 `yaml:"workerID"`
DatacenterID int64 `yaml:"datacenterID"`
GrpcUrl string `yaml:"grpcUrl"`
}
type DatabaseConfig struct {

2
go.mod
View File

@@ -42,6 +42,8 @@ require (
golang.org/x/sync v0.16.0 // indirect
golang.org/x/sys v0.35.0 // indirect
golang.org/x/text v0.28.0 // indirect
google.golang.org/genproto/googleapis/rpc v0.0.0-20250707201910-8d1bb00bc6a7 // indirect
google.golang.org/grpc v1.75.0 // indirect
google.golang.org/protobuf v1.36.8 // indirect
gorm.io/driver/postgres v1.6.0 // indirect
gorm.io/gorm v1.30.2 // indirect

4
go.sum
View File

@@ -126,6 +126,10 @@ golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
google.golang.org/genproto/googleapis/rpc v0.0.0-20250707201910-8d1bb00bc6a7 h1:pFyd6EwwL2TqFf8emdthzeX+gZE1ElRq3iM8pui4KBY=
google.golang.org/genproto/googleapis/rpc v0.0.0-20250707201910-8d1bb00bc6a7/go.mod h1:qQ0YXyHHx3XkvlzUtpXDkS29lDSafHMZBAZDc03LQ3A=
google.golang.org/grpc v1.75.0 h1:+TW+dqTd2Biwe6KKfhE5JpiYIBWq865PhKGSXiivqt4=
google.golang.org/grpc v1.75.0/go.mod h1:JtPAzKiq4v1xcAB2hydNlWI2RnF85XXcV0mhKXr2ecQ=
google.golang.org/protobuf v1.36.6 h1:z1NpPI8ku2WgiWnf+t9wTPsn6eP1L7ksHUlkfLvd9xY=
google.golang.org/protobuf v1.36.6/go.mod h1:jduwjTPXsFjZGTmRluh+L6NjiWu7pchiJ2/5YcXBHnY=
google.golang.org/protobuf v1.36.8 h1:xHScyCOEuuwZEc6UtSOvPbAT4zRh0xcNRYekJwfqyMc=

36
grpc-client/user_client.go Normal file
View File

@@ -0,0 +1,36 @@
package grpcClient
import (
"fmt"
"go-desk-service/config"
userpb "go-desk-service/proto/gen"
"log"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
var UserClient userpb.UserServiceClient
func UserClientInit() {
appConfig, err := config.LoadConfig()
if err != nil {
fmt.Println(err)
fmt.Println("读取配置失败")
}
conn, err := grpc.NewClient(
appConfig.GrpcUrl, // gRPC服务器地址
grpc.WithTransportCredentials(insecure.NewCredentials()), // 使用非安全连接（测试用）
)
if err != nil {
log.Fatalf("连接失败: %v", err)
}
defer conn.Close()
client := userpb.NewUserServiceClient(conn)
UserClient = client
}
func GetUserClient() userpb.UserServiceClient {
return UserClient
}

22
libs/error_code.go Normal file
View File

@@ -0,0 +1,22 @@
package libs
type ErrorInfo struct {
Code int
Data any
Msg string
}
var ErrorCode = map[string]*ErrorInfo{
"LoginFailed": {Code: 0, Data: "", Msg: "账号异常"},
"LoginSuccessful": {Code: 200, Data: "", Msg: "登录成功"},
"NetworkTimeout": {Code: 1, Data: "", Msg: "网络超时"},
"AccountHasBeenTakenOffline": {Code: 2, Data: "", Msg: "账号已下线，请重新登录"},
"OtherError": {Code: 3, Data: "", Msg: "其他错误"},
"TransferSuccessful": {Code: 200, Data: "", Msg: "转账成功"},
"TransferFailed": {Code: 4, Data: "", Msg: "转账失败"},
"AccountRequiresEmailVerification": {Code: 0, Data: "", Msg: "账号需要邮箱验证"},
"AccountRequiresSMSVerification": {Code: 0, Data: "", Msg: "账号需要短信验证"},
"AccountRequiresRobotVerification": {Code: 0, Data: "", Msg: "账号需要机器人验证"},
"BrowserOpenFail": {Code: 0, Data: "", Msg: "浏览器开启失败"},
"AccountFailed": {Code: 0, Data: "", Msg: "账号不存在"},
}

56
proto/user.proto Normal file
View File

@@ -0,0 +1,56 @@
syntax = "proto3";
package user;
option go_package = "./proto/gen;userpb";
// 登录请求
message LoginRequest {
string username = 1;
string password = 2;
}
// 登录响应
message LoginResponse {
int64 user_id = 1;
string username = 2;
string email = 3;
string access_token = 4;
int32 expires_in = 5; // token过期时间(秒)
}
// 用户注册请求
message RegisterRequest {
string username = 1;
string password = 2;
string email = 3;
string nickname = 4;
string phone = 5;
}
// 用户注册响应
message RegisterResponse {
int64 user_id = 1;
string username = 2;
string email = 3;
}
// 令牌验证请求
message ValidateTokenRequest {
string access_token = 1;
}
// 令牌验证响应
message ValidateTokenResponse {
int64 user_id = 1;
bool is_valid = 2;
}
// 用户服务定义
service UserService {
// 用户登录
rpc Login(LoginRequest) returns (LoginResponse);
// 用户注册
rpc Register(RegisterRequest) returns (RegisterResponse);
// 验证访问令牌
rpc ValidateToken(ValidateTokenRequest) returns (ValidateTokenResponse);
}

2
router/index.go
View File

@@ -4,8 +4,10 @@ import "github.com/gin-gonic/gin"
var routerTest Test
var websocks Websocks
var user User
func Init(app *gin.Engine) {
routerTest.Init(app)
websocks.Init(app)
user.Init(app)
}

16
router/user.go Normal file
View File

@@ -0,0 +1,16 @@
package router
import (
"go-desk-service/api"
"github.com/gin-gonic/gin"
)
type User struct{}
var ApiUser api.User
func (*User) Init(app *gin.Engine) {
group := app.Group("/user")
group.POST("/login", ApiUser.Login)
}