🎉 完成基本的演示和样例

internal/store/db/client.go

@@ -0,0 +1,45 @@
+package db
+
+import (
+	"context"
+	"uauth/internal/opt"
+	"uauth/internal/tool"
+
+	"gorm.io/gorm"
+)
+
+var (
+	Default *Client
+)
+
+type Client struct {
+	ctx   context.Context
+	cli   *gorm.DB
+	ttype string
+}
+
+func (c *Client) Type() string {
+	return c.ttype
+}
+
+func (c *Client) Session(ctxs ...context.Context) *gorm.DB {
+	var ctx context.Context
+	if len(ctxs) > 0 && ctxs[0] != nil {
+		ctx = ctxs[0]
+	} else {
+		ctx = tool.Timeout(30)
+	}
+
+	session := c.cli.Session(&gorm.Session{Context: ctx})
+
+	if opt.Cfg.Debug {
+		session = session.Debug()
+	}
+
+	return session
+}
+
+func (c *Client) Close() {
+	d, _ := c.cli.DB()
+	d.Close()
+}

internal/store/db/db_test.go

@@ -0,0 +1,9 @@
+package db
+
+import (
+	"testing"
+)
+
+func TestOpen(t *testing.T) {
+
+}

internal/store/db/init.go

@@ -0,0 +1,52 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func New(ctx context.Context, uri string) (*Client, error) {
+	strs := strings.Split(uri, "::")
+
+	if len(strs) != 2 {
+		return nil, fmt.Errorf("db.Init: opt db uri invalid: %s", uri)
+	}
+
+	c := &Client{ttype: strs[0]}
+
+	var (
+		err error
+		dsn = strs[1]
+	)
+
+	switch strs[0] {
+	case "sqlite":
+		c.cli, err = gorm.Open(sqlite.Open(dsn))
+	case "mysql":
+		c.cli, err = gorm.Open(mysql.Open(dsn))
+	case "postgres":
+		c.cli, err = gorm.Open(postgres.Open(dsn))
+	default:
+		return nil, fmt.Errorf("db type only support: [sqlite, mysql, postgres], unsupported db type: %s", strs[0])
+	}
+
+	if err != nil {
+		return nil, fmt.Errorf("db.Init: open %s with dsn:%s, err: %w", strs[0], dsn, err)
+	}
+
+	return c, nil
+}
+
+func Init(ctx context.Context, uri string) (err error) {
+	if Default, err = New(ctx, uri); err != nil {
+		return err
+	}
+
+	return nil
+}