opms_backend/opms_libary/db/db.go

package db

import (
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/frame/g"

	myerrors "dashoo.cn/opms_libary/myerrors"
)

type ServiceBase struct {
	Tenant    string
	SafeModel *gdb.Model
	DB        gdb.DB
	TableName string
}

func (s *ServiceBase) Init(tenant string, tableName string) error {
	//db, err := gdb.Instance(tenant)
	s.Tenant = tenant
	db := g.DB(tenant)
	if db == nil {
		return myerrors.NewSysError(nil, "数据库初始化异常")
	}
	//if err != nil {
	//	return gerror.New("获取数据库连接失败")
	//}
	s.DB = db
	s.TableName = tableName
	//s.Model = db.Table(tableName)
	s.SafeModel = s.DB.Table(tableName).Safe()
	return nil
}

// Exists 判断数据是否在数据表里已存在
func (s *ServiceBase) Exists(tableName string, where ...interface{}) (bool, error) {
	exist, err := s.DB.Table(tableName).Fields("1 as id").FindOne(where...)
	if err != nil {
		return false, myerrors.NewDbError(err)
	}
	if len(exist) > 0 {
		return true, nil
	}
	return false, nil
}