- init

2 years ago · 0ed48aeae1
commit 0ed48aeae1
7 changed files with 183 additions and 0 deletions
--- a/.drone.yml
+++ b/.drone.yml
@ -0,0 +1,17 @@
+kind: pipeline
+type: docker
+name: clone
+
+steps:
+  - name: Test
+    image: golang:1.18
+    commands:
+      - go env -w GO111MODULE=on
+      - go env -w GOPROXY=https://goproxy.cn,direct
+      - go test -v ./...
+  - name: Benchmark
+    image: golang:1.18
+    commands:
+      - go env -w GO111MODULE=on
+      - go env -w GOPROXY=https://goproxy.cn,direct
+      - go test -bench=. -benchmem
--- a/.gitignore
+++ b/.gitignore
@ -0,0 +1,9 @@
+.env
+.git
+.svn
+.idea
+.vscode
+*.log
+goinit.sh
+gomod.sh
+/vendor/
--- a/go.mod
+++ b/go.mod
@ -0,0 +1,16 @@
+module go.dtapp.net/gomysql
+
+go 1.18
+
+require (
+	github.com/sirupsen/logrus v1.8.1
+	gorm.io/driver/mysql v1.3.3
+	gorm.io/gorm v1.23.5
+)
+
+require (
+	github.com/go-sql-driver/mysql v1.6.0 // indirect
+	github.com/jinzhu/inflection v1.0.0 // indirect
+	github.com/jinzhu/now v1.1.5 // indirect
+	golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a // indirect
+)
--- a/go.sum
+++ b/go.sum
@ -0,0 +1,23 @@
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
+github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
+github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
+github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
+github.com/jinzhu/now v1.1.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
+github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
+github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
+github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a h1:dGzPydgVsqGcTRVwiLJ1jVbufYwmzD3LfVPLKsKg+0k=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+gorm.io/driver/mysql v1.3.3 h1:jXG9ANrwBc4+bMvBcSl8zCfPBaVoPyBEBshA8dA93X8=
+gorm.io/driver/mysql v1.3.3/go.mod h1:ChK6AHbHgDCFZyJp0F+BmVGb06PSIoh9uVYKAlRbb2U=
+gorm.io/gorm v1.23.1/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk=
+gorm.io/gorm v1.23.5 h1:TnlF26wScKSvknUC/Rn8t0NLLM22fypYBlvj1+aH6dM=
+gorm.io/gorm v1.23.5/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk=
--- a/gomysql.go
+++ b/gomysql.go
@ -0,0 +1,108 @@
+package gomysql
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"log"
+	"os"
+	"path"
+	"time"
+)
+
+type App struct {
+	Db     *gorm.DB // 驱动
+	Dns    string   // 地址
+	Log    bool     // 日志
+	LogUrl string   // 日志路径
+}
+
+type writer struct{}
+
+// 日志路径
+var logsUrl = ""
+
+func (w writer) Printf(format string, args ...interface{}) {
+
+	// 判断路径
+
+	now := time.Now()
+	logFilePath := ""
+	if dir, err := os.Getwd(); err == nil {
+		logFilePath = dir + logsUrl
+	}
+	if err := os.MkdirAll(logFilePath, 0777); err != nil {
+		fmt.Println(err.Error())
+	}
+	logFileName := "gorm." + now.Format("2006-01-02") + ".log"
+
+	// 日志文件
+	fileName := path.Join(logFilePath, logFileName)
+	if _, err := os.Stat(fileName); err != nil {
+		if _, err := os.Create(fileName); err != nil {
+			fmt.Println(err.Error())
+		}
+	}
+	// 写入文件
+	src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
+	if err != nil {
+		fmt.Println("err", err)
+	}
+
+	// 实例化
+	l := logrus.New()
+
+	// 设置输出
+	l.Out = src
+
+	// 设置日志格式 JSONFormatter=json TextFormatter=text
+	l.SetFormatter(&logrus.JSONFormatter{
+		TimestampFormat: "2006-01-02 15:04:05",
+	})
+
+	l.Println(args...)
+}
+
+func (app *App) InitClient() {
+
+	log.Printf("mysql config：%+v\n", app)
+
+	// 判断路径
+	if app.LogUrl == "" {
+		logsUrl = "/logs/mysql"
+	}
+
+	var err error
+
+	if app.Log == true {
+		app.Db, err = gorm.Open(mysql.Open(app.Dns), &gorm.Config{
+			Logger: logger.New(
+				writer{},
+				logger.Config{
+					SlowThreshold:             time.Second, // 慢 SQL 阈值
+					LogLevel:                  logger.Info, // 日志级别
+					IgnoreRecordNotFoundError: true,        // 忽略ErrRecordNotFound（记录未找到）错误
+					Colorful:                  false,       // 禁用彩色打印
+				},
+			),
+		})
+	} else {
+		app.Db, err = gorm.Open(mysql.Open(app.Dns), &gorm.Config{})
+	}
+
+	if err != nil {
+		panic(fmt.Sprintf("数据库【mysql】连接失败：%v", err))
+	}
+
+	sqlDB, err := app.Db.DB()
+	if err != nil {
+		panic(fmt.Sprintf("数据库【mysql】连接服务器失败：%v", err))
+	}
+
+	sqlDB.SetMaxIdleConns(10)                   // 设置空闲连接池中连接的最大数量
+	sqlDB.SetMaxOpenConns(100)                  // 设置打开数据库连接的最大数量。
+	sqlDB.SetConnMaxLifetime(time.Second * 600) // 设置了连接可复用的最大时间。
+
+}
--- a/version.go
+++ b/version.go
@ -0,0 +1,3 @@
+package gomysql
+
+const Version = "1.0.0"
--- a/version_test.go
+++ b/version_test.go
@ -0,0 +1,7 @@
+package gomysql
+
+import "testing"
+
+func TestVersion(t *testing.T) {
+	t.Log(Version)
+}