diff --git a/pkg/common/db/controller/friend_request.go b/pkg/common/db/controller/friend_request.go deleted file mode 100644 index d4c2b3032..000000000 --- a/pkg/common/db/controller/friend_request.go +++ /dev/null @@ -1,110 +0,0 @@ -package controller - -import ( - "Open_IM/pkg/common/db/relation" - "context" - "gorm.io/gorm" -) - -type FriendRequestInterface interface { - Create(ctx context.Context, friends []*relation.FriendRequest) (err error) - Delete(ctx context.Context, fromUserID, toUserID string) (err error) - UpdateByMap(ctx context.Context, ownerUserID string, args map[string]interface{}) (err error) - Update(ctx context.Context, friends []*relation.FriendRequest) (err error) - Find(ctx context.Context, ownerUserID string) (friends []*relation.FriendRequest, err error) - Take(ctx context.Context, fromUserID, toUserID string) (friend *relation.FriendRequest, err error) - FindToUserID(ctx context.Context, toUserID string) (friends []*relation.FriendRequest, err error) - FindFromUserID(ctx context.Context, fromUserID string) (friends []*relation.FriendRequest, err error) -} - -type FriendRequestController struct { - database FriendRequestInterface -} - -func NewFriendRequestController(db *gorm.DB) *FriendRequestController { - return &FriendRequestController{database: NewFriendRequestDatabase(db)} -} - -func (f *FriendRequestController) Create(ctx context.Context, friends []*relation.FriendRequest) (err error) { - return f.database.Create(ctx, friends) -} -func (f *FriendRequestController) Delete(ctx context.Context, fromUserID, toUserID string) (err error) { - return f.database.Delete(ctx, fromUserID, toUserID) -} -func (f *FriendRequestController) UpdateByMap(ctx context.Context, ownerUserID string, args map[string]interface{}) (err error) { - return f.database.UpdateByMap(ctx, ownerUserID, args) -} -func (f *FriendRequestController) Update(ctx context.Context, friends []*relation.FriendRequest) (err error) { - return f.database.Update(ctx, friends) -} -func (f *FriendRequestController) Find(ctx context.Context, ownerUserID string) (friends []*relation.FriendRequest, err error) { - return f.database.Find(ctx, ownerUserID) -} -func (f *FriendRequestController) Take(ctx context.Context, fromUserID, toUserID string) (friend *relation.FriendRequest, err error) { - return f.database.Take(ctx, fromUserID, toUserID) -} -func (f *FriendRequestController) FindToUserID(ctx context.Context, toUserID string) (friends []*relation.FriendRequest, err error) { - return f.database.FindToUserID(ctx, toUserID) -} -func (f *FriendRequestController) FindFromUserID(ctx context.Context, fromUserID string) (friends []*relation.FriendRequest, err error) { - return f.database.FindFromUserID(ctx, fromUserID) -} - -type FriendRequestDatabaseInterface interface { - Create(ctx context.Context, friends []*relation.FriendRequest) (err error) - Delete(ctx context.Context, fromUserID, toUserID string) (err error) - UpdateByMap(ctx context.Context, ownerUserID string, args map[string]interface{}) (err error) - Update(ctx context.Context, friends []*relation.FriendRequest) (err error) - Find(ctx context.Context, ownerUserID string) (friends []*relation.FriendRequest, err error) - Take(ctx context.Context, fromUserID, toUserID string) (friend *relation.FriendRequest, err error) - FindToUserID(ctx context.Context, toUserID string) (friends []*relation.FriendRequest, err error) - FindFromUserID(ctx context.Context, fromUserID string) (friends []*relation.FriendRequest, err error) -} - -type FriendRequestDatabase struct { - sqlDB *relation.FriendRequest - friend *FriendDatabase -} - -func (f *FriendRequestDatabase) Update(ctx context.Context, friends []*relation.FriendRequest) (err error) { - return f.sqlDB.DB.Transaction(func(tx *gorm.DB) error { - if err := f.sqlDB.Update(ctx, friends); err != nil { - return err - } - if err := f.friend.Update(); err != nil { - return err - } - return nil - }) -} - -func NewFriendRequestDatabase(db *gorm.DB) *FriendRequestDatabase { - sqlDB := relation.NewFriendRequest(db) - database := &FriendRequestDatabase{ - sqlDB: sqlDB, - } - return database -} - -func (f *FriendRequestDatabase) Create(ctx context.Context, friends []*relation.FriendRequest) (err error) { - return f.sqlDB.Create(ctx, friends) -} -func (f *FriendRequestDatabase) Delete(ctx context.Context, fromUserID, toUserID string) (err error) { - return f.sqlDB.Delete(ctx, fromUserID, toUserID) -} -func (f *FriendRequestDatabase) UpdateByMap(ctx context.Context, ownerUserID string, args map[string]interface{}) (err error) { - return f.sqlDB.UpdateByMap(ctx, ownerUserID, args) -} - -func (f *FriendRequestDatabase) Find(ctx context.Context, ownerUserID string) (friends []*relation.FriendRequest, err error) { - return f.sqlDB.Find(ctx, ownerUserID) -} -func (f *FriendRequestDatabase) Take(ctx context.Context, fromUserID, toUserID string) (friend *relation.FriendRequest, err error) { - return f.sqlDB.Take(ctx, fromUserID, toUserID) -} -func (f *FriendRequestDatabase) FindToUserID(ctx context.Context, toUserID string) (friends []*relation.FriendRequest, err error) { - return f.sqlDB.FindToUserID(ctx, toUserID) -} -func (f *FriendRequestDatabase) FindFromUserID(ctx context.Context, fromUserID string) (friends []*relation.FriendRequest, err error) { - return f.sqlDB.FindFromUserID(ctx, fromUserID) -}