55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"go_dreamfactory/pb"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/liwei1dao/lego/core"
|
|
"github.com/liwei1dao/lego/sys/log"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
const ( //Redis
|
|
DB_UserTable core.SqlTable = "user" //用户表
|
|
DB_UserIdTable core.SqlTable = "userid" //用户id表
|
|
)
|
|
|
|
//校验数据库初始化工作是否完成
|
|
func (this DB) checkUserIdInit() (err error) {
|
|
ctx, _ := context.WithTimeout(context.Background(), time.Second*60)
|
|
count, err := this.mgo.CountDocuments(DB_UserIdTable, bson.M{})
|
|
if err != nil || count == 0 {
|
|
//批量插入数据
|
|
leng := 1000000
|
|
cIds := make([]interface{}, leng)
|
|
for i, _ := range cIds {
|
|
cIds[i] = 1000000 + i
|
|
}
|
|
data := make([]interface{}, leng)
|
|
r := rand.New(rand.NewSource(time.Now().Unix()))
|
|
n := 0
|
|
for _, i := range r.Perm(leng) {
|
|
data[n] = bson.M{"_id": i}
|
|
n++
|
|
}
|
|
var (
|
|
err error
|
|
)
|
|
begin := time.Now()
|
|
if _, err = this.mgo.InsertManyByCtx(DB_UserIdTable, ctx, data); err != nil {
|
|
return fmt.Errorf("checkUserIdInit err=%s", err.Error())
|
|
}
|
|
log.Debugf("checkUserIdInit succ time consuming:%v", time.Now().Sub(begin))
|
|
}
|
|
return
|
|
}
|
|
|
|
//更新用户数据到DB
|
|
func (this *DB) UpdateUser(data *pb.DB_UserData) (err error) {
|
|
|
|
return
|
|
}
|