chore: save work

database/build/count_deployment.go

@@ -0,0 +1,32 @@
+// Copyright (c) 2023 Target Brands, Inc. All rights reserved.
+//
+// Use of this source code is governed by the LICENSE file in this repository.
+
+package build
+
+import (
+	"github.com/go-vela/types/constants"
+	"github.com/go-vela/types/library"
+	"github.com/sirupsen/logrus"
+)
+
+// CountBuildsForDeployment gets the count of builds by deployment from the database.
+func (e *engine) CountBuildsForDeployment(r *library.Repo, filters map[string]interface{}) (int64, error) {
+	e.logger.WithFields(logrus.Fields{
+		"org":  r.GetOrg(),
+		"repo": r.GetName(),
+	}).Tracef("getting count of builds for repo %s from the database", r.GetFullName())
+
+	// variable to store query results
+	var b int64
+
+	// send query to the database and store result in variable
+	err := e.client.
+		Table(constants.TableBuild).
+		Where("repo_id = ?", r.GetID()).
+		Where(filters).
+		Count(&b).
+		Error
+
+	return b, err
+}

database/build/count_org.go

@@ -21,7 +21,8 @@ func (e *engine) CountBuildsForOrg(org string, filters map[string]interface{}) (
 	// send query to the database and store result in variable
 	err := e.client.
 		Table(constants.TableBuild).
-		Joins("JOIN repos ON builds.repo_id = repos.id and repos.org = ?", org).
+		Joins("JOIN repos ON builds.repo_id = repos.id").
+		Where("repos.org = ?", org).
 		Where(filters).
 		Count(&b).
 		Error

database/build/count_repo.go

@@ -10,22 +10,23 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
-// CountBuildsForUser gets the count of builds by user ID from the database.
+// CountBuildsForRepo gets the count of builds by repo ID from the database.
 func (e *engine) CountBuildsForRepo(r *library.Repo, filters map[string]interface{}) (int64, error) {
 	e.logger.WithFields(logrus.Fields{
-		"user": u.GetName(),
-	}).Tracef("getting count of builds for user %s from the database", u.GetName())
+		"org":  r.GetOrg(),
+		"repo": r.GetName(),
+	}).Tracef("getting count of builds for repo %s from the database", r.GetFullName())
 
 	// variable to store query results
 	var b int64
 
 	// send query to the database and store result in variable
 	err := e.client.
 		Table(constants.TableBuild).
-		Where("user_id = ?", u.GetID()).
+		Where("repo_id = ?", r.GetID()).
 		Where(filters).
-		Count(&r).
+		Count(&b).
 		Error
 
-	return r, err
+	return b, err
 }