sqldb.go

package gsqldb

import (
	"fmt"
	"github.com/cryptowilliam/goutil/basic/gerrors"
	"github.com/cryptowilliam/goutil/container/gstring"
	"github.com/cryptowilliam/goutil/database/gdriver"
	"github.com/cryptowilliam/goutil/database/gdsn"
	_ "github.com/go-sql-driver/mysql"
	"xorm.io/xorm"
)

type (
	SqlDB struct {
		ng  *xorm.Engine
		dvr gdriver.Driver
	}
)

func Dial(dsn string) (*SqlDB, error) {
	pDSN, err := gdsn.Parse(dsn)
	if err != nil {
		return nil, err
	}

	r := &SqlDB{dvr: pDSN.Driver}
	switch pDSN.Driver {
	case gdriver.MySQL:
		r.ng, err = xorm.NewEngine("mysql", dsn)
	case gdriver.Mssql:
		r.ng, err = xorm.NewEngine("mssql", dsn)
	case gdriver.PgSQL:
		r.ng, err = xorm.NewEngine("postgres", dsn)
	case gdriver.SQLite:
		r.ng, err = xorm.NewEngine("sqlite3", dsn)
	case gdriver.Oracle:
		r.ng, err = xorm.NewEngine("oracle", dsn)
	case gdriver.TiDB:
		r.ng, err = xorm.NewEngine("mysql", dsn)
	case gdriver.CockroachDB:
		r.ng, err = xorm.NewEngine("postgres", dsn)
	default:
		return nil, gerrors.New("unsupported database driver %s", pDSN.Driver.String())
	}

	return r, err
}

func (s *SqlDB) Tables() ([]string, error) {
	tables, err := s.ng.DBMetas()
	if err != nil {
		return nil, err
	}

	var res []string
	for _, v := range tables {
		res = append(res, v.Name)
	}
	return res, nil
}

// 根据结构体中存在的非空数据来查询单条数据
func (s *SqlDB) SelectOne(condAndOutput interface{}) (bool, error) {
	return s.ng.Get(condAndOutput)
}

// 根据cond...结构体中存在的非空数据来查询全部数据
func (s *SqlDB) SelectAll(output interface{}, cond ...interface{}) error {
	return s.ng.Find(output, cond...)
}

// 插入单条数据
func (s *SqlDB) InsertOne(data interface{}) (int64, error) {
	return s.ng.InsertOne(data)
}

// 根据cond...结构体中存在的非空数据来Upsert单条数据
func (s *SqlDB) UpsertOne(data, cond interface{}) (int64, error) {
	n, err := s.ng.InsertOne(data)
	if s.dvr == gdriver.MySQL && n == 0 && gstring.StartWith(err.Error(), "Error 1062") { // Error 1062: duplicate primary key
		return s.ng.Update(data, cond)
	}
	return n, err
}

// 根据cond...结构体中存在的非空数据来Update单条数据
func (s *SqlDB) UpdateOne(data, cond interface{}) (int64, error) {
	return s.ng.Update(data, cond)
}

// 根据cond结构体中存在的非空数据来查询是否存在，同时cond也是要目标table名
// table: use to known which table to query
func (s *SqlDB) Exist(cond interface{}) (bool, error) {
	return s.ng.Exist(cond)
}

// 根据cond结构体中存在的非空数据来删除记录，同时cond也是要目标table名
// 此接口只允许根据某个属性的特定值进行删除，不允许空条件或者条件中的字段为空，如果条件中有多个字段，则必须同时满足
// 如果要删除全部内容，而不是根据某个属性的特定值进行删除，那么应该使用Clear接口
func (s *SqlDB) Delete(cond interface{}) (int64, error) {
	return s.ng.Unscoped().Delete(cond)
}

// 清空表格内容
func (s *SqlDB) ClearAll(table string) error {
	if s.dvr == gdriver.MySQL {
		_, err := s.ng.Unscoped().Exec(fmt.Sprintf("TRUNCATE TABLE %s", table))
		return err
	}
	return gerrors.New("Clear function doesn't support %s for now", s.dvr)
}

func (s *SqlDB) Close() error {
	return s.ng.Close()
}