Add the productionOnly option

Author: noah

internal/interactor/deployment.go

@@ -39,10 +39,11 @@ type (
 	SearchDeploymentsOfUserOptions struct {
 		ListOptions
 
-		Statuses []deployment.Status
-		Owned    bool
-		From     time.Time
-		To       time.Time
+		Statuses       []deployment.Status
+		Owned          bool
+		ProductionOnly bool
+		From           time.Time
+		To             time.Time
 	}
 
 	// ListInactiveDeploymentsLessThanTimeOptions specifies the optional parameters that

internal/pkg/store/deployment.go

@@ -35,27 +35,8 @@ func (s *Store) SearchDeploymentsOfUser(ctx context.Context, u *ent.User, opt *i
 		return deployment.StatusIn(ss...)
 	}
 
-	// Search deployments that were triggered by the user.
-	if opt.Owned {
-		return s.c.Deployment.
-			Query().
-			Where(
-				deployment.And(
-					deployment.UserIDEQ(u.ID),
-					statusIn(opt.Statuses),
-					deployment.CreatedAtGTE(opt.From.UTC()),
-					deployment.CreatedAtLT(opt.To.UTC()),
-				),
-			).
-			Order(ent.Desc(deployment.FieldCreatedAt)).
-			Offset(offset(opt.Page, opt.PerPage)).
-			Limit(opt.PerPage).
-			WithRepo().
-			WithUser().
-			All(ctx)
-	}
-
-	return s.c.Deployment.
+	// Build the query searching all deployments under the accessible repositories.
+	qry := s.c.Deployment.
 		Query().
 		Where(func(s *sql.Selector) {
 			p := sql.Table(perm.Table)
@@ -78,8 +59,19 @@ func (s *Store) SearchDeploymentsOfUser(ctx context.Context, u *ent.User, opt *i
 		Offset(offset(opt.Page, opt.PerPage)).
 		Limit(opt.PerPage).
 		WithRepo().
-		WithUser().
-		All(ctx)
+		WithUser()
+
+	// Search only deployments that were triggered by the user.
+	if opt.Owned {
+		qry.Where(deployment.UserIDEQ(u.ID))
+	}
+
+	// Search only deployments for production environment.
+	if opt.ProductionOnly {
+		qry.Where(deployment.ProductionEnvironment(true))
+	}
+
+	return qry.All(ctx)
 }
 
 func (s *Store) ListInactiveDeploymentsLessThanTime(ctx context.Context, opt *i.ListInactiveDeploymentsLessThanTimeOptions) ([]*ent.Deployment, error) {

internal/pkg/store/deployment_test.go

@@ -43,6 +43,7 @@ func TestStore_SearchDeployments(t *testing.T) {
 		SetRef("main").
 		SetEnv("local").
 		SetStatus(deployment.StatusCreated).
+		SetProductionEnvironment(true).
 		SetUserID(1).
 		SetRepoID(1).
 		SaveX(ctx)
@@ -99,7 +100,29 @@ func TestStore_SearchDeployments(t *testing.T) {
 		}
 	})
 
-	t.Run("Returns waiting deployments under the accessible repositories.", func(t *testing.T) {
+	t.Run("Returns all deployment triggered by myself.", func(t *testing.T) {
+		store := NewStore(client)
+
+		res, err := store.SearchDeploymentsOfUser(ctx,
+			&ent.User{ID: 1},
+			&i.SearchDeploymentsOfUserOptions{
+				ListOptions: i.ListOptions{Page: 1, PerPage: 30},
+				Statuses:    []deployment.Status{},
+				Owned:       true,
+				From:        time.Now().UTC().Add(-time.Minute),
+				To:          time.Now().UTC(),
+			})
+		if err != nil {
+			t.Fatalf("SearchDeployments return an error: %s", err)
+		}
+
+		expected := 2
+		if len(res) != expected {
+			t.Fatalf("SearchDeployments = %v, wanted %v", res, expected)
+		}
+	})
+
+	t.Run("Returns deployments which are 'created'.", func(t *testing.T) {
 		store := NewStore(client)
 
 		res, err := store.SearchDeploymentsOfUser(ctx,
@@ -121,25 +144,25 @@ func TestStore_SearchDeployments(t *testing.T) {
 		}
 	})
 
-	t.Run("Returns all deployment triggered by myself.", func(t *testing.T) {
+	t.Run("Returns deployments for production environment.", func(t *testing.T) {
 		store := NewStore(client)
 
 		res, err := store.SearchDeploymentsOfUser(ctx,
 			&ent.User{ID: 1},
 			&i.SearchDeploymentsOfUserOptions{
 				ListOptions: i.ListOptions{Page: 1, PerPage: 30},
-				Statuses:    []deployment.Status{},
-				Owned:       true,
+				Statuses:    []deployment.Status{deployment.StatusCreated},
+				Owned:       false,
 				From:        time.Now().UTC().Add(-time.Minute),
 				To:          time.Now().UTC(),
 			})
 		if err != nil {
 			t.Fatalf("SearchDeployments return an error: %s", err)
 		}
 
-		expected := 2
+		expected := 1
 		if len(res) != expected {
-			t.Fatalf("SearchDeployments = %v, wanted %v", res, expected)
+			t.Fatalf("SearchDeployments = %v, wanted %v", len(res), expected)
 		}
 	})
 }

internal/server/api/v1/search/search.go

@@ -38,68 +38,54 @@ func (s *Search) SearchDeployments(c *gin.Context) {
 	ctx := c.Request.Context()
 
 	var (
-		statuses = c.DefaultQuery("statuses", "")
-		owned    = c.DefaultQuery("owned", "true")
-		from     = c.DefaultQuery("from", time.Now().Add(-activeDuration).Format(time.RFC3339))
-		to       = c.DefaultQuery("to", time.Now().Format(time.RFC3339))
-		page     = c.DefaultQuery("page", "1")
-		perPage  = c.DefaultQuery("per_page", "30")
-	)
-
-	var (
-		ss  = make([]deployment.Status, 0)
-		o   bool
-		f   time.Time
-		t   time.Time
-		p   int
-		pp  int
-		err error
+		statuses       = []deployment.Status{}
+		owned          bool
+		productionOnly bool
+		from, to       time.Time
+		page, perPage  int
+		err            error
 	)
 
 	// Validate query parameters.
-	for _, st := range strings.Split(statuses, ",") {
-		if st != "" {
-			ss = append(ss, deployment.Status(st))
+	for _, s := range strings.Split(c.DefaultQuery("statuses", ""), ",") {
+		if s != "" {
+			statuses = append(statuses, deployment.Status(s))
 		}
 	}
 
-	if o, err = strconv.ParseBool(owned); err != nil {
-		gb.ResponseWithError(
-			c,
-			e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "The owned must be boolean.", err),
-		)
+	if owned, err = strconv.ParseBool(c.DefaultQuery("owned", "true")); err != nil {
+		gb.ResponseWithError(c, e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "The owned must be boolean.", err))
+		return
+	}
+
+	if productionOnly, err = strconv.ParseBool(c.DefaultQuery("production_only", "false")); err != nil {
+		gb.ResponseWithError(c, e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "The production must be boolean.", err))
 		return
 	}
 
-	if f, err = time.Parse(time.RFC3339, from); err != nil {
+	if from, err = time.Parse(time.RFC3339, c.DefaultQuery("from", time.Now().Add(-activeDuration).Format(time.RFC3339))); err != nil {
 		gb.ResponseWithError(
 			c,
 			e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "Invalid format of \"from\" parameter, RFC3339 format only.", err),
 		)
 		return
 	}
 
-	if t, err = time.Parse(time.RFC3339, to); err != nil {
+	if to, err = time.Parse(time.RFC3339, c.DefaultQuery("to", time.Now().Format(time.RFC3339))); err != nil {
 		gb.ResponseWithError(
 			c,
 			e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "Invalid format of \"to\" parameter, RFC3339 format only.", err),
 		)
 		return
 	}
 
-	if p, err = strconv.Atoi(page); err != nil {
-		gb.ResponseWithError(
-			c,
-			e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "Invalid format of \"page\" parameter.", err),
-		)
+	if page, err = strconv.Atoi(c.DefaultQuery("page", "1")); err != nil {
+		gb.ResponseWithError(c, e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "The page must be number.", err))
 		return
 	}
 
-	if pp, err = strconv.Atoi(perPage); err != nil {
-		gb.ResponseWithError(
-			c,
-			e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "Invalid format of \"per_page\" parameter.", err),
-		)
+	if perPage, err = strconv.Atoi(c.DefaultQuery("per_page", "1")); err != nil {
+		gb.ResponseWithError(c, e.NewErrorWithMessage(e.ErrorCodeParameterInvalid, "The per_page must be number.", err))
 		return
 	}
 
@@ -112,11 +98,12 @@ func (s *Search) SearchDeployments(c *gin.Context) {
 	u := v.(*ent.User)
 
 	if ds, err = s.i.SearchDeploymentsOfUser(ctx, u, &i.SearchDeploymentsOfUserOptions{
-		ListOptions: i.ListOptions{Page: p, PerPage: pp},
-		Statuses:    ss,
-		Owned:       o,
-		From:        f,
-		To:          t,
+		ListOptions:    i.ListOptions{Page: page, PerPage: perPage},
+		Statuses:       statuses,
+		Owned:          owned,
+		ProductionOnly: productionOnly,
+		From:           from,
+		To:             to,
 	}); err != nil {
 		s.log.Check(gb.GetZapLogLevel(err), "Failed to search deployments.").Write(zap.Error(err))
 		gb.ResponseWithError(c, err)