wip: 登录和认证

pkg/database/db/db.go

@@ -0,0 +1,49 @@
+package db
+
+import (
+	"context"
+	"gorm.io/gorm"
+)
+
+type Config struct {
+	Debug  bool
+	DryRun bool
+}
+
+type DB interface {
+	Session(ctx context.Context, configs ...Config) *gorm.DB
+}
+
+type db struct {
+	tx *gorm.DB
+}
+
+var (
+	Default DB
+)
+
+func (db *db) Session(ctx context.Context, configs ...Config) *gorm.DB {
+	var (
+		sc      = &gorm.Session{Context: ctx}
+		session *gorm.DB
+	)
+
+	if len(configs) == 0 {
+		session = db.tx.Session(sc)
+		return session
+	}
+
+	cfg := configs[0]
+
+	if cfg.DryRun {
+		sc.DryRun = true
+	}
+
+	session = db.tx.Session(sc)
+
+	if cfg.Debug {
+		session = session.Debug()
+	}
+
+	return session
+}

pkg/database/db/new.go

@@ -0,0 +1,48 @@
+package db
+
+import (
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+var defaultSqlite = "data.db"
+
+func New(opts ...OptionFn) (DB, error) {
+	var (
+		err  error
+		conf = &config{
+			sqlite: &defaultSqlite,
+		}
+		tx *gorm.DB
+	)
+
+	for _, opt := range opts {
+		opt(conf)
+	}
+
+	if conf.mysql != nil {
+		tx, err = gorm.Open(mysql.Open(*conf.mysql))
+		goto CHECK
+	}
+
+	if conf.pg != nil {
+		tx, err = gorm.Open(postgres.Open(*conf.pg))
+		goto CHECK
+	}
+
+	tx, err = gorm.Open(sqlite.Open(*conf.sqlite))
+
+CHECK:
+	if err != nil {
+		return nil, err
+	}
+
+	return &db{tx: tx}, nil
+}
+
+func Init(opts ...OptionFn) (err error) {
+	Default, err = New(opts...)
+	return err
+}

pkg/database/db/new_test.go

@@ -0,0 +1,25 @@
+package db
+
+import (
+	"testing"
+)
+
+func TestNew(t *testing.T) {
+	//mdb, err := New(WithMysql("127.0.0.1", 3306, "root", "MyPassw0rd", "mydb"))
+	//if err != nil {
+	//	t.Fatal(err)
+	//}
+	//
+	//type User struct {
+	//	Id       uint64 `gorm:"primaryKey"`
+	//	Username string `gorm:"unique"`
+	//}
+	//
+	//if err = mdb.Session(t.Context()).AutoMigrate(&User{}); err != nil {
+	//	t.Fatal(err)
+	//}
+	//
+	//if err = mdb.Session(t.Context()).Create(&User{Username: "zyp"}).Error; err != nil {
+	//	t.Fatal(err)
+	//}
+}

pkg/database/db/option.go

@@ -0,0 +1,45 @@
+package db
+
+import (
+	"context"
+	"fmt"
+)
+
+type config struct {
+	ctx    context.Context
+	mysql  *string
+	pg     *string
+	sqlite *string
+}
+
+type OptionFn func(*config)
+
+func WithCtx(ctx context.Context) OptionFn {
+	return func(c *config) {
+		if ctx != nil {
+			c.ctx = ctx
+		}
+	}
+}
+
+func WithMysql(host string, port int, user string, password string, database string) OptionFn {
+	return func(c *config) {
+		dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, password, host, port, database)
+		c.mysql = &dsn
+	}
+}
+
+func WithPg(host string, port int, user string, password string, database string) OptionFn {
+	return func(c *config) {
+		dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai", host, user, password, database, port)
+		c.pg = &dsn
+	}
+}
+
+func WithSqlite(path string) OptionFn {
+	return func(c *config) {
+		if path != "" {
+			c.sqlite = &path
+		}
+	}
+}