MaysWind
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package datastore
|
|
|
|
import (
|
|
"xorm.io/xorm"
|
|
|
|
"github.com/mayswind/lab/pkg/errs"
|
|
)
|
|
|
|
// DataStore represents a data storage containing a series of database shards
|
|
type DataStore struct {
|
|
databases []*Database
|
|
}
|
|
|
|
// Choose returns a database instance by sharding key
|
|
func (s *DataStore) Choose(key int64) *Database {
|
|
return s.databases[0]
|
|
}
|
|
|
|
// Query returns a new database session in a specific database by sharding key
|
|
func (s *DataStore) Query(key int64) *xorm.Session {
|
|
return s.Choose(key).NewSession()
|
|
}
|
|
|
|
// DoTransaction runs a new database transaction in a specific database by sharding key
|
|
func (s *DataStore) DoTransaction(key int64, fn func(sess *xorm.Session) error) (err error) {
|
|
return s.Choose(key).DoTransaction(fn)
|
|
}
|
|
|
|
// SyncStructs updates database structs by database models
|
|
func (s *DataStore) SyncStructs(beans ...interface{}) error {
|
|
var err error
|
|
|
|
for i := 0; i < len(s.databases); i++ {
|
|
err = s.databases[i].Sync2(beans...)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// NewDataStore returns a new data storage by a series of database
|
|
func NewDataStore(databases ...*Database) (*DataStore, error) {
|
|
if len(databases) < 1 {
|
|
return nil, errs.ErrDatabaseIsNull
|
|
}
|
|
|
|
return &DataStore{
|
|
databases: databases,
|
|
}, nil
|
|
}
|