完善聊天功能 初步构建十子棋功能

repo/chatMessage.go

@@ -0,0 +1,21 @@
+package repo
+
+import (
+	"gorm.io/gorm"
+	"work_cation/models"
+)
+
+type chatMessageRepo struct{}
+
+var ChatMsg = &chatMessageRepo{}
+
+func (*chatMessageRepo) GetUserMsgs(db *gorm.DB, uuid string, limit int) []models.ChatMessage {
+	var msgs []models.ChatMessage
+	// ASC：升序（默认），DESC：降序。
+	db.Order("create_tx ASC").Limit(limit).Where("user_id = ?", uuid).Find(&msgs)
+	return msgs
+}
+
+func (*chatMessageRepo) Create(db *gorm.DB, msg *models.ChatMessage) error {
+	return db.Create(msg).Error
+}

repo/user.go

@@ -45,5 +45,5 @@ func (u *userRepo) GetUserInfo(db *gorm.DB) *models.Users {
 
 func (u *userRepo) Update(db *gorm.DB, newUser *models.Users) error {
 	u.isNew = false
-	return db.Model(&models.Users{}).Where("ID = ?", newUser.ID).Updates(newUser).Error
+	return db.Model(&models.Users{}).Where("UserID = ?", newUser.ID).Updates(newUser).Error
 }

repo/userFollow.go

@@ -10,7 +10,7 @@ type userFollowRepo struct{}
 var UserFollow = &userFollowRepo{}
 
 func (u *userFollowRepo) Follow(db *gorm.DB, user *models.UserFollows) error {
-	return db.Create(user).Error
+	return db.Model(&models.UserFollows{}).Create(user).Error
 }
 
 func (u *userFollowRepo) GetUser(db *gorm.DB, uuid string) *models.UserFollows {
@@ -25,7 +25,7 @@ func (u *userFollowRepo) GetUser(db *gorm.DB, uuid string) *models.UserFollows {
 	return &user
 }
 func (u *userFollowRepo) UnFollow(db *gorm.DB, user *models.UserFollows) error {
-	return db.Where("id = ?", user.ID).Delete(user).Error
+	return db.Model(&models.UserFollows{}).Where("id = ?", user.ID).Delete(user).Error
 }
 func (u *userFollowRepo) All(db *gorm.DB) ([]models.UserFollows, error) {
 	var users []models.UserFollows