134 lines
3.2 KiB
Go
134 lines
3.2 KiB
Go
package mongo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/Leufolium/test/dbstruct"
|
|
"github.com/qiniu/qmgo"
|
|
)
|
|
|
|
type UserIdMap struct {
|
|
Seq int64 `json:"seq" bson:"_id"` //用户业务id序列号
|
|
UserId int64 `json:"user_id" bson:"user_id"` //映射后用户业务id序列号
|
|
}
|
|
|
|
type Mongo struct {
|
|
clientMix *qmgo.Client
|
|
}
|
|
|
|
func NewMongo() (mongo *Mongo, err error) {
|
|
mongo = new(Mongo)
|
|
|
|
mongo.clientMix, err = NewMongoClient()
|
|
if err != nil {
|
|
fmt.Printf("NewMongoClient fail, cfg: %v, err: %v", nil, err)
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
const (
|
|
DBUserIdSeq = "user_id_seq"
|
|
COLUserIdSeq = "user_id_seq"
|
|
COLUserIdMap = "user_id_map"
|
|
|
|
DBAccountIdSeq = "account_id_seq"
|
|
COLAccountIdSeq = "account_id_seq"
|
|
|
|
DBAccount = "account"
|
|
COLAccount = "account"
|
|
|
|
DBLogin = "login"
|
|
COLLogin = "login"
|
|
)
|
|
|
|
func (m *Mongo) getColUserIdSeq() *qmgo.Collection {
|
|
return m.clientMix.Database(DBUserIdSeq).Collection(COLUserIdSeq)
|
|
}
|
|
|
|
func (m *Mongo) getColUserIdMap() *qmgo.Collection {
|
|
return m.clientMix.Database(DBUserIdSeq).Collection(COLUserIdMap)
|
|
}
|
|
|
|
// Account表
|
|
func (m *Mongo) getColAccount() *qmgo.Collection {
|
|
return m.clientMix.Database(DBAccount).Collection(COLAccount)
|
|
}
|
|
|
|
// 登录表
|
|
func (m *Mongo) getColLogin() *qmgo.Collection {
|
|
return m.clientMix.Database(DBLogin).Collection(COLLogin)
|
|
}
|
|
|
|
// AccountIdSeq序列表
|
|
func (m *Mongo) getColAccountIdSeq() *qmgo.Collection {
|
|
return m.clientMix.Database(DBAccountIdSeq).Collection(COLAccountIdSeq)
|
|
}
|
|
|
|
func (m *Mongo) CreateBatch(ctx context.Context, userIdSeqs []*UserIdMap) error {
|
|
col := m.getColUserIdMap()
|
|
_, err := col.InsertMany(ctx, userIdSeqs)
|
|
return err
|
|
}
|
|
|
|
// account相关
|
|
func (m *Mongo) CreateAccount(ctx context.Context, account *dbstruct.Account) error {
|
|
col := m.getColAccount()
|
|
_, err := col.InsertOne(ctx, account)
|
|
return err
|
|
}
|
|
|
|
func (m *Mongo) CreateLogin(ctx context.Context, login *dbstruct.Login) error {
|
|
col := m.getColLogin()
|
|
_, err := col.InsertOne(ctx, login)
|
|
return err
|
|
}
|
|
|
|
func (m *Mongo) GetAndUpdateAccountIdSeq(ctx context.Context) (accountIdSeq *dbstruct.AccountIdSeq, err error) {
|
|
col := m.getColAccountIdSeq()
|
|
|
|
change := qmgo.Change{
|
|
Update: qmgo.M{"$inc": qmgo.M{"seq": 1}},
|
|
Upsert: true,
|
|
ReturnNew: false,
|
|
}
|
|
|
|
accountIdSeqInstance := dbstruct.AccountIdSeq{}
|
|
if err = col.Find(ctx, qmgo.M{"_id": "account_id_seq_id"}).Apply(change, &accountIdSeqInstance); err != nil {
|
|
fmt.Printf("change error : %v", err)
|
|
return
|
|
}
|
|
|
|
return &accountIdSeqInstance, err
|
|
}
|
|
|
|
func (m *Mongo) GetAndUpdateUserIdSeq(ctx context.Context) (userIdSeq *dbstruct.UserIdSeq, err error) {
|
|
col := m.getColUserIdSeq()
|
|
|
|
change := qmgo.Change{
|
|
Update: qmgo.M{"$inc": qmgo.M{"seq": 1}},
|
|
Upsert: true,
|
|
ReturnNew: false,
|
|
}
|
|
|
|
userIdSeqInstance := dbstruct.UserIdSeq{}
|
|
if err = col.Find(ctx, qmgo.M{"_id": "user_id_seq_id"}).Apply(change, &userIdSeqInstance); err != nil {
|
|
fmt.Printf("change error : %v", err)
|
|
return
|
|
}
|
|
|
|
return &userIdSeqInstance, err
|
|
}
|
|
|
|
func (m *Mongo) GetMappedUserId(ctx context.Context, userIdSeq int64) (*dbstruct.UserIdMap, error) {
|
|
userIdMap := &dbstruct.UserIdMap{}
|
|
col := m.getColUserIdMap()
|
|
query := qmgo.M{
|
|
"_id": userIdSeq,
|
|
}
|
|
err := col.Find(ctx, query).One(&userIdMap)
|
|
return userIdMap, err
|
|
}
|