From 58139791f397443425e1add7662594e663d67d0a Mon Sep 17 00:00:00 2001 From: zhaoxiaorong Date: Fri, 7 Feb 2025 17:08:55 +0800 Subject: [PATCH] errcode --- errcode/errcode.go | 13 +++++---- go.sum | 3 ++ infra/fts.go | 44 +++++++++++++++++++++++++++++ infra/handler.go | 7 +++++ infra/new.go | 16 +++++++++++ infra/parser.go | 20 ++++++++++++++ infra/response.go | 69 ++++++++++++++++++++++++++++++++++++++++++++++ infra/service.go | 54 ++++++++++++++++++++++++++++++++++++ 8 files changed, 220 insertions(+), 6 deletions(-) create mode 100644 infra/fts.go create mode 100644 infra/handler.go create mode 100644 infra/new.go create mode 100644 infra/parser.go create mode 100644 infra/response.go create mode 100644 infra/service.go diff --git a/errcode/errcode.go b/errcode/errcode.go index 6500259..fbc5649 100644 --- a/errcode/errcode.go +++ b/errcode/errcode.go @@ -1,7 +1,8 @@ package errcode import ( - "github.com/gofiber/fiber/v2" + "google.golang.org/grpc/codes" + "google.golang.org/grpc/status" ) // header error code ,start:100 @@ -88,14 +89,14 @@ var ( ) func NewError(code int, msg string) error { - return fiber.NewError(code, msg) + return status.New(codes.Code(code), msg).Err() } // custom error,status code:500 -func ErrFatal(msg string) error { - return fiber.NewError(500, msg) +func ErrFatal(code int, msg string) error { + return status.New(codes.Code(code), msg).Err() } -func ErrNotFound(msg string) error { - return fiber.NewError(404, msg+" Not Found") +func ErrNotFound(code int, msg string) error { + return status.New(codes.Code(code), msg).Err() } diff --git a/go.sum b/go.sum index 99d68b3..5f4c73a 100644 --- a/go.sum +++ b/go.sum @@ -673,12 +673,15 @@ google.golang.org/genproto v0.0.0-20200804131852-c06518451d9c/go.mod h1:FWY/as6D google.golang.org/genproto v0.0.0-20200825200019-8632dd797987/go.mod h1:FWY/as6DDZQgahTzZj3fqbO1CbirC29ZNUFHwi0/+no= google.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d h1:VBu5YqKPv6XiJ199exd8Br+Aetz+o08F+PLMnwJQHAY= google.golang.org/genproto v0.0.0-20230822172742-b8732ec3820d/go.mod h1:yZTlhN0tQnXo3h00fuXNCxJdLdIdnVFVBaRJ5LWBbw4= +google.golang.org/genproto v0.0.0-20231012201019-e917dd12ba7a h1:fwgW9j3vHirt4ObdHoYNwuO24BEZjSzbh+zPaNWoiY8= google.golang.org/genproto v0.0.0-20231012201019-e917dd12ba7a/go.mod h1:EMfReVxb80Dq1hhioy0sOsY9jCE46YDgHlJ7fWVUWRE= google.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d h1:DoPTO70H+bcDXcd39vOqb2viZxgqeBeSGtZ55yZU4/Q= google.golang.org/genproto/googleapis/api v0.0.0-20230822172742-b8732ec3820d/go.mod h1:KjSP20unUpOx5kyQUFa7k4OJg0qeJ7DEZflGDu2p6Bk= +google.golang.org/genproto/googleapis/api v0.0.0-20231002182017-d307bd883b97 h1:W18sezcAYs+3tDZX4F80yctqa12jcP1PUS2gQu1zTPU= google.golang.org/genproto/googleapis/api v0.0.0-20231002182017-d307bd883b97/go.mod h1:iargEX0SFPm3xcfMI0d1domjg0ZF4Aa0p2awqyxhvF0= google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d h1:uvYuEyMHKNt+lT4K3bN6fGswmK8qSvcreM3BwjDh+y4= google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d/go.mod h1:+Bk1OCOj40wS2hwAMA+aCW9ypzm63QTBBHp6lQ3p+9M= +google.golang.org/genproto/googleapis/rpc v0.0.0-20231016165738-49dd2c1f3d0b h1:ZlWIi1wSK56/8hn4QcBp/j9M7Gt3U/3hZw3mC7vDICo= google.golang.org/genproto/googleapis/rpc v0.0.0-20231016165738-49dd2c1f3d0b/go.mod h1:swOH3j0KzcDDgGUWr+SNpyTen5YrXjS3eyPzFYKc6lc= google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38= diff --git a/infra/fts.go b/infra/fts.go new file mode 100644 index 0000000..4f0ea96 --- /dev/null +++ b/infra/fts.go @@ -0,0 +1,44 @@ +package infra + +import ( + "bytes" + "io" + "net/http" + + "git.apinb.com/bsm-sdk/core/errcode" +) + +var FTS fts + +type fts struct { + Endpoint string +} + +func (o *fts) SaveTo(buf *bytes.Buffer, haeder map[string]string) (string, error) { + // 创建一个 POST 请求 + req, err := http.NewRequest("POST", o.Endpoint, buf) + if err != nil { + return "", errcode.NewError(110, "Request to the URL") + } + + // 设置请求头为二进制流 + req.Header.Set("Content-Type", "application/octet-stream") + for key, val := range haeder { + req.Header.Set(key, val) + } + + // 发送请求 + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return "", errcode.NewError(111, "Client Do") + } + defer resp.Body.Close() + + // 读取响应体 + respBody, err := io.ReadAll(resp.Body) + if err != nil { + return "", errcode.NewError(112, "Read response body") + } + return string(respBody), nil +} diff --git a/infra/handler.go b/infra/handler.go new file mode 100644 index 0000000..af74385 --- /dev/null +++ b/infra/handler.go @@ -0,0 +1,7 @@ +package infra + +import "github.com/gofiber/fiber/v2" + +func Health(ctx *fiber.Ctx) error { + return ctx.SendStatus(200) +} diff --git a/infra/new.go b/infra/new.go new file mode 100644 index 0000000..bf16509 --- /dev/null +++ b/infra/new.go @@ -0,0 +1,16 @@ +package infra + +import ( + "git.apinb.com/bsm-sdk/core/cache/redis" + "gorm.io/gorm" +) + +var ( + DB *gorm.DB + RedisCache *redis.RedisClient +) + +func New(db *gorm.DB, redisCache *redis.RedisClient) { + DB = db + RedisCache = redisCache +} diff --git a/infra/parser.go b/infra/parser.go new file mode 100644 index 0000000..220a798 --- /dev/null +++ b/infra/parser.go @@ -0,0 +1,20 @@ +package infra + +import ( + "github.com/gofiber/fiber/v2" +) + +type Auth struct { + CustomerID int64 // 企业编号 + CustomerCode string // 企业标识 + UserID int64 // 用户编号 + UserIdentity string // 用户标识 +} + +func ParserCtx(ctx *fiber.Ctx) *Auth { + customerID := ctx.Locals("customerID").(int64) + customerCode := ctx.Locals("customerCode").(string) + userID := ctx.Locals("userID").(int64) + userIdentity := ctx.Locals("userIdentity").(string) + return &Auth{CustomerID: customerID, CustomerCode: customerCode, UserID: userID, UserIdentity: userIdentity} +} diff --git a/infra/response.go b/infra/response.go new file mode 100644 index 0000000..cfbd47e --- /dev/null +++ b/infra/response.go @@ -0,0 +1,69 @@ +package infra + +import ( + "net/http" + "time" + + "git.apinb.com/bsm-sdk/core/vars" + "github.com/gofiber/fiber/v2" +) + +var Response Reply + +type Reply struct { + Ret int `json:"ret"` + Msg string `json:"msg"` + Data interface{} `json:"data"` +} + +type ReplyIdent struct { + ID uint `json:"id"` + Identiy string `json:"identity"` +} + +func (r *Reply) Error(ctx *fiber.Ctx, err error) error { + // Status code defaults to 500 + reply := Reply{ + Ret: http.StatusInternalServerError, + } + if e, ok := err.(*fiber.Error); ok { + reply.Ret = e.Code + } + + // 获取执行信息 + ExcuteTime(ctx) + + reply.Msg = err.Error() + + // Send error + return ctx.Status(fiber.StatusInternalServerError).JSON(reply) +} +func (r *Reply) Success(ctx *fiber.Ctx, data interface{}) error { + reply := Reply{ + Ret: http.StatusOK, + } + + if data == nil { + reply.Data = "" + } else { + reply.Data = data + } + + // 获取执行信息 + ExcuteTime(ctx) + + return ctx.Status(fiber.StatusOK).JSON(reply) +} + +func ExcuteTime(ctx *fiber.Ctx) error { + // 从context中获取开始时间 + startTime, ok := ctx.Locals("startTime").(time.Time) + if !ok { + // 如果转换失败，则默认为当前时间 + startTime = time.Now() + } + + exectime := time.Since(startTime) + ctx.Response().Header.Add("Execute-Node", vars.HostName+","+exectime.String()) + return nil +} diff --git a/infra/service.go b/infra/service.go new file mode 100644 index 0000000..59bc091 --- /dev/null +++ b/infra/service.go @@ -0,0 +1,54 @@ +package infra + +import ( + "context" + "log" + "time" + + "git.apinb.com/bsm-sdk/core/print" + "git.apinb.com/bsm-sdk/core/utils" + clientv3 "go.etcd.io/etcd/client/v3" +) + +var ( + Service service + RootPrefix string = "/services/" +) + +type service struct{} + +func (s *service) Register(cli *clientv3.Client, serviceName string, port int) error { + lease := clientv3.NewLease(cli) + grantResp, err := lease.Grant(context.TODO(), 5) + if err != nil { + return err + } + + serviceAddr := utils.GetLocationIP() + ":" + utils.Int2String(port) + + key := RootPrefix + serviceName + "/" + utils.Int642String(time.Now().UnixNano()) + _, err = cli.KV.Put(context.TODO(), key, serviceAddr, clientv3.WithLease(grantResp.ID)) + if err != nil { + return err + } + + keepAliveChan, err := lease.KeepAlive(context.TODO(), grantResp.ID) + if err != nil { + return err + } + + print.Info("[Traingo Register] Service Key: %s", key) + print.Info("[Traingo Register] Service Val: %s", serviceAddr) + + print.Success("[Traingo Register] Service Register Complete.") + + go func() { + for keepAliveResp := range keepAliveChan { + if keepAliveResp == nil { + log.Println("LeaseKeepAlive", "Failed") + return + } + } + }() + return nil +}