98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package sdbc
|
|
|
|
import (
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/sirupsen/logrus"
|
|
"gitter.top/common/lormatter"
|
|
"google.golang.org/protobuf/proto"
|
|
"gorm.io/gorm"
|
|
"reflect"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
var formatter = &lormatter.Formatter{
|
|
ShowTime: true,
|
|
}
|
|
formatter.Register()
|
|
}
|
|
|
|
type tabler interface {
|
|
TableName() string
|
|
}
|
|
|
|
type Config struct {
|
|
Dbname string // 数据库名
|
|
MaxIdleConn int // 最大空闲连接
|
|
MaxOpenConn int // 最大连接
|
|
MaxLifetime time.Duration // 最大生存时长
|
|
Debug bool // 是否开启debug
|
|
}
|
|
|
|
type Driver struct {
|
|
dbname string
|
|
client *gorm.DB
|
|
}
|
|
|
|
func NewSDBC(cfg *Config) *Driver {
|
|
if cfg == nil {
|
|
logrus.Fatalf("config can not be nil")
|
|
}
|
|
var (
|
|
driver = new(Driver)
|
|
err error
|
|
)
|
|
driver.client, err = gorm.Open(sqlite.Open(cfg.Dbname), &gorm.Config{})
|
|
if err != nil {
|
|
logrus.Fatalf("open sqlite file failed: %v", err)
|
|
}
|
|
if cfg.Debug {
|
|
driver.client = driver.client.Debug()
|
|
}
|
|
rawDB, err := driver.client.DB()
|
|
if err != nil {
|
|
logrus.Fatalf("get sqlite instance failed: %v", err)
|
|
}
|
|
if cfg.MaxOpenConn != 0 {
|
|
rawDB.SetMaxOpenConns(cfg.MaxOpenConn)
|
|
}
|
|
if cfg.MaxIdleConn != 0 {
|
|
rawDB.SetMaxIdleConns(cfg.MaxIdleConn)
|
|
}
|
|
if cfg.MaxLifetime != 0 {
|
|
rawDB.SetConnMaxLifetime(cfg.MaxLifetime)
|
|
}
|
|
driver.dbname = cfg.Dbname
|
|
return driver
|
|
}
|
|
|
|
func (s *Driver) GetClient() *gorm.DB {
|
|
return s.client
|
|
}
|
|
|
|
func (s *Driver) BindModel(prototype any) Operator {
|
|
if prototype == nil {
|
|
logrus.Panic("model can not be nil")
|
|
}
|
|
if _, impled := prototype.(tabler); !impled {
|
|
logrus.Panic("model does not implement tabler interface")
|
|
}
|
|
protomsg, impled := prototype.(proto.Message)
|
|
if !impled {
|
|
logrus.Panic("model does not implement message interface")
|
|
}
|
|
scope := &scope{
|
|
&model{
|
|
Type: protomsg,
|
|
modelKind: reflect.TypeOf(protomsg),
|
|
modelName: string(protomsg.ProtoReflect().Descriptor().FullName()),
|
|
tableName: prototype.(tabler).TableName(),
|
|
},
|
|
s,
|
|
}
|
|
if err := s.client.AutoMigrate(prototype); err != nil {
|
|
logrus.Panicf("auto create table failed: %v", err)
|
|
}
|
|
return scope
|
|
}
|