return summary by status and add required field

api-description/web-api.swagger.yaml

@@ -5049,6 +5049,18 @@ definitions:
       - EXPERIMENT
       - OPERATION
     default: UNKNOWN
+  ListExperimentsResponseSummary:
+    type: object
+    properties:
+      totalWaitingCount:
+        type: string
+        format: int64
+      totalRunningCount:
+        type: string
+        format: int64
+      totalStoppedCount:
+        type: string
+        format: int64
   ListFlagTriggersResponseFlagTriggerWithUrl:
     type: object
     properties:
@@ -6763,6 +6775,12 @@ definitions:
         type: string
     required:
       - environmentId
+      - featureId
+      - startAt
+      - stopAt
+      - goalIds
+      - name
+      - baseVariationId
   experimentCreateExperimentResponse:
     type: object
     properties:
@@ -6952,6 +6970,8 @@ definitions:
       totalCount:
         type: string
         format: int64
+      summary:
+        $ref: '#/definitions/ListExperimentsResponseSummary'
   experimentListGoalsRequestOrderBy:
     type: string
     enum:

pkg/experiment/api/experiment.go

@@ -168,7 +168,7 @@ func (s *experimentService) ListExperiments(
 		return nil, dt.Err()
 	}
 	experimentStorage := v2es.NewExperimentStorage(s.mysqlClient)
-	experiments, nextCursor, totalCount, err := experimentStorage.ListExperiments(
+	experiments, nextCursor, totalCount, summary, err := experimentStorage.ListExperiments(
 		ctx,
 		whereParts,
 		orders,
@@ -196,6 +196,7 @@ func (s *experimentService) ListExperiments(
 		Experiments: experiments,
 		Cursor:      strconv.Itoa(nextCursor),
 		TotalCount:  totalCount,
+		Summary:     summary,
 	}, nil
 }
 

pkg/experiment/storage/v2/experiment.go

@@ -33,6 +33,8 @@ var (
 
 	//go:embed sql/experiment/select_experiments.sql
 	selectExperimentsSQL string
+	//go:embed sql/experiment/count_experiment.sql
+	countExperimentSQL string
 )
 
 type ExperimentStorage interface {
@@ -44,7 +46,7 @@ type ExperimentStorage interface {
 		whereParts []mysql.WherePart,
 		orders []*mysql.Order,
 		limit, offset int,
-	) ([]*proto.Experiment, int, int64, error)
+	) ([]*proto.Experiment, int, int64, *proto.ListExperimentsResponse_Summary, error)
 }
 
 type experimentStorage struct {
@@ -262,14 +264,14 @@ func (s *experimentStorage) ListExperiments(
 	whereParts []mysql.WherePart,
 	orders []*mysql.Order,
 	limit, offset int,
-) ([]*proto.Experiment, int, int64, error) {
+) ([]*proto.Experiment, int, int64, *proto.ListExperimentsResponse_Summary, error) {
 	whereSQL, whereArgs := mysql.ConstructWhereSQLString(whereParts)
 	orderBySQL := mysql.ConstructOrderBySQLString(orders)
 	limitOffsetSQL := mysql.ConstructLimitOffsetSQLString(limit, offset)
 	query := fmt.Sprintf(selectExperimentsSQL, whereSQL, orderBySQL, limitOffsetSQL)
 	rows, err := s.qe.QueryContext(ctx, query, whereArgs...)
 	if err != nil {
-		return nil, 0, 0, err
+		return nil, 0, 0, nil, err
 	}
 	defer rows.Close()
 	experiments := make([]*proto.Experiment, 0, limit)
@@ -299,27 +301,26 @@ func (s *experimentStorage) ListExperiments(
 			&mysql.JSONObject{Val: &experiment.Goals},
 		)
 		if err != nil {
-			return nil, 0, 0, err
+			return nil, 0, 0, nil, err
 		}
 		experiment.Status = proto.Experiment_Status(status)
 		experiments = append(experiments, &experiment)
 	}
 	if rows.Err() != nil {
-		return nil, 0, 0, err
+		return nil, 0, 0, nil, err
 	}
 	nextOffset := offset + len(experiments)
 	var totalCount int64
-	countQuery := fmt.Sprintf(`
-		SELECT
-			COUNT(1)
-		FROM
-			experiment
-		%s
-		`, whereSQL,
+	summary := &proto.ListExperimentsResponse_Summary{}
+	countQuery := fmt.Sprintf(countExperimentSQL, whereSQL)
+	err = s.qe.QueryRowContext(ctx, countQuery, whereArgs...).Scan(
+		&totalCount,
+		&summary.TotalWaitingCount,
+		&summary.TotalRunningCount,
+		&summary.TotalStoppedCount,
 	)
-	err = s.qe.QueryRowContext(ctx, countQuery, whereArgs...).Scan(&totalCount)
 	if err != nil {
-		return nil, 0, 0, err
+		return nil, 0, 0, nil, err
 	}
-	return experiments, nextOffset, totalCount, nil
+	return experiments, nextOffset, totalCount, summary, nil
 }

pkg/experiment/storage/v2/experiment_test.go

@@ -247,7 +247,7 @@ func TestListExperiments(t *testing.T) {
 		if p.setup != nil {
 			p.setup(storage)
 		}
-		experiments, cursor, _, err := storage.ListExperiments(
+		experiments, cursor, _, _, err := storage.ListExperiments(
 			context.Background(),
 			p.whereParts,
 			p.orders,

pkg/experiment/storage/v2/sql/experiment/count_experiment.sql

@@ -0,0 +1,8 @@
+SELECT
+    COUNT(1) AS total,
+    COALESCE(SUM(IF(status = 0, 1, 0)), 0) AS waiting,
+    COALESCE(SUM(IF(status = 1, 1, 0)), 0) AS running,
+    COALESCE(SUM(IF(status = 2 OR status = 3, 1, 0)), 0) AS stopped
+FROM
+    experiment
+%s