Error code standardization

pkg/common/db/table/relation/friend.go

@@ -1,6 +1,9 @@
 package relation
 
-import "time"
+import (
+	"context"
+	"time"
+)
 
 const (
 	FriendModelTableName = "friends"
@@ -21,4 +24,30 @@ func (FriendModel) TableName() string {
 }
 
 type FriendModelInterface interface {
+	// 插入多条记录
+	Create(ctx context.Context, friends []*FriendModel) (err error)
+	// 删除ownerUserID指定的好友
+	Delete(ctx context.Context, ownerUserID string, friendUserIDs []string) (err error)
+	// 更新ownerUserID单个好友信息 更新零值
+	UpdateByMap(ctx context.Context, ownerUserID string, friendUserID string, args map[string]interface{}) (err error)
+	// 更新好友信息的非零值
+	Update(ctx context.Context, friends []*FriendModel) (err error)
+	// 更新好友备注（也支持零值 ）
+	UpdateRemark(ctx context.Context, ownerUserID, friendUserID, remark string) (err error)
+	// 获取单个好友信息，如没找到 返回错误
+	Take(ctx context.Context, ownerUserID, friendUserID string) (friend *FriendModel, err error)
+	// 查找好友关系，如果是双向关系，则都返回
+	FindUserState(ctx context.Context, userID1, userID2 string) (friends []*FriendModel, err error)
+	// 获取 owner指定的好友列表 如果有friendUserIDs不存在，也不返回错误
+	FindFriends(ctx context.Context, ownerUserID string, friendUserIDs []string) (friends []*FriendModel, err error)
+	// 获取哪些人添加了friendUserID 如果有ownerUserIDs不存在，也不返回错误
+	FindReversalFriends(ctx context.Context, friendUserID string, ownerUserIDs []string) (friends []*FriendModel, err error)
+	// 获取ownerUserID好友列表 支持翻页
+	FindOwnerFriends(ctx context.Context, ownerUserID string, pageNumber, showNumber int32) (friends []*FriendModel, total int64, err error)
+	// 获取哪些人添加了friendUserID 支持翻页
+	FindInWhoseFriends(ctx context.Context, friendUserID string, pageNumber, showNumber int32) (friends []*FriendModel, total int64, err error)
+	// 获取好友UserID列表
+	FindFriendUserIDs(ctx context.Context, ownerUserID string) (friendUserIDs []string, err error)
+
+	NewTx(tx any) FriendModelInterface
 }

pkg/common/db/table/relation/friend_request.go

@@ -1,6 +1,9 @@
 package relation
 
-import "time"
+import (
+	"context"
+	"time"
+)
 
 const FriendRequestModelTableName = "friend_requests"
 
@@ -21,4 +24,21 @@ func (FriendRequestModel) TableName() string {
 }
 
 type FriendRequestModelInterface interface {
+	// 插入多条记录
+	Create(ctx context.Context, friendRequests []*FriendRequestModel) (err error)
+	// 删除记录
+	Delete(ctx context.Context, fromUserID, toUserID string) (err error)
+	// 更新零值
+	UpdateByMap(ctx context.Context, formUserID string, toUserID string, args map[string]interface{}) (err error)
+	// 更新多条记录 （非零值）
+	Update(ctx context.Context, friendRequests []*FriendRequestModel) (err error)
+	// 获取来指定用户的好友申请  未找到 不返回错误
+	Find(ctx context.Context, fromUserID, toUserID string) (friendRequest *FriendRequestModel, err error)
+	Take(ctx context.Context, fromUserID, toUserID string) (friendRequest *FriendRequestModel, err error)
+	// 获取toUserID收到的好友申请列表
+	FindToUserID(ctx context.Context, toUserID string, pageNumber, showNumber int32) (friendRequests []*FriendRequestModel, total int64, err error)
+	// 获取fromUserID发出去的好友申请列表
+	FindFromUserID(ctx context.Context, fromUserID string, pageNumber, showNumber int32) (friendRequests []*FriendRequestModel, total int64, err error)
+
+	NewTx(tx any) FriendRequestModelInterface
 }

pkg/common/db/table/relation/user.go

@@ -1,6 +1,9 @@
 package relation
 
-import "time"
+import (
+	"context"
+	"time"
+)
 
 const (
 	UserModelTableName = "users"
@@ -22,8 +25,18 @@ type UserModel struct {
 }
 
 func (UserModel) TableName() string {
-	return GroupRequestModelTableName
+	return UserModelTableName
 }
 
 type UserModelInterface interface {
+	Create(ctx context.Context, users []*UserModel) (err error)
+	UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error)
+	Update(ctx context.Context, users []*UserModel) (err error)
+	// 获取指定用户信息  不存在，也不返回错误
+	Find(ctx context.Context, userIDs []string) (users []*UserModel, err error)
+	// 获取某个用户信息  不存在，则返回错误
+	Take(ctx context.Context, userID string) (user *UserModel, err error)
+	// 获取用户信息 不存在，不返回错误
+	Page(ctx context.Context, pageNumber, showNumber int32) (users []*UserModel, count int64, err error)
+	GetAllUserID(ctx context.Context) (userIDs []string, err error)
 }