上传数据库结构模板

go.mod

@@ -4,8 +4,10 @@ go 1.18
 
 require (
 	github.com/gorilla/websocket v1.4.2
+	github.com/json-iterator/go v1.1.12
 	github.com/liwei1dao/lego v0.0.0-20220531091126-a21bb3766fbc
 	github.com/stretchr/testify v1.7.1
+	go.mongodb.org/mongo-driver v1.5.1
 	google.golang.org/protobuf v1.28.0
 )
 
@@ -52,7 +54,6 @@ require (
 	github.com/hashicorp/golang-lru v0.5.4 // indirect
 	github.com/hashicorp/serf v0.9.7 // indirect
 	github.com/jmespath/go-jmespath v0.4.0 // indirect
-	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/juju/ratelimit v1.0.1 // indirect
 	github.com/julienschmidt/httprouter v1.3.0 // indirect
 	github.com/kavu/go_reuseport v1.5.0 // indirect
@@ -106,7 +107,6 @@ require (
 	github.com/xdg-go/stringprep v1.0.2 // indirect
 	github.com/xtaci/kcp-go v5.4.20+incompatible // indirect
 	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
-	go.mongodb.org/mongo-driver v1.5.1 // indirect
 	go.opentelemetry.io/otel v1.6.3 // indirect
 	go.opentelemetry.io/otel/trace v1.6.3 // indirect
 	go.uber.org/atomic v1.7.0 // indirect

modules/login/login_comp.go

@@ -20,7 +20,7 @@ type LoginComp struct {
 func (this *LoginComp) Login(ctx context.Context, session comm.IUserSession, req *pb.UserLoginReq) error {
 	log.Debugf("User - Login: session:%v rsp:%v", session.ToString(), req)
 
-	db_user, err := db.Defsys.FindUserByAccount(req.Name)
+	db_user, err := db.Defsys.User_FindUserByAccount(req.Name)
 	if err != nil {
 		if err != mongo.ErrNoDocuments {
 			return err
@@ -29,7 +29,7 @@ func (this *LoginComp) Login(ctx context.Context, session comm.IUserSession, req
 
 	if db_user.UserId == 0 {
 		db_user.Account = req.Name
-		err = db.Defsys.CreateUser(db_user)
+		err = db.Defsys.User_CreateUser(db_user)
 		if err != nil {
 			return err
 		}
@@ -42,7 +42,7 @@ func (this *LoginComp) Login(ctx context.Context, session comm.IUserSession, req
 		GatewayServiceId: session.GetGatewayServiceId(),
 		UserData:         db_user,
 	}
-	err = cache.Defsys.UpdateUser(cache_user)
+	err = cache.Defsys.User_UpdateUser(cache_user)
 	if err != nil {
 		return err
 	}

sys/cache/core.go

@@ -1,10 +1,9 @@
 package cache
 
-import "go_dreamfactory/pb"
-
 type (
 	ISys interface {
-		UpdateUser(data *pb.Cache_UserData) (err error)
+		IUser
+		IPack
 	}
 )
 
@@ -29,5 +28,3 @@ func NewSys(option ...Option) (sys ISys, err error) {
 	Defsys, err = newSys(options)
 	return
 }
-
-

sys/cache/mail.go

@@ -0,0 +1,7 @@
+package cache
+
+const ( //Redis
+	Redis_mailCache string = "mail:%d"
+)
+
+type IMail interface{}

sys/cache/pack.go

@@ -0,0 +1,7 @@
+package cache
+
+const ( //Redis
+	Redis_PackCache string = "pack:%d"
+)
+
+type IPack interface{}

sys/cache/user.go

@@ -9,6 +9,11 @@ const ( //Redis
 	Redis_UserCache string = "user:%d" //会话列表
 )
 
-func (this *Cache) UpdateUser(data *pb.Cache_UserData) (err error) {
+type IUser interface {
-	return this.redis.Set(fmt.Sprintf(Redis_UserCache, data.UserData.UserId), data, 0)
+	User_UpdateUser(data *pb.Cache_UserData) (err error)
+}
+
+func (this *Cache) User_UpdateUser(data *pb.Cache_UserData) (err error) {
+	err = this.redis.Set(fmt.Sprintf(Redis_UserCache, data.UserData.UserId), data, -1)
+	return
 }

sys/cache/user_test.go

@@ -32,6 +32,6 @@ func TestUpdateUser(t *testing.T) {
 			Account: "aaa",
 		},
 	}
-	err := cache.UpdateUser(user)
+	err := cache.User_UpdateUser(user)
 	require.Nil(t, err)
 }

sys/db/core.go

@@ -1,13 +1,10 @@
 package db
 
-import "go_dreamfactory/pb"
-
 type (
 	ISys interface {
-		FindUserByAccount(account string) (*pb.DB_UserData, error)
+		IUser
-		FindUserById(id uint32) (*pb.DB_UserData, error)
+		IPack
-		CreateUser(user *pb.DB_UserData) error
+		IMail
-		UpdateUser(data *pb.DB_UserData) (err error)
 	}
 )
 

sys/db/mail.go

@@ -0,0 +1,4 @@
+package db
+
+type IMail interface {
+}

sys/db/pack.go

@@ -0,0 +1,4 @@
+package db
+
+type IPack interface {
+}

sys/db/user.go

@@ -18,11 +18,18 @@ const ( //Redis
 	DB_UserIdTable core.SqlTable = "userid" //用户id表
 )
 
+type IUser interface {
+	User_FindUserByAccount(account string) (*pb.DB_UserData, error)
+	User_FindUserById(id uint32) (*pb.DB_UserData, error)
+	User_CreateUser(user *pb.DB_UserData) error
+	User_UpdateUser(data *pb.DB_UserData) (err error)
+}
+
 type UserId struct {
 	UserId uint32 `bson:"_id"`
 }
 
-func (this *DB) FindUserByAccount(account string) (*pb.DB_UserData, error) {
+func (this *DB) User_FindUserByAccount(account string) (*pb.DB_UserData, error) {
 	filter := bson.D{
 		{"account", account},
 	}
@@ -32,7 +39,7 @@ func (this *DB) FindUserByAccount(account string) (*pb.DB_UserData, error) {
 	return user, err
 }
 
-func (this *DB) FindUserById(id uint32) (*pb.DB_UserData, error) {
+func (this *DB) User_FindUserById(id uint32) (*pb.DB_UserData, error) {
 	filter := bson.D{
 		{"_id", id},
 	}
@@ -42,7 +49,7 @@ func (this *DB) FindUserById(id uint32) (*pb.DB_UserData, error) {
 	return user, err
 }
 
-func (this *DB) CreateUser(user *pb.DB_UserData) error {
+func (this *DB) User_CreateUser(user *pb.DB_UserData) error {
 	userId := &UserId{}
 	err := this.mgo.FindOneAndDelete(DB_UserIdTable, bson.M{}).Decode(userId)
 	if err != nil {
@@ -54,6 +61,20 @@ func (this *DB) CreateUser(user *pb.DB_UserData) error {
 	return err
 }
 
+//更新用户数据到DB
+func (this *DB) User_UpdateUser(data *pb.DB_UserData) error {
+	err := this.mgo.FindOneAndUpdate(
+		DB_UserTable,
+		bson.M{"_id": data.UserId},
+		bson.M{"$set": bson.M{
+			"niceName": data.NiceName,
+			"email":    data.Email,
+		}},
+		options.FindOneAndUpdate().SetUpsert(false).SetReturnDocument(options.After),
+	).Decode(data)
+	return err
+}
+
 //校验数据库初始化工作是否完成
 func (this DB) checkUserIdInit() (err error) {
 	ctx, _ := context.WithTimeout(context.Background(), time.Second*60)
@@ -83,17 +104,3 @@ func (this DB) checkUserIdInit() (err error) {
 	}
 	return
 }
-
-//更新用户数据到DB
-func (this *DB) UpdateUser(data *pb.DB_UserData) error {
-	err := this.mgo.FindOneAndUpdate(
-		DB_UserTable,
-		bson.M{"_id": data.UserId},
-		bson.M{"$set": bson.M{
-			"niceName": data.NiceName,
-			"email":    data.Email,
-		}},
-		options.FindOneAndUpdate().SetUpsert(false).SetReturnDocument(options.After),
-	).Decode(data)
-	return err
-}

sys/db/user_test.go

@@ -32,16 +32,16 @@ func TestCreate(t *testing.T) {
 		Email:    "1111@legu.com",
 	}
 
-	err := db.CreateUser(user)
+	err := db.User_CreateUser(user)
 	require.Nil(t, err)
 }
 
 func TestFindOne(t *testing.T) {
-	user, err := db.FindUserById(1)
+	user, err := db.User_FindUserById(1)
 	require.Nil(t, err)
 	assert.Equal(t, "legu1", user.Account)
 
-	user2, err := db.FindUserByAccount("legu1")
+	user2, err := db.User_FindUserByAccount("legu1")
 	require.Nil(t, err)
 	assert.Equal(t, "legu1", user2.Account)
 
@@ -52,7 +52,7 @@ func TestUpdate(t *testing.T) {
 		UserId: 10001,
 		Email:  "new@qq.com",
 	}
-	err := db.UpdateUser(user)
+	err := db.User_UpdateUser(user)
 	require.Nil(t, err)
 
 	assert.Equal(t, "new@qq.com", user.Email)