mayfly-go/server/internal/db/dbm/sqlite/meta.go

package sqlite

import (
	"database/sql"
	"errors"
	"mayfly-go/internal/db/dbm/dbi"
	"os"
)

func init() {
	dbi.Register(dbi.DbTypeSqlite, new(Meta))
}

type Meta struct {
}

func (md *Meta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
	// 用host字段来存sqlite的文件路径
	// 检查文件是否存在,否则报错，基于sqlite会自动创建文件，为了服务器文件安全，所以先确定文件存在再连接，不自动创建
	if _, err := os.Stat(d.Host); err != nil {
		return nil, errors.New("数据库文件不存在")
	}

	db, err := sql.Open("sqlite", d.Host)
	if err != nil {
		return nil, err
	}
	_, err = db.Exec("PRAGMA busy_timeout = 50000;")
	return db, err
}

func (sm *Meta) GetDialect(conn *dbi.DbConn) dbi.Dialect {
	return &SqliteDialect{dc: conn}
}

func (sm *Meta) GetMetaData(conn *dbi.DbConn) *dbi.MetaDataX {
	return dbi.NewMetaDataX(&SqliteMetaData{dc: conn})
}