feat: add oliver elastic support; update db(sqlite)

internal/database/db/client.go

@@ -2,25 +2,29 @@ package db
 
 import (
 	"context"
-	"gorm.io/gorm"
+	"io"
 	"ultone/internal/opt"
 	"ultone/internal/tool"
+
+	"gorm.io/gorm"
 )
 
 var (
-	cli = &client{}
+	Default *Client
 )
 
-type client struct {
-	cli   *gorm.DB
-	ttype string
+type Client struct {
+	ctx       context.Context
+	cli       *gorm.DB
+	ttype     string
+	cfgSqlite *cfgSqlite
 }
 
-func Type() string {
-	return cli.ttype
+func (c *Client) Type() string {
+	return c.ttype
 }
 
-func New(ctxs ...context.Context) *gorm.DB {
+func (c *Client) Session(ctxs ...context.Context) *gorm.DB {
 	var ctx context.Context
 	if len(ctxs) > 0 && ctxs[0] != nil {
 		ctx = ctxs[0]
@@ -28,7 +32,7 @@ func New(ctxs ...context.Context) *gorm.DB {
 		ctx = tool.Timeout(30)
 	}
 
-	session := cli.cli.Session(&gorm.Session{Context: ctx})
+	session := c.cli.Session(&gorm.Session{Context: ctx})
 
 	if opt.Debug {
 		session = session.Debug()
@@ -36,3 +40,22 @@ func New(ctxs ...context.Context) *gorm.DB {
 
 	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
+}