fix: add the logic mysql

Author: coderi421

app/pkg/code/code_generated.go

@@ -3,7 +3,7 @@ package code
 //go:generate codeg -type=int
 const (
 	// ErrConnectDB - 500: Init db error.
-	ErrConnectDB int = iota + 100601
+	ErrConnectDB int = iota + 100101
 
 	// ErrConnectGRPC - 500: Connect to grpc error.
 	ErrConnectGRPC

app/pkg/code/db.go

@@ -38,4 +38,5 @@
 | ErrSmsSend | 100404 | 400 | Send sms error |
 | ErrCodeNotExist | 100405 | 400 | Sms code incorrect or expired |
 | ErrCodeInCorrect | 100406 | 400 | Sms code incorrect |
+| ErrCodeInvalidParam | 100407 | 400 | Invalid param |
 

app/pkg/code/error_code_generated.md

@@ -20,4 +20,7 @@ const (
 
 	// ErrCodeInCorrect - 400: Sms code incorrect.
 	ErrCodeInCorrect
+
+	// ErrCodeInvalidParam - 400: Invalid param.
+	ErrCodeInvalidParam
 )

app/pkg/code/user.go

@@ -0,0 +1,70 @@
+package options
+
+import (
+	"time"
+
+	"github.com/spf13/pflag"
+)
+
+type MySQLOptions struct {
+	Host     string `json:"host,omitempty" mapstructure:"host,omitempty"`
+	Port     string `json:"port,omitempty" mapstructure:"port,omitempty"`
+	Username string `json:"username,omitempty" mapstructure:"username,omitempty"`
+	Password string `json:"password,omitempty" mapstructure:"password,omitempty"`
+	Database string `json:"database" mapstructure:"database"`
+	LogLevel int    `json:"log-level,omitempty" mapstructure:"log-level,omitempty"`
+	//连接池会用到
+	MaxIdleConnections    int           `json:"max-idle-connections,omitempty" mapstructure:"max-idle-connections,omitempty"`
+	MaxConnectionLifetime time.Duration `json:"max-connection-lifetime,omitempty" mapstructure:"max-connection-lifetime,omitempty"`
+	MaxOpenConnections    int           `json:"max-open-connections,omitempty" mapstructure:"max-open-connections,omitempty"`
+}
+
+func NewMySQLOptions() *MySQLOptions {
+	return &MySQLOptions{
+		Host:                  "127.0.0.1",
+		Port:                  "3306",
+		Username:              "",
+		Password:              "",
+		Database:              "",
+		MaxIdleConnections:    100,
+		MaxOpenConnections:    100,
+		MaxConnectionLifetime: time.Duration(10) * time.Second,
+		LogLevel:              1,
+	}
+}
+
+// Validate verifies flags passed to ServerOptions.
+func (so *MySQLOptions) Validate() []error {
+	errs := []error{}
+	return errs
+}
+
+// AddFlags adds flags related to server storage for a specific APIServer to the specified FlagSet.
+func (mo *MySQLOptions) AddFlags(fs *pflag.FlagSet) {
+	fs.StringVar(&mo.Host, "mysql.host", mo.Host, ""+
+		"MySQL service host address. If left blank, the following related mysql options will be ignored.")
+
+	fs.StringVar(&mo.Port, "mysql.port", mo.Port, ""+
+		"MySQL service port")
+
+	fs.StringVar(&mo.Username, "mysql.username", mo.Username, ""+
+		"Username for access to mysql service.")
+
+	fs.StringVar(&mo.Password, "mysql.password", mo.Password, ""+
+		"Password for access to mysql, should be used pair with password.")
+
+	fs.StringVar(&mo.Database, "mysql.database", mo.Database, ""+
+		"Database name for the server to use.")
+
+	fs.IntVar(&mo.MaxIdleConnections, "mysql.max-idle-connections", mo.MaxOpenConnections, ""+
+		"Maximum idle connections allowed to connect to mysql.")
+
+	fs.IntVar(&mo.MaxOpenConnections, "mysql.max-open-connections", mo.MaxOpenConnections, ""+
+		"Maximum open connections allowed to connect to mysql.")
+
+	fs.DurationVar(&mo.MaxConnectionLifetime, "mysql.max-connection-life-time", mo.MaxConnectionLifetime, ""+
+		"Maximum connection life time allowed to connecto to mysql.")
+
+	fs.IntVar(&mo.LogLevel, "mysql.log-mode", mo.LogLevel, ""+
+		"Specify gorm log level.")
+}

app/pkg/options/mysql.go

@@ -11,10 +11,11 @@ var _ app.CliOptions = &Config{}
 
 func New() *Config {
 	return &Config{
-		Log:       log.NewOptions(),
-		Server:    options.NewServerOptions(),
-		Registry:  options.NewRegistryOptions(),
-		Telemetry: options.NewTelemetryOptions(),
+		Log:          log.NewOptions(),
+		Server:       options.NewServerOptions(),
+		Registry:     options.NewRegistryOptions(),
+		Telemetry:    options.NewTelemetryOptions(),
+		MySQLOptions: options.NewMySQLOptions(),
 	}
 }
 
@@ -25,7 +26,8 @@ type Config struct {
 	// 注册中心
 	Registry *options.RegistryOptions `json:"registry" mapstructure:"registry"`
 	//	链路追踪
-	Telemetry *options.TelemetryOptions `json:"telemetry" mapstructure:"telemetry"`
+	Telemetry    *options.TelemetryOptions `json:"telemetry" mapstructure:"telemetry"`
+	MySQLOptions *options.MySQLOptions     `json:"mysql" mapstructure:"mysql"`
 }
 
 // Flags implements app.CliOptions interface.Add flags to the specified FlagSet object.
@@ -35,6 +37,7 @@ func (c *Config) Flags() (fss cliflag.NamedFlagSets) {
 	c.Server.AddFlags(fss.FlagSet("server"))
 	c.Registry.AddFlags(fss.FlagSet("registry"))
 	c.Telemetry.AddFlags(fss.FlagSet("telemetry"))
+	c.MySQLOptions.AddFlags(fss.FlagSet("mysql"))
 	return fss
 }
 
@@ -45,5 +48,6 @@ func (c *Config) Validate() (errors []error) {
 	errors = append(errors, c.Server.Validate()...)
 	errors = append(errors, c.Registry.Validate()...)
 	errors = append(errors, c.Telemetry.Validate()...)
+	errors = append(errors, c.MySQLOptions.Validate()...)
 	return
 }

app/user/srv/config/config.go

@@ -0,0 +1,49 @@
+package db
+
+import (
+	"fmt"
+	"sync"
+
+	"github.com/CoderI421/gframework/app/pkg/code"
+	"github.com/CoderI421/gframework/app/pkg/options"
+	"github.com/CoderI421/gframework/pkg/errors"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+var (
+	dbFactory *gorm.DB
+	once      sync.Once
+)
+
+func GetDBFactoryOr(mysqlOpts *options.MySQLOptions) (*gorm.DB, error) {
+	if mysqlOpts == nil && dbFactory == nil {
+		return nil, errors.WithCode(code.ErrConnectDB, "failed to get mysql store factory")
+
+	}
+	var err error
+	once.Do(func() {
+		dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+			mysqlOpts.Username,
+			mysqlOpts.Password,
+			mysqlOpts.Host,
+			mysqlOpts.Port,
+			mysqlOpts.Database)
+		dbFactory, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+		if err != nil {
+			return
+		}
+		sqlDB, _ := dbFactory.DB()
+		//允许连接多少个mysql
+		sqlDB.SetMaxOpenConns(mysqlOpts.MaxOpenConnections)
+		//允许最大的空闲的连接数
+		sqlDB.SetMaxIdleConns(mysqlOpts.MaxIdleConnections)
+		//重用连接的最大时长
+		sqlDB.SetConnMaxLifetime(mysqlOpts.MaxConnectionLifetime)
+
+	})
+	if dbFactory == nil || err != nil {
+		return nil, errors.WithCode(code.ErrConnectDB, "failed to get mysql store factory")
+	}
+	return dbFactory, nil
+}

app/user/srv/data/v1/db/mysql.go

@@ -3,14 +3,16 @@ package srv
 import (
 	"fmt"
 
+	"github.com/CoderI421/gframework/app/user/srv/data/v1/db"
+	"github.com/CoderI421/gframework/pkg/log"
+
 	"github.com/CoderI421/gframework/gmicro/core/trace"
 
 	upbv1 "github.com/CoderI421/gframework/api/user/v1"
 	srv1 "github.com/CoderI421/gframework/app/user/srv/service/v1"
 
 	"github.com/CoderI421/gframework/app/user/srv/config"
 	"github.com/CoderI421/gframework/app/user/srv/controller/user"
-	"github.com/CoderI421/gframework/app/user/srv/data/v1/mock"
 	"github.com/CoderI421/gframework/gmicro/server/rpcserver"
 )
 
@@ -24,7 +26,13 @@ func NewUserRPCServer(cfg *config.Config) (*rpcserver.Server, error) {
 		Batcher:  cfg.Telemetry.Batcher,
 	})
 
-	data := mock.NewUsers()
+	//data := mock.NewUsers()
+
+	gormDB, err := db.GetDBFactoryOr(cfg.MySQLOptions)
+	if err != nil {
+		log.Fatal(err.Error())
+	}
+	data := db.NewUser(gormDB)
 	srv := srv1.NewUserService(data)
 	userver := user.NewUserServer(srv)
 

app/user/srv/rpc.go

@@ -27,4 +27,15 @@ telemetry:
   Name: user-srv
   Endpoint: http://127.0.0.1:14268/api/traces
   Sampler: 1.0
-  Bathcer: jaeger
+  Bathcer: jaeger
+
+mysql:
+  host: "127.0.0.1" # MySQL 机器 ip，默认 127.0.0.1
+  port: "3306"
+  password: "123456" # MySQL 用户密码
+  username: "root" # MySQL 用户名(建议授权最小权限集)
+  database: "shopping_go" # iam 系统所用的数据库名
+  max-idle-connections: 100 # MySQL 最大空闲连接数，默认 100
+  max-open-connections: 100 # MySQL 最大打开的连接数，默认 100
+  max-connection-life-time: 10s # 空闲连接最大存活时间，默认 10s
+  log-level: 4 # GORM log level, 1: silent, 2:error, 3:warn, 4:info

configs/user/srv.yaml

@@ -58,6 +58,7 @@ require (
 	github.com/go-logr/logr v1.2.4 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/go-playground/form/v4 v4.2.0 // indirect
+	github.com/go-sql-driver/mysql v1.7.0 // indirect
 	github.com/goccy/go-json v0.10.0 // indirect
 	github.com/golang/protobuf v1.5.3 // indirect
 	github.com/gorilla/mux v1.8.0 // indirect
@@ -110,4 +111,5 @@ require (
 	golang.org/x/text v0.9.0 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
+	gorm.io/driver/mysql v1.5.1 // indirect
 )