[feature]:支持Sqlite数据库 (#1467)

* add sqlite3 support

* 修改gorm.io/driver/sqlite 为 github.com/glebarez/sqlite 适配windows用户无cgo环境

---------

Co-authored-by: Kafumio <linzehong1_2020@qq.com>
Co-authored-by: Kafumio <73083337+Kafumio@users.noreply.github.com>
Co-authored-by: sliverhorn <sliver_horn@qq.com>

server/service/system/sys_auto_code_interface.go

@@ -19,13 +19,14 @@ func (autoCodeService *AutoCodeService) Database(businessDB string) Database {
 			return AutoCodeMysql
 		case "pgsql":
 			return AutoCodePgsql
+		case "sqlite":
+			return AutoCodeSqlite
 		default:
 			return AutoCodeMysql
 		}
 	} else {
 		for _, info := range global.GVA_CONFIG.DBList {
 			if info.AliasName == businessDB {
-
 				switch info.Type {
 				case "mysql":
 					return AutoCodeMysql
@@ -35,6 +36,8 @@ func (autoCodeService *AutoCodeService) Database(businessDB string) Database {
 					return AutoCodePgsql
 				case "oracle":
 					return AutoCodeOracle
+				case "sqlite":
+					return AutoCodeSqlite
 				default:
 					return AutoCodeMysql
 				}

server/service/system/sys_auto_code_oracle.go

@@ -15,7 +15,6 @@ type autoCodeOracle struct{}
 func (s *autoCodeOracle) GetDB(businessDB string) (data []response.Db, err error) {
 	var entities []response.Db
 	sql := `SELECT lower(username) AS "database" FROM all_users`
-
 	err = global.GVA_DBList[businessDB].Raw(sql).Scan(&entities).Error
 	return entities, err
 }

server/service/system/sys_auto_code_sqlite.go

@@ -0,0 +1,82 @@
+package system
+
+import (
+	"fmt"
+	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"github.com/flipped-aurora/gin-vue-admin/server/model/system/response"
+	"path/filepath"
+	"strings"
+)
+
+var AutoCodeSqlite = new(autoCodeSqlite)
+
+type autoCodeSqlite struct{}
+
+// GetDB 获取数据库的所有数据库名
+// Author [piexlmax](https://github.com/piexlmax)
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (a *autoCodeSqlite) GetDB(businessDB string) (data []response.Db, err error) {
+	var entities []response.Db
+	sql := "PRAGMA database_list;"
+	var databaseList []struct {
+		File string `gorm:"column:file"`
+	}
+	if businessDB == "" {
+		err = global.GVA_DB.Raw(sql).Find(&databaseList).Error
+	} else {
+		err = global.GVA_DBList[businessDB].Raw(sql).Find(&databaseList).Error
+	}
+	for _, database := range databaseList {
+		if database.File != "" {
+			fileName := filepath.Base(database.File)
+			fileExt := filepath.Ext(fileName)
+			fileNameWithoutExt := strings.TrimSuffix(fileName, fileExt)
+
+			entities = append(entities, response.Db{fileNameWithoutExt})
+		}
+	}
+	// entities = append(entities, response.Db{global.GVA_CONFIG.Sqlite.Dbname})
+	return entities, err
+}
+
+// GetTables 获取数据库的所有表名
+// Author [piexlmax](https://github.com/piexlmax)
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (a *autoCodeSqlite) GetTables(businessDB string, dbName string) (data []response.Table, err error) {
+	var entities []response.Table
+	sql := `SELECT name FROM sqlite_master WHERE type='table'`
+	tabelNames := []string{}
+	if businessDB == "" {
+		err = global.GVA_DB.Raw(sql).Find(&tabelNames).Error
+	} else {
+		err = global.GVA_DBList[businessDB].Raw(sql).Find(&tabelNames).Error
+	}
+	for _, tabelName := range tabelNames {
+		entities = append(entities, response.Table{tabelName})
+	}
+	return entities, err
+}
+
+// GetColumn 获取指定数据表的所有字段名,类型值等
+// Author [piexlmax](https://github.com/piexlmax)
+// Author [SliverHorn](https://github.com/SliverHorn)
+func (a *autoCodeSqlite) GetColumn(businessDB string, tableName string, dbName string) (data []response.Column, err error) {
+	var entities []response.Column
+	sql := fmt.Sprintf("PRAGMA table_info(%s);", tableName)
+	var columnInfos []struct {
+		Name string `gorm:"column:name"`
+		Type string `gorm:"column:type"`
+	}
+	if businessDB == "" {
+		err = global.GVA_DB.Raw(sql).Scan(&columnInfos).Error
+	} else {
+		err = global.GVA_DBList[businessDB].Raw(sql).Scan(&columnInfos).Error
+	}
+	for _, columnInfo := range columnInfos {
+		entities = append(entities, response.Column{
+			ColumnName: columnInfo.Name,
+			DataType:   columnInfo.Type,
+		})
+	}
+	return entities, err
+}

server/service/system/sys_initdb.go

@@ -14,6 +14,7 @@ import (
 const (
 	Mysql           = "mysql"
 	Pgsql           = "pgsql"
+	Sqlite          = "sqlite"
 	InitSuccess     = "\n[%v] --> 初始数据成功!\n"
 	InitDataExist   = "\n[%v] --> %v 的初始数据已存在!\n"
 	InitDataFailed  = "\n[%v] --> %v 初始数据失败! \nerr: %+v\n"
@@ -102,6 +103,9 @@ func (initDBService *InitDBService) InitDB(conf request.InitDB) (err error) {
 	case "pgsql":
 		initHandler = NewPgsqlInitHandler()
 		ctx = context.WithValue(ctx, "dbtype", "pgsql")
+	case "sqlite":
+		initHandler = NewSqliteInitHandler()
+		ctx = context.WithValue(ctx, "dbtype", "sqlite")
 	default:
 		initHandler = NewMysqlInitHandler()
 		ctx = context.WithValue(ctx, "dbtype", "mysql")
@@ -161,7 +165,6 @@ func createTables(ctx context.Context, inits initSlice) error {
 		} else {
 			next = n
 		}
-
 	}
 	return nil
 }

server/service/system/sys_initdb_sqlite.go

@@ -0,0 +1,86 @@
+package system
+
+import (
+	"context"
+	"errors"
+	"github.com/flipped-aurora/gin-vue-admin/server/config"
+	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
+	"github.com/flipped-aurora/gin-vue-admin/server/utils"
+	"github.com/glebarez/sqlite"
+	"github.com/gofrs/uuid"
+	"github.com/gookit/color"
+	"gorm.io/gorm"
+	"path/filepath"
+)
+
+type SqliteInitHandler struct{}
+
+func NewSqliteInitHandler() *SqliteInitHandler {
+	return &SqliteInitHandler{}
+}
+
+// WriteConfig mysql回写配置
+func (h SqliteInitHandler) WriteConfig(ctx context.Context) error {
+	c, ok := ctx.Value("config").(config.Sqlite)
+	if !ok {
+		return errors.New("mysql config invalid")
+	}
+	global.GVA_CONFIG.System.DbType = "sqlite"
+	global.GVA_CONFIG.Sqlite = c
+	global.GVA_CONFIG.JWT.SigningKey = uuid.Must(uuid.NewV4()).String()
+	cs := utils.StructToMap(global.GVA_CONFIG)
+	for k, v := range cs {
+		global.GVA_VP.Set(k, v)
+	}
+	return global.GVA_VP.WriteConfig()
+}
+
+// EnsureDB 创建数据库并初始化 sqlite
+func (h SqliteInitHandler) EnsureDB(ctx context.Context, conf *request.InitDB) (next context.Context, err error) {
+	if s, ok := ctx.Value("dbtype").(string); !ok || s != "sqlite" {
+		return ctx, ErrDBTypeMismatch
+	}
+
+	c := conf.ToSqliteConfig()
+	next = context.WithValue(ctx, "config", c)
+	if c.Dbname == "" {
+		return ctx, nil
+	} // 如果没有数据库名, 则跳出初始化数据
+
+	dsn := conf.SqliteEmptyDsn()
+
+	var db *gorm.DB
+	if db, err = gorm.Open(sqlite.Open(dsn), &gorm.Config{
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}); err != nil {
+		return ctx, err
+	}
+	global.GVA_CONFIG.AutoCode.Root, _ = filepath.Abs("..")
+	next = context.WithValue(next, "db", db)
+	return next, err
+}
+
+func (h SqliteInitHandler) InitTables(ctx context.Context, inits initSlice) error {
+	return createTables(ctx, inits)
+}
+
+func (h SqliteInitHandler) InitData(ctx context.Context, inits initSlice) error {
+	next, cancel := context.WithCancel(ctx)
+	defer func(c func()) { c() }(cancel)
+	for _, init := range inits {
+		if init.DataInserted(next) {
+			color.Info.Printf(InitDataExist, Sqlite, init.InitializerName())
+			continue
+		}
+		if n, err := init.InitializeData(next); err != nil {
+			color.Info.Printf(InitDataFailed, Sqlite, init.InitializerName(), err)
+			return err
+		} else {
+			next = n
+			color.Info.Printf(InitDataSuccess, Sqlite, init.InitializerName())
+		}
+	}
+	color.Info.Printf(InitSuccess, Sqlite)
+	return nil
+}