commit

Author: dxchub

Makefile

@@ -0,0 +1,35 @@
+GOHOSTOS:=$(shell go env GOHOSTOS)
+GOPATH:=$(shell go env GOPATH)
+VERSION=$(shell git describe --tags --always)
+BUILDTIME=$(shell date "+%Y-%m-%d %H:%M:%S")
+GITCOMMITID=$(shell git rev-parse HEAD)
+
+ifeq ($(GOHOSTOS), windows)
+	#the `find.exe` is different from `find` in bash/shell.
+	#to see https://docs.microsoft.com/en-us/windows-server/administration/windows-commands/find.
+	#changed to use git-bash.exe to run find cli or other cli friendly, caused of every developer has a Git.
+	#Git_Bash= $(subst cmd\,bin\bash.exe,$(dir $(shell where git)))
+	Git_Bash=$(subst \,/,$(subst cmd\,bin\bash.exe,$(dir $(shell where git))))
+	INTERNAL_PROTO_FILES=$(shell $(Git_Bash) -c "find internal -name *.proto")
+	API_PROTO_FILES=$(shell $(Git_Bash) -c "find api -name *.proto")
+else
+	INTERNAL_PROTO_FILES=$(shell find internal -name *.proto)
+	API_PROTO_FILES=$(shell find api -name *.proto)
+endif
+
+.PHONY: init
+# init env
+init:
+	go install google.golang.org/protobuf/cmd/protoc-gen-go@latest
+	go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@latest
+	go install github.com/go-kratos/kratos/cmd/kratos/v2@latest
+	go install github.com/go-kratos/kratos/cmd/protoc-gen-go-http/v2@latest
+	go install github.com/google/gnostic/cmd/protoc-gen-openapi@latest
+
+.PHONT: errors
+errors:
+	protoc --proto_path=./api \
+	       --proto_path=./third_party \
+		   --go_out=paths=import:. \
+		   --go-errors_out=paths=import:. \
+		   $(API_PROTO_FILES)

api/xerr/proto/xerr.proto

@@ -0,0 +1,18 @@
+syntax = "proto3";
+
+package xerr.v1;
+
+import "errors/errors.proto";
+
+// 多语言特定包名，用于源代码引用
+option go_package = "./api/xerr;xerr";
+
+enum Reason {
+  option (errors.default_code) = 500;
+
+  // 鉴权错误包括:token错误、token过期、token无效、无权限访问等
+  AUTH = 0 [(errors.code) = 401];
+
+  // 数据操作错误包括:数据不存在、数据已存在、数据已删除等
+  DB_OPTION = 4 [(errors.code) = 403];
+}

api/xerr/xerr.pb.go

@@ -0,0 +1,33 @@
+package service
+
+import (
+	"fmt"
+	"time"
+)
+
+type Info struct {
+	Id       string
+	Name     string
+	Version  string
+	Metadata map[string]string
+}
+
+func NewServiceInfo(name, version, id string) Info {
+	if id == "" {
+		id = fmt.Sprint(time.Now().Unix())
+	}
+	return Info{
+		Name:     name,
+		Version:  version,
+		Id:       id,
+		Metadata: map[string]string{},
+	}
+}
+
+func (this *Info) GetInstanceId() string {
+	return this.Name + "." + this.Id
+}
+
+func (this *Info) SetMataData(k, v string) {
+	this.Metadata[k] = v
+}

api/xerr/xerr_errors.pb.go

@@ -0,0 +1,32 @@
+package service
+
+import (
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/exporters/jaeger"
+	"go.opentelemetry.io/otel/sdk/resource"
+	traceSdk "go.opentelemetry.io/otel/sdk/trace"
+	semConv "go.opentelemetry.io/otel/semconv/v1.4.0"
+)
+
+func NewTracerProvider(endpoint, env string, info *Info) error {
+	exp, err := jaeger.New(jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(endpoint)))
+	if err != nil {
+		return err
+	}
+
+	tp := traceSdk.NewTracerProvider(
+		traceSdk.WithSampler(traceSdk.ParentBased(traceSdk.TraceIDRatioBased(1.0))),
+		traceSdk.WithBatcher(exp),
+		traceSdk.WithResource(resource.NewSchemaless(
+			semConv.ServiceNameKey.String(info.Name),
+			semConv.ServiceVersionKey.String(info.Version),
+			semConv.ServiceInstanceIDKey.String(info.Id),
+			attribute.String("env", env),
+		)),
+	)
+
+	otel.SetTracerProvider(tp)
+
+	return nil
+}

pkg/service/info.go

@@ -0,0 +1 @@
+package service

pkg/service/tracer.go

@@ -0,0 +1,16 @@
+package share
+
+import (
+	"context"
+	"fmt"
+	"github.com/aswancen/libs/pkg/xmw"
+)
+
+// GetRpcToken 获取Token
+func GetRpcToken(ctx context.Context) (*xmw.AuthInfo, error) {
+	value := ctx.Value(xmw.CtxAuthKey)
+	if token, IsOk := value.(*xmw.AuthInfo); IsOk {
+		return token, nil
+	}
+	return nil, fmt.Errorf("token not found")
+}

pkg/service/writer.go

@@ -0,0 +1,31 @@
+package share
+
+import (
+	"time"
+
+	"github.com/sony/sonyflake"
+)
+
+type IdGenerator struct {
+	flake *sonyflake.Sonyflake
+}
+
+func NewIdGenerator(baseTime string) *IdGenerator {
+	loc, _ := time.LoadLocation("Asia/Shanghai")
+	dt, err := time.ParseInLocation("2006-01-02", baseTime, loc)
+	if err != nil {
+		panic(err)
+	}
+	settings := sonyflake.Settings{
+		StartTime:      dt,
+		MachineID:      nil,
+		CheckMachineID: nil,
+	}
+	flake := sonyflake.NewSonyflake(settings)
+	return &IdGenerator{flake: flake}
+}
+
+func (this *IdGenerator) GetId() int64 {
+	nextID, _ := this.flake.NextID()
+	return int64(nextID)
+}

pkg/share/auth.go

@@ -0,0 +1,16 @@
+package share
+
+import "gorm.io/gorm"
+
+// Paginate 分页器(offset从1开始)
+func Paginate(page int, pageSize int) func(db *gorm.DB) *gorm.DB {
+	if pageSize > 10000 {
+		pageSize = 10000
+	}
+	if pageSize < 1 {
+		pageSize = 10
+	}
+	return func(db *gorm.DB) *gorm.DB {
+		return db.Offset((page - 1) * pageSize).Limit(pageSize)
+	}
+}