init: web_hgcg_server 初始化提交
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
Claude Opus 4.7
@@ -0,0 +1,324 @@
|
||||
// Code generated by gen mm. DO NOT EDIT.
|
||||
package usermod
|
||||
|
||||
import (
|
||||
"context"
|
||||
"videoweb/pub/common/log"
|
||||
"videoweb/pub/common/srvdef"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
/*
|
||||
|
||||
var (
|
||||
collUser *mongos.Collection
|
||||
)
|
||||
|
||||
type getIndexeser interface {
|
||||
GetIndexes() []mongo.IndexModel
|
||||
}
|
||||
|
||||
func Init(ctx context.Context) {
|
||||
collUser = mongos.Shop.Coll(TN.User)
|
||||
|
||||
|
||||
// create indexes for collection 'user's
|
||||
if v, ok := interface{}(&User{}).(getIndexeser); ok {
|
||||
if indexes := v.GetIndexes(); len(indexes) > 0 {
|
||||
if _, err := collUser.CreateIndex(ctx, indexes); err != nil {
|
||||
log.Fatal("collUser.CreateIndex failed", log.E(err))
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
*/
|
||||
/*
|
||||
// InsertOneUser 创建一个User
|
||||
func InsertOneUser(ctx context.Context, data *User) *srvdef.SrvErr {
|
||||
_, err := db.InsertOne(ctx, data)
|
||||
if err != nil {
|
||||
log.Error("usermod:InsertOneUser db.InsertOne failed",
|
||||
log.Any("doc", data),
|
||||
log.E(err))
|
||||
return srvdef.New(srvdef.ErrDbInsertError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
*/
|
||||
|
||||
// InsertManyUsers 创建多个User
|
||||
func InsertManyUsers(ctx context.Context, data []*User, opts ...*options.InsertManyOptions) *srvdef.SrvErr {
|
||||
_, err := db.InsertMany(ctx, &data)
|
||||
if err != nil {
|
||||
log.Error("usermod:InsertManyUsers db.InsertMany failed",
|
||||
log.Any("doc", data),
|
||||
log.E(err))
|
||||
return srvdef.New(srvdef.ErrDbInsertError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/*
|
||||
// FindOneUser 查找一个User
|
||||
func FindOneUser(ctx context.Context, filter bson.M, opts ...*options.FindOneOptions) (
|
||||
*User, *srvdef.SrvErr) {
|
||||
var data *User
|
||||
if err := db.FindOneEx(ctx, &data, filter, opts...); err != nil {
|
||||
if err != mongo.ErrNoDocuments {
|
||||
log.Warn("usermod:FindOneUser db.FindOneEx failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil, srvdef.New(srvdef.ErrNoData, "", "")
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
*/
|
||||
// FindOneUserById 根据Id查找一个User
|
||||
func FindOneUserById(ctx context.Context, id int64, opts ...*options.FindOneOptions) (
|
||||
*User, *srvdef.SrvErr) {
|
||||
filter := bson.M{
|
||||
FN.User.Id: id,
|
||||
}
|
||||
var data *User
|
||||
if err := db.FindOneEx(ctx, &data, filter, opts...); err != nil {
|
||||
if err != mongo.ErrNoDocuments {
|
||||
log.Warn("usermod:FindOneUser db.FindOneEx failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil, srvdef.New(srvdef.ErrNoData, "", "")
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
|
||||
// FindOneUserWithBind 用自定义结构查找一个User
|
||||
func FindOneUserWithBind(ctx context.Context, bind interface{}, filter bson.M, opts ...*options.FindOneOptions) *srvdef.SrvErr {
|
||||
if err := db.FindOneEx(ctx, bind, filter, opts...); err != nil {
|
||||
if err != mongo.ErrNoDocuments {
|
||||
log.Warn("usermod:FindOneUserWithBind db.FindOneEx failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return srvdef.New(srvdef.ErrNoData, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// FindOneUserAndUpdate 更新查找一个User
|
||||
func FindOneUserAndUpdate(ctx context.Context, filter bson.M, update bson.M, opts ...*options.FindOneAndUpdateOptions) (
|
||||
*User, *srvdef.SrvErr) {
|
||||
var data User
|
||||
err := db.FindOneAndUpdate(ctx, &data, filter, update, opts...)
|
||||
if err != nil {
|
||||
log.Warn("usermod:FindOneUserAndUpdate db.FindOneAndUpdate failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return &data, nil
|
||||
}
|
||||
|
||||
/*
|
||||
|
||||
// DelOneUser 删除一个User
|
||||
func DelOneUser(ctx context.Context, filter bson.M) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
count, err := db.DeleteOne(ctx, filter)
|
||||
if err != nil {
|
||||
log.Warn("usermod:DelOneUser db.DeleteOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "")
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
*/
|
||||
|
||||
// DelOneUserById 根据Id删除一个User
|
||||
func DelOneUserById(ctx context.Context, id int64, opts ...*options.FindOneOptions) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
filter := bson.M{
|
||||
FN.User.Id: id,
|
||||
}
|
||||
return DelOneUser(ctx, filter)
|
||||
}
|
||||
|
||||
// DelManyUsers 删除多个User
|
||||
func DelManyUsers(ctx context.Context, filter bson.M) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
count, err := db.DeleteMany(ctx, filter)
|
||||
if err != nil {
|
||||
log.Warn("usermod:DelManyUsers} db.DeleteMany failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return 0, srvdef.New(srvdef.ErrDbDeleteError, "", "")
|
||||
}
|
||||
return count, nil
|
||||
} /*
|
||||
// UpdateOneUser 更新一个User
|
||||
func UpdateOneUser(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
|
||||
_, err := db.UpdateOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("usermod:UpdateOneUser db.UpdateOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return nil
|
||||
}*/
|
||||
// MustUpdateOneUser 更新一个User
|
||||
func MustUpdateOneUser(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
|
||||
result, err := db.UpdateOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("usermod:UpdateOneUser db.UpdateOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
if result.MatchedCount != 1 {
|
||||
return srvdef.New(srvdef.ErrNoData, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpdateOneUserWithResult 更新一个User
|
||||
func UpdateOneUserWithResult(ctx context.Context, filter bson.M, update bson.M, opts ...*options.UpdateOptions) (*mongo.UpdateResult, *srvdef.SrvErr) {
|
||||
result, err := db.UpdateOne(ctx, filter, update, opts...)
|
||||
if err != nil {
|
||||
log.Warn("usermod:UpdateOneUserWithResult db.UpdateOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
if result.MatchedCount != 1 {
|
||||
return nil, srvdef.New(srvdef.ErrNoData, "", "")
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// UpdateManyUsers 更新多个User
|
||||
func UpdateManyUsers(ctx context.Context, filter bson.M, update bson.M) *srvdef.SrvErr {
|
||||
result, err := db.UpdateMany(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("usermod:UpdateManyUsers db.UpdateMany failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
if result.MatchedCount <= 0 {
|
||||
log.Warn("usermod:UpdateManyUsers db.UpdateMany failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/*
|
||||
|
||||
// CountUsers 统计User数量
|
||||
func CountUsers(ctx context.Context, filter bson.M) (
|
||||
int64, *srvdef.SrvErr) {
|
||||
count, err := db.Count(ctx, filter)
|
||||
if err != nil {
|
||||
log.Warn("usermod:CountUsers db.Count failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return 0, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
*/
|
||||
|
||||
// FindManyUsers 查找一批User
|
||||
func FindManyUsers(ctx context.Context, filter bson.M,
|
||||
opts ...*options.FindOptions) ([]*User, *srvdef.SrvErr) {
|
||||
data := make([]*User, 0)
|
||||
if err := db.Find(ctx, &data, filter, opts...); err != nil {
|
||||
log.Warn("usermod:FindManyUsers db.Find failed", log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
|
||||
// FindManyUsers 查找一批User
|
||||
func FindManyUsersAsInt64Map(ctx context.Context, filter bson.M, keyFn func(*User) int64, opts ...*options.FindOptions) (map[int64]*User, *srvdef.SrvErr) {
|
||||
data := make([]*User, 0)
|
||||
if err := db.Find(ctx, &data, filter, opts...); err != nil {
|
||||
log.Warn("usermod:FindManyUsers db.Find failed", log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
result := make(map[int64]*User)
|
||||
for _, item := range data {
|
||||
key := keyFn(item)
|
||||
if _, ok := result[key]; !ok {
|
||||
result[key] = item
|
||||
}
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// FindManyUsersWithBind 用自己定义结构查找一批User数据
|
||||
func FindManyUsersWithBind(ctx context.Context, bind interface{}, filter bson.M,
|
||||
opts ...*options.FindOptions) *srvdef.SrvErr {
|
||||
if err := db.Find(ctx, bind, filter, opts...); err != nil {
|
||||
log.Warn("usermod:FindManyUsersWithBind db.Find failed", log.E(err), log.Any("filter", filter))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/*
|
||||
|
||||
// AggregateUsers 聚合查找一批User
|
||||
func AggregateUsers(ctx context.Context,
|
||||
bind interface{}, pipeline []bson.M) *srvdef.SrvErr {
|
||||
if err := db.Aggregate(ctx, bind, pipeline); err != nil {
|
||||
log.Warn("usermod:AggregateUsers db.Aggregate failed",
|
||||
log.E(err), log.Any("pipeline", pipeline))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
*/
|
||||
|
||||
// ObjectAggregateUsers 聚合查找一批User数据
|
||||
func ObjectAggregateUsers(ctx context.Context,
|
||||
bind interface{}, pipeline []bson.M) *srvdef.SrvErr {
|
||||
if err := db.ObjectAggregate(ctx, bind, pipeline); err != nil {
|
||||
log.Warn("usermod:ObjectAggregateUsers db.Aggregate failed",
|
||||
log.E(err), log.Any("pipeline", pipeline))
|
||||
return srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// BulkWriteUsers 批量写入或者更新User数据
|
||||
func BulkWriteUsers(ctx context.Context,
|
||||
docs []mongo.WriteModel, opts ...*options.BulkWriteOptions) *srvdef.SrvErr {
|
||||
_, err := db.BulkWrite(ctx, docs, opts...)
|
||||
if err != nil {
|
||||
return srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// UpsertOneUser UpsertOneUser
|
||||
func UpsertOneUser(ctx context.Context, filter bson.M, update bson.M) (*mongo.UpdateResult, *srvdef.SrvErr) {
|
||||
result, err := db.UpsertOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
log.Warn("usermod:UpsertOneUser db.UpdateOne failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbUpdateError, "", "")
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// DistinctUser 查找指定字段的不同值
|
||||
func DistinctUser(ctx context.Context, filter bson.M, fieldName string) ([]interface{}, *srvdef.SrvErr) {
|
||||
result, err := db.Distince(ctx, filter, fieldName)
|
||||
if err != nil {
|
||||
log.Warn("usermod:DistinctUser db.Distince failed",
|
||||
log.E(err), log.Any("filter", filter))
|
||||
return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
Reference in New Issue
Block a user