feat: wrap message by fluent ui toast

internal/db/client.go

@@ -0,0 +1,61 @@
+package db
+
+import (
+	"context"
+	"github.com/loveuer/nf-disk/internal/opt"
+	"github.com/loveuer/nf-disk/internal/tool"
+	"io"
+
+	"gorm.io/gorm"
+)
+
+var (
+	Default *Client
+)
+
+type Client struct {
+	ctx       context.Context
+	cli       *gorm.DB
+	ttype     string
+	cfgSqlite *cfgSqlite
+}
+
+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.Debug {
+		session = session.Debug()
+	}
+
+	return session
+}
+
+func (c *Client) Close() {
+	d, _ := c.cli.DB()
+	d.Close()
+}
+
+// Dump
+// Only for sqlite with mem mode to dump data to bytes(io.Reader)
+func (c *Client) Dump() (reader io.ReadSeekCloser, ok bool) {
+	if c.ttype != "sqlite" {
+		return nil, false
+	}
+
+	if c.cfgSqlite.fsType != "mem" {
+		return nil, false
+	}
+
+	return c.cfgSqlite.memDump.Dump(), true
+}

internal/db/db_test.go

@@ -0,0 +1,44 @@
+package db
+
+import (
+	"context"
+	"io"
+	"os"
+	"testing"
+)
+
+func TestOpen(t *testing.T) {
+	myClient, err := New(context.TODO(), "sqlite::", OptSqliteByMem())
+	if err != nil {
+		t.Fatalf("TestOpen: New err = %v", err)
+	}
+
+	type Start struct {
+		Id   int     `json:"id" gorm:"column:id;primaryKey"`
+		Name string  `json:"name" gorm:"column:name"`
+		Dis  float64 `json:"dis" gorm:"column:dis"`
+	}
+
+	if err = myClient.Session().AutoMigrate(&Start{}); err != nil {
+		t.Fatalf("TestOpen: AutoMigrate err = %v", err)
+	}
+
+	if err = myClient.Session().Create(&Start{Name: "sun", Dis: 6631.76}).Error; err != nil {
+		t.Fatalf("TestOpen: Create err = %v", err)
+	}
+
+	if err = myClient.Session().Create(&Start{Name: "mar", Dis: 786.35}).Error; err != nil {
+		t.Fatalf("TestOpen: Create err = %v", err)
+	}
+
+	if reader, ok := myClient.Dump(); ok {
+		bs, err := io.ReadAll(reader)
+		if err != nil {
+			t.Fatalf("TestOpen: ReadAll err = %v", err)
+		}
+
+		os.WriteFile("dump.db", bs, 0644)
+	}
+
+	myClient.Close()
+}

internal/db/init.go

@@ -0,0 +1,54 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+func New(ctx context.Context, uri string, opts ...Option) (*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], cfgSqlite: &cfgSqlite{fsType: "file"}}
+	for _, f := range opts {
+		f(c)
+	}
+
+	var (
+		err error
+		dsn = strs[1]
+	)
+
+	switch strs[0] {
+	case "sqlite":
+		err = openSqlite(c, 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, opts ...Option) (err error) {
+	if Default, err = New(ctx, uri, opts...); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/db/option.go

@@ -0,0 +1,27 @@
+package db
+
+import (
+	_ "github.com/loveuer/go-sqlite3/embed"
+	"io"
+)
+
+type Option func(c *Client)
+
+func OptSqliteByUrl(address string) Option {
+	return func(c *Client) {
+		c.cfgSqlite.fsType = "url"
+	}
+}
+
+type SqliteMemDumper interface {
+	Dump() io.ReadSeekCloser
+}
+
+// 如果传 nil 则表示新生成一个 mem 的 sqlite
+// 如果传了一个合法的 reader 则会从这个 reader 初始化 database
+func OptSqliteByMem(reader io.ReadCloser) Option {
+	return func(c *Client) {
+		c.cfgSqlite.memReader = reader
+		c.cfgSqlite.fsType = "mem"
+	}
+}

internal/db/sqlite.go

@@ -0,0 +1,63 @@
+package db
+
+import (
+	"fmt"
+	"io"
+	"time"
+
+	_ "github.com/loveuer/go-sqlite3/embed"
+	"github.com/loveuer/go-sqlite3/vfs/memdb"
+	"github.com/loveuer/go-sqlite3/vfs/readervfs"
+	"github.com/ncruces/go-sqlite3/gormlite"
+	"github.com/psanford/httpreadat"
+	"gorm.io/gorm"
+)
+
+type cfgSqlite struct {
+	fsType    string // file, mem(bytes), url
+	memDump   *memdb.MemDB
+	memReader io.ReadCloser
+}
+
+func openSqlite(c *Client, dsn string) error {
+	var (
+		db  gorm.Dialector
+		err error
+	)
+
+	switch c.cfgSqlite.fsType {
+	case "file":
+		db = gormlite.Open("file:" + dsn)
+	case "url":
+		name := fmt.Sprintf("%d.db", time.Now().UnixNano())
+		readervfs.Create(name, httpreadat.New(dsn))
+		uri := fmt.Sprintf("file:%s?vfs=reader", name)
+		db = gormlite.Open(uri)
+	case "mem":
+		var (
+			bs   []byte
+			name = fmt.Sprintf("%d.db", time.Now().UnixNano())
+		)
+
+		if c.cfgSqlite.memReader == nil {
+			bs = make([]byte, 0)
+		} else {
+			if bs, err = io.ReadAll(c.cfgSqlite.memReader); err != nil {
+				return err
+			}
+		}
+
+		memDump := memdb.Create(name, bs)
+		c.cfgSqlite.memDump = memDump
+		uri := fmt.Sprintf("file:/%s?vfs=memdb", name)
+		db = gormlite.Open(uri)
+	default:
+		return fmt.Errorf("unsupported sqlite fs type: %s", c.cfgSqlite.fsType)
+	}
+
+	if c.cli, err = gorm.Open(db); err != nil {
+		return err
+	}
+
+	return nil
+}