diff --git a/internal/controller/subscription/subscription_v1_add_subscription.go b/internal/controller/subscription/subscription_v1_add_subscription.go index 85513b3..d013a0c 100644 --- a/internal/controller/subscription/subscription_v1_add_subscription.go +++ b/internal/controller/subscription/subscription_v1_add_subscription.go @@ -15,6 +15,7 @@ func (c *ControllerV1) AddSubscription(ctx context.Context, req *v1.AddSubscript ChannelId: req.Id, UserId: req.UserId, }).Insert() + _, err = dao.GoChannels.Ctx(ctx).Where("id",req.Id).Increment("subscription_count",1) if err != nil { return dto.Error(err.Error()), nil } diff --git a/internal/controller/subscription/subscription_v1_delete_subscription.go b/internal/controller/subscription/subscription_v1_delete_subscription.go index 5a85b90..9eef3d8 100644 --- a/internal/controller/subscription/subscription_v1_delete_subscription.go +++ b/internal/controller/subscription/subscription_v1_delete_subscription.go @@ -11,6 +11,7 @@ import ( func (c *ControllerV1) DeleteSubscription(ctx context.Context, req *v1.DeleteSubscriptionReq) (*dto.Result, error) { res := &v1.DeleteSubscriptionRes{} _, err := dao.GoChannelSubscriptions.Ctx(ctx).Wheref("channel_id =? and user_id =?", req.Id, req.UserId).Delete() + _, err = dao.GoChannels.Ctx(ctx).Where("id",req.Id).Decrement("subscription_count",1) if err != nil { return dto.Error(err.Error()), nil }