Add checkin count in attendance and event api

Signed-off-by: Asai Neko <sugar@sne.moe>

data/attendance.go

@@ -149,6 +149,29 @@ func (self *Attendance) CountUsersByEventID(ctx context.Context, eventID uuid.UU
 		Where("event_id = ?", eventID).
 		Count(&count).Error
 
+	if err == gorm.ErrRecordNotFound {
+		return 0, nil
+	}
+
+	if err != nil {
+		return 0, err
+	}
+
+	return count, nil
+}
+
+func (self *Attendance) CountCheckedInUsersByEventID(ctx context.Context, eventID uuid.UUID) (int64, error) {
+	var count int64
+
+	err := Database.WithContext(ctx).
+		Model(&Attendance{}).
+		Where("event_id = ? AND checkin_at IS NOT NULL AND checkin_at > ?", eventID, time.Time{}). // 过滤未签到用户
+		Count(&count).Error
+
+	if err == gorm.ErrRecordNotFound {
+		return 0, nil
+	}
+
 	if err != nil {
 		return 0, err
 	}

data/event.go

@@ -32,6 +32,7 @@ type EventIndexDoc struct {
 	StartTime    time.Time `json:"start_time"`
 	EndTime      time.Time `json:"end_time"`
 	Thumbnail    string    `json:"thumbnail"`
+	CheckinCount int64     `json:"checkin_count"`
 }
 
 func (self *Event) GetEventById(ctx context.Context, eventId uuid.UUID) (*Event, error) {

service/service_event/get_event_info.go

@@ -45,6 +45,27 @@ func (self *EventServiceImpl) GetEventInfo(payload *EventInfoPayload) (result *E
 		return
 	}
 
+	checkinCount, err := new(data.Attendance).CountCheckedInUsersByEventID(payload.Context, payload.Data.EventId)
+	if err != nil {
+		exception := new(exception.Builder).
+			SetStatus(exception.StatusUser).
+			SetService(exception.ServiceEvent).
+			SetEndpoint(exception.EndpointEventServiceInfo).
+			SetType(exception.TypeCommon).
+			SetOriginal(exception.CommonErrorDatabase).
+			SetError(err).
+			Throw(payload.Context)
+
+		result = &EventInfoResult{
+			Common: shared.CommonResult{
+				HttpCode:  500,
+				Exception: exception,
+			},
+		}
+
+		return
+	}
+
 	result = &EventInfoResult{
 		Common: shared.CommonResult{
 			HttpCode: 200,
@@ -64,6 +85,7 @@ func (self *EventServiceImpl) GetEventInfo(payload *EventInfoPayload) (result *E
 			StartTime:    event.StartTime,
 			EndTime:      event.EndTime,
 			Thumbnail:    event.Thumbnail,
+			CheckinCount: checkinCount,
 		},
 	}