chore: add context to schedule functions

api/schedule/create.go

@@ -77,6 +77,7 @@ func CreateSchedule(c *gin.Context) {
 	// capture middleware values
 	u := user.Retrieve(c)
 	r := repo.Retrieve(c)
+	ctx := c.Request.Context()
 	allowlist := c.Value("allowlistschedule").([]string)
 	minimumFrequency := c.Value("scheduleminimumfrequency").(time.Duration)
 
@@ -145,7 +146,7 @@ func CreateSchedule(c *gin.Context) {
 	}
 
 	// send API call to capture the schedule from the database
-	dbSchedule, err := database.FromContext(c).GetScheduleForRepo(r, input.GetName())
+	dbSchedule, err := database.FromContext(c).GetScheduleForRepo(ctx, r, input.GetName())
 	if err == nil && dbSchedule.GetActive() {
 		retErr := fmt.Errorf("unable to create schedule: %s is already active", input.GetName())
 
@@ -170,7 +171,7 @@ func CreateSchedule(c *gin.Context) {
 		dbSchedule.SetActive(true)
 
 		// send API call to update the schedule
-		err = database.FromContext(c).UpdateSchedule(dbSchedule, true)
+		err = database.FromContext(c).UpdateSchedule(ctx, dbSchedule, true)
 		if err != nil {
 			retErr := fmt.Errorf("unable to set schedule %s to active: %w", dbSchedule.GetName(), err)
 
@@ -180,10 +181,10 @@ func CreateSchedule(c *gin.Context) {
 		}
 
 		// send API call to capture the updated schedule
-		s, _ = database.FromContext(c).GetScheduleForRepo(r, dbSchedule.GetName())
+		s, _ = database.FromContext(c).GetScheduleForRepo(ctx, r, dbSchedule.GetName())
 	} else {
 		// send API call to create the schedule
-		err = database.FromContext(c).CreateSchedule(s)
+		err = database.FromContext(c).CreateSchedule(ctx, s)
 		if err != nil {
 			retErr := fmt.Errorf("unable to create new schedule %s: %w", r.GetName(), err)
 
@@ -193,7 +194,7 @@ func CreateSchedule(c *gin.Context) {
 		}
 
 		// send API call to capture the created schedule
-		s, _ = database.FromContext(c).GetScheduleForRepo(r, input.GetName())
+		s, _ = database.FromContext(c).GetScheduleForRepo(ctx, r, input.GetName())
 	}
 
 	c.JSON(http.StatusCreated, s)

api/schedule/delete.go

@@ -65,6 +65,7 @@ func DeleteSchedule(c *gin.Context) {
 	r := repo.Retrieve(c)
 	u := user.Retrieve(c)
 	s := schedule.Retrieve(c)
+	ctx := c.Request.Context()
 
 	// update engine logger with API metadata
 	//
@@ -75,7 +76,7 @@ func DeleteSchedule(c *gin.Context) {
 		"user": u.GetName(),
 	}).Infof("deleting schedule %s", s.GetName())
 
-	err := database.FromContext(c).DeleteSchedule(s)
+	err := database.FromContext(c).DeleteSchedule(ctx, s)
 	if err != nil {
 		retErr := fmt.Errorf("unable to delete schedule %s: %w", s.GetName(), err)
 

api/schedule/list.go

@@ -76,6 +76,7 @@ import (
 func ListSchedules(c *gin.Context) {
 	// capture middleware values
 	r := repo.Retrieve(c)
+	ctx := c.Request.Context()
 
 	// update engine logger with API metadata
 	//
@@ -109,7 +110,7 @@ func ListSchedules(c *gin.Context) {
 	perPage = util.MaxInt(1, util.MinInt(100, perPage))
 
 	// send API call to capture the list of schedules for the repo
-	s, t, err := database.FromContext(c).ListSchedulesForRepo(r, page, perPage)
+	s, t, err := database.FromContext(c).ListSchedulesForRepo(ctx, r, page, perPage)
 	if err != nil {
 		retErr := fmt.Errorf("unable to get schedules for repo %s: %w", r.GetFullName(), err)
 

api/schedule/update.go

@@ -73,6 +73,7 @@ func UpdateSchedule(c *gin.Context) {
 	// capture middleware values
 	r := repo.Retrieve(c)
 	s := schedule.Retrieve(c)
+	ctx := c.Request.Context()
 	scheduleName := util.PathParameter(c, "schedule")
 	minimumFrequency := c.Value("scheduleminimumfrequency").(time.Duration)
 
@@ -123,7 +124,7 @@ func UpdateSchedule(c *gin.Context) {
 	}
 
 	// update the schedule within the database
-	err = database.FromContext(c).UpdateSchedule(s, true)
+	err = database.FromContext(c).UpdateSchedule(ctx, s, true)
 	if err != nil {
 		retErr := fmt.Errorf("unable to update scheduled %s: %w", scheduleName, err)
 
@@ -133,7 +134,7 @@ func UpdateSchedule(c *gin.Context) {
 	}
 
 	// capture the updated scheduled
-	s, _ = database.FromContext(c).GetScheduleForRepo(r, scheduleName)
+	s, _ = database.FromContext(c).GetScheduleForRepo(ctx, r, scheduleName)
 
 	c.JSON(http.StatusOK, s)
 }

cmd/vela-server/schedule.go

@@ -5,6 +5,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"strings"
 	"time"
@@ -26,11 +27,11 @@ import (
 
 const baseErr = "unable to schedule build"
 
-func processSchedules(compiler compiler.Engine, database database.Interface, metadata *types.Metadata, queue queue.Service, scm scm.Service) error {
+func processSchedules(ctx context.Context, compiler compiler.Engine, database database.Interface, metadata *types.Metadata, queue queue.Service, scm scm.Service) error {
 	logrus.Infof("processing active schedules to create builds")
 
 	// send API call to capture the list of active schedules
-	schedules, err := database.ListActiveSchedules()
+	schedules, err := database.ListActiveSchedules(ctx)
 	if err != nil {
 		return err
 	}
@@ -42,7 +43,7 @@ func processSchedules(compiler compiler.Engine, database database.Interface, met
 		// This is needed to ensure we are not dealing with a stale schedule since we fetch
 		// all schedules once and iterate through that list which can take a significant
 		// amount of time to get to the end of the list.
-		schedule, err := database.GetSchedule(s.GetID())
+		schedule, err := database.GetSchedule(ctx, s.GetID())
 		if err != nil {
 			logrus.WithError(err).Warnf("%s for %s", baseErr, schedule.GetName())
 
@@ -84,7 +85,7 @@ func processSchedules(compiler compiler.Engine, database database.Interface, met
 		}
 
 		if trigger && schedule.GetActive() {
-			err = processSchedule(schedule, compiler, database, metadata, queue, scm)
+			err = processSchedule(ctx, schedule, compiler, database, metadata, queue, scm)
 			if err != nil {
 				logrus.WithError(err).Warnf("%s for %s", baseErr, schedule.GetName())
 
@@ -97,7 +98,7 @@ func processSchedules(compiler compiler.Engine, database database.Interface, met
 }
 
 //nolint:funlen // ignore function length and number of statements
-func processSchedule(s *library.Schedule, compiler compiler.Engine, database database.Interface, metadata *types.Metadata, queue queue.Service, scm scm.Service) error {
+func processSchedule(ctx context.Context, s *library.Schedule, compiler compiler.Engine, database database.Interface, metadata *types.Metadata, queue queue.Service, scm scm.Service) error {
 	// sleep for 1s - 3s before processing the schedule
 	//
 	// This should prevent multiple servers from processing a schedule at the same time by
@@ -350,7 +351,7 @@ func processSchedule(s *library.Schedule, compiler compiler.Engine, database dat
 	}
 
 	// send API call to update schedule for ensuring scheduled_at field is set
-	err = database.UpdateSchedule(s, false)
+	err = database.UpdateSchedule(ctx, s, false)
 	if err != nil {
 		return fmt.Errorf("unable to update schedule %s/%s: %w", r.GetFullName(), s.GetName(), err)
 	}

cmd/vela-server/server.go

@@ -192,7 +192,7 @@ func server(c *cli.Context) error {
 			// along with a scale factor of 0.1.
 			time.Sleep(wait.Jitter(base, 0.1))
 
-			err = processSchedules(compiler, database, metadata, queue, scm)
+			err = processSchedules(ctx, compiler, database, metadata, queue, scm)
 			if err != nil {
 				logrus.WithError(err).Warn("unable to process schedules")
 			} else {

database/database.go

@@ -5,6 +5,7 @@
 package database
 
 import (
+	"context"
 	"fmt"
 	"time"
 
@@ -54,6 +55,8 @@ type (
 		client *gorm.DB
 		// engine configuration settings used in database functions
 		config *config
+		// engine context used in database functions
+		ctx context.Context
 		// sirupsen/logrus logger used in database functions
 		logger *logrus.Entry
 
@@ -85,6 +88,7 @@ func New(opts ...EngineOpt) (Interface, error) {
 	e.client = new(gorm.DB)
 	e.config = new(config)
 	e.logger = new(logrus.Entry)
+	e.ctx = context.TODO()
 
 	// apply all provided configuration options
 	for _, opt := range opts {
@@ -143,7 +147,7 @@ func New(opts ...EngineOpt) (Interface, error) {
 	}
 
 	// create database agnostic engines for resources
-	err = e.NewResources()
+	err = e.NewResources(e.ctx)
 	if err != nil {
 		return nil, err
 	}

database/opts.go

@@ -4,7 +4,10 @@
 
 package database
 
-import "time"
+import (
+	"context"
+	"time"
+)
 
 // EngineOpt represents a configuration option to initialize the database engine.
 type EngineOpt func(*engine) error
@@ -88,3 +91,12 @@ func WithSkipCreation(skipCreation bool) EngineOpt {
 		return nil
 	}
 }
+
+// WithContext sets the context in the database engine.
+func WithContext(ctx context.Context) EngineOpt {
+	return func(e *engine) error {
+		e.ctx = ctx
+
+		return nil
+	}
+}

database/resource.go

@@ -5,6 +5,7 @@
 package database
 
 import (
+	"context"
 	"github.com/go-vela/server/database/build"
 	"github.com/go-vela/server/database/hook"
 	"github.com/go-vela/server/database/log"
@@ -19,7 +20,7 @@ import (
 )
 
 // NewResources creates and returns the database agnostic engines for resources.
-func (e *engine) NewResources() error {
+func (e *engine) NewResources(ctx context.Context) error {
 	var err error
 
 	// create the database agnostic engine for builds
@@ -77,6 +78,7 @@ func (e *engine) NewResources() error {
 
 	// create the database agnostic engine for schedules
 	e.ScheduleInterface, err = schedule.New(
+		schedule.WithContext(e.ctx),
 		schedule.WithClient(e.client),
 		schedule.WithLogger(e.logger),
 		schedule.WithSkipCreation(e.config.SkipCreation),

database/resource_test.go

@@ -5,6 +5,7 @@
 package database
 
 import (
+	"context"
 	"testing"
 
 	"github.com/DATA-DOG/go-sqlmock"
@@ -94,7 +95,7 @@ func TestDatabase_Engine_NewResources(t *testing.T) {
 	// run tests
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			err := test.database.NewResources()
+			err := test.database.NewResources(context.TODO())
 
 			if test.failure {
 				if err == nil {

database/schedule/count.go

@@ -5,11 +5,12 @@
 package schedule
 
 import (
+	"context"
 	"github.com/go-vela/types/constants"
 )
 
 // CountSchedules gets the count of all schedules from the database.
-func (e *engine) CountSchedules() (int64, error) {
+func (e *engine) CountSchedules(ctx context.Context) (int64, error) {
 	e.logger.Tracef("getting count of all schedules from the database")
 
 	// variable to store query results

database/schedule/count_active.go

@@ -5,11 +5,12 @@
 package schedule
 
 import (
+	"context"
 	"github.com/go-vela/types/constants"
 )
 
 // CountActiveSchedules gets the count of all active schedules from the database.
-func (e *engine) CountActiveSchedules() (int64, error) {
+func (e *engine) CountActiveSchedules(ctx context.Context) (int64, error) {
 	e.logger.Tracef("getting count of all active schedules from the database")
 
 	// variable to store query results

database/schedule/count_active_test.go

@@ -5,6 +5,7 @@
 package schedule
 
 import (
+	"context"
 	"reflect"
 	"testing"
 
@@ -46,12 +47,12 @@ func TestSchedule_Engine_CountActiveSchedules(t *testing.T) {
 	_sqlite := testSqlite(t)
 	defer func() { _sql, _ := _sqlite.client.DB(); _sql.Close() }()
 
-	err := _sqlite.CreateSchedule(_scheduleOne)
+	err := _sqlite.CreateSchedule(context.TODO(), _scheduleOne)
 	if err != nil {
 		t.Errorf("unable to create test schedule for sqlite: %v", err)
 	}
 
-	err = _sqlite.CreateSchedule(_scheduleTwo)
+	err = _sqlite.CreateSchedule(context.TODO(), _scheduleTwo)
 	if err != nil {
 		t.Errorf("unable to create test schedule for sqlite: %v", err)
 	}
@@ -80,7 +81,7 @@ func TestSchedule_Engine_CountActiveSchedules(t *testing.T) {
 	// run tests
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			got, err := test.database.CountActiveSchedules()
+			got, err := test.database.CountActiveSchedules(context.TODO())
 
 			if test.failure {
 				if err == nil {

database/schedule/count_repo.go

@@ -5,13 +5,14 @@
 package schedule
 
 import (
+	"context"
 	"github.com/go-vela/types/constants"
 	"github.com/go-vela/types/library"
 	"github.com/sirupsen/logrus"
 )
 
 // CountSchedulesForRepo gets the count of schedules by repo ID from the database.
-func (e *engine) CountSchedulesForRepo(r *library.Repo) (int64, error) {
+func (e *engine) CountSchedulesForRepo(ctx context.Context, r *library.Repo) (int64, error) {
 	e.logger.WithFields(logrus.Fields{
 		"org":  r.GetOrg(),
 		"repo": r.GetName(),

database/schedule/count_repo_test.go

@@ -5,6 +5,7 @@
 package schedule
 
 import (
+	"context"
 	"reflect"
 	"testing"
 
@@ -50,12 +51,12 @@ func TestSchedule_Engine_CountSchedulesForRepo(t *testing.T) {
 	_sqlite := testSqlite(t)
 	defer func() { _sql, _ := _sqlite.client.DB(); _sql.Close() }()
 
-	err := _sqlite.CreateSchedule(_scheduleOne)
+	err := _sqlite.CreateSchedule(context.TODO(), _scheduleOne)
 	if err != nil {
 		t.Errorf("unable to create test schedule for sqlite: %v", err)
 	}
 
-	err = _sqlite.CreateSchedule(_scheduleTwo)
+	err = _sqlite.CreateSchedule(context.TODO(), _scheduleTwo)
 	if err != nil {
 		t.Errorf("unable to create test schedule for sqlite: %v", err)
 	}
@@ -84,7 +85,7 @@ func TestSchedule_Engine_CountSchedulesForRepo(t *testing.T) {
 	// run tests
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			got, err := test.database.CountSchedulesForRepo(_repo)
+			got, err := test.database.CountSchedulesForRepo(context.TODO(), _repo)
 
 			if test.failure {
 				if err == nil {