package notifications import ( "context" "apps/backend/internal/middleware" "apps/backend/internal/response" "apps/backend/internal/svc" "apps/backend/internal/types" "github.com/zeromicro/go-zero/core/logx" ) type UnreadNotificationCountLogic struct { logx.Logger ctx context.Context svcCtx *svc.ServiceContext } func NewUnreadNotificationCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UnreadNotificationCountLogic { return &UnreadNotificationCountLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx} } func (l *UnreadNotificationCountLogic) UnreadNotificationCount() (*types.UnreadCountData, error) { if l.svcCtx.AppNotif == nil { return nil, response.Biz(503, 503001, "notifications not configured") } uid, ok := middleware.UIDFrom(l.ctx) if !ok { return nil, response.Biz(401, 401001, "missing authorization") } c, err := l.svcCtx.AppNotif.UnreadCount(l.ctx, uid) if err != nil { return nil, err } return &types.UnreadCountData{Count: c}, nil }