enhance: supply parent context to all db calls (#1168)

* enhance: supply parent context to all db calls

* fix: lint

database/build/clean.go

@@ -26,6 +26,7 @@ func (e *engine) CleanBuilds(ctx context.Context, msg string, before int64) (int
 
 	// send query to the database
 	result := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Where("created < ?", before).
 		Where("status = 'running' OR status = 'pending'").

database/build/count.go

@@ -17,6 +17,7 @@ func (e *engine) CountBuilds(ctx context.Context) (int64, error) {
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Count(&b).
 		Error

database/build/count_deployment.go

@@ -22,6 +22,7 @@ func (e *engine) CountBuildsForDeployment(ctx context.Context, d *library.Deploy
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Where("source = ?", d.GetURL()).
 		Where(filters).

database/build/count_org.go

@@ -21,6 +21,7 @@ func (e *engine) CountBuildsForOrg(ctx context.Context, org string, filters map[
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Joins("JOIN repos ON builds.repo_id = repos.id").
 		Where("repos.org = ?", org).

database/build/count_repo.go

@@ -23,6 +23,7 @@ func (e *engine) CountBuildsForRepo(ctx context.Context, r *api.Repo, filters ma
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Where("repo_id = ?", r.GetID()).
 		Where(filters).

database/build/count_status.go

@@ -17,6 +17,7 @@ func (e *engine) CountBuildsForStatus(ctx context.Context, status string, filter
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Where("status = ?", status).
 		Where(filters).

database/build/create.go

@@ -30,7 +30,10 @@ func (e *engine) CreateBuild(ctx context.Context, b *api.Build) (*api.Build, err
 	build = build.Crop()
 
 	// send query to the database
-	err = e.client.Table(constants.TableBuild).Create(build).Error
+	err = e.client.
+		WithContext(ctx).
+		Table(constants.TableBuild).
+		Create(build).Error
 	if err != nil {
 		return nil, err
 	}

database/build/delete.go

@@ -22,6 +22,7 @@ func (e *engine) DeleteBuild(ctx context.Context, b *api.Build) error {
 
 	// send query to the database
 	return e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Delete(build).
 		Error

database/build/get.go

@@ -19,6 +19,7 @@ func (e *engine) GetBuild(ctx context.Context, id int64) (*api.Build, error) {
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/get_repo.go

@@ -25,6 +25,7 @@ func (e *engine) GetBuildForRepo(ctx context.Context, r *api.Repo, number int) (
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/index.go

@@ -47,23 +47,31 @@ func (e *engine) CreateBuildIndexes(ctx context.Context) error {
 	e.logger.Tracef("creating indexes for builds table")
 
 	// create the created column index for the builds table
-	err := e.client.Exec(CreateCreatedIndex).Error
+	err := e.client.
+		WithContext(ctx).
+		Exec(CreateCreatedIndex).Error
 	if err != nil {
 		return err
 	}
 
 	// create the repo_id column index for the builds table
-	err = e.client.Exec(CreateRepoIDIndex).Error
+	err = e.client.
+		WithContext(ctx).
+		Exec(CreateRepoIDIndex).Error
 	if err != nil {
 		return err
 	}
 
 	// create the source column index for the builds table
-	err = e.client.Exec(CreateSourceIndex).Error
+	err = e.client.
+		WithContext(ctx).
+		Exec(CreateSourceIndex).Error
 	if err != nil {
 		return err
 	}
 
 	// create the status column index for the builds table
-	return e.client.Exec(CreateStatusIndex).Error
+	return e.client.
+		WithContext(ctx).
+		Exec(CreateStatusIndex).Error
 }

database/build/last_repo.go

@@ -26,6 +26,7 @@ func (e *engine) LastBuildForRepo(ctx context.Context, r *api.Repo, branch strin
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/list.go

@@ -32,6 +32,7 @@ func (e *engine) ListBuilds(ctx context.Context) ([]*api.Build, error) {
 
 	// send query to the database and store result in variable
 	err = e.client.
+		WithContext(ctx).
 		Preload("Repo").
 		Preload("Repo.Owner").
 		Table(constants.TableBuild).

database/build/list_dashboard.go

@@ -23,7 +23,10 @@ func (e *engine) ListBuildsForDashboardRepo(ctx context.Context, r *api.Repo, br
 	b := new([]types.Build)
 	builds := []*api.Build{}
 
-	query := e.client.Table(constants.TableBuild).Where("repo_id = ?", r.GetID())
+	query := e.client.
+		WithContext(ctx).
+		Table(constants.TableBuild).
+		Where("repo_id = ?", r.GetID())
 
 	if len(branches) > 0 {
 		query = query.Where("branch IN (?)", branches)

database/build/list_org.go

@@ -40,6 +40,7 @@ func (e *engine) ListBuildsForOrg(ctx context.Context, org string, filters map[s
 	offset := perPage * (page - 1)
 
 	err = e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/list_pending_running.go

@@ -20,6 +20,7 @@ func (e *engine) ListPendingAndRunningBuilds(ctx context.Context, after string)
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Select("builds.created, builds.number, builds.status, repos.full_name").
 		InnerJoins("INNER JOIN repos ON builds.repo_id = repos.id").

database/build/list_pending_running_repo.go

@@ -20,6 +20,7 @@ func (e *engine) ListPendingAndRunningBuildsForRepo(ctx context.Context, repo *a
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/list_repo.go

@@ -41,6 +41,7 @@ func (e *engine) ListBuildsForRepo(ctx context.Context, r *api.Repo, filters map
 	offset := perPage * (page - 1)
 
 	err = e.client.
+		WithContext(ctx).
 		Table(constants.TableBuild).
 		Preload("Repo").
 		Preload("Repo.Owner").

database/build/table.go

@@ -108,11 +108,15 @@ func (e *engine) CreateBuildTable(ctx context.Context, driver string) error {
 	switch driver {
 	case constants.DriverPostgres:
 		// create the builds table for Postgres
-		return e.client.Exec(CreatePostgresTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreatePostgresTable).Error
 	case constants.DriverSqlite:
 		fallthrough
 	default:
 		// create the builds table for Sqlite
-		return e.client.Exec(CreateSqliteTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreateSqliteTable).Error
 	}
 }

database/build/update.go

@@ -30,7 +30,10 @@ func (e *engine) UpdateBuild(ctx context.Context, b *api.Build) (*api.Build, err
 	build = build.Crop()
 
 	// send query to the database
-	err = e.client.Table(constants.TableBuild).Save(build).Error
+	err = e.client.
+		WithContext(ctx).
+		Table(constants.TableBuild).
+		Save(build).Error
 	if err != nil {
 		return nil, err
 	}

database/dashboard/create.go

@@ -26,7 +26,10 @@ func (e *engine) CreateDashboard(ctx context.Context, d *api.Dashboard) (*api.Da
 	}
 
 	// send query to the database
-	result := e.client.Table(constants.TableDashboard).Create(dashboard)
+	result := e.client.
+		WithContext(ctx).
+		Table(constants.TableDashboard).
+		Create(dashboard)
 
 	return dashboard.ToAPI(), result.Error
 }

database/dashboard/delete.go

@@ -22,6 +22,7 @@ func (e *engine) DeleteDashboard(ctx context.Context, d *api.Dashboard) error {
 
 	// send query to the database
 	return e.client.
+		WithContext(ctx).
 		Table(constants.TableDashboard).
 		Delete(dashboard).
 		Error

database/dashboard/get.go

@@ -19,6 +19,7 @@ func (e *engine) GetDashboard(ctx context.Context, id string) (*api.Dashboard, e
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDashboard).
 		Where("id = ?", id).
 		Take(d).

database/dashboard/table.go

@@ -50,11 +50,15 @@ func (e *engine) CreateDashboardTable(ctx context.Context, driver string) error
 	switch driver {
 	case constants.DriverPostgres:
 		// create the dashboards table for Postgres
-		return e.client.Exec(CreatePostgresTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreatePostgresTable).Error
 	case constants.DriverSqlite:
 		fallthrough
 	default:
 		// create the dashboards table for Sqlite
-		return e.client.Exec(CreateSqliteTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreateSqliteTable).Error
 	}
 }

database/dashboard/update.go

@@ -26,7 +26,10 @@ func (e *engine) UpdateDashboard(ctx context.Context, d *api.Dashboard) (*api.Da
 	}
 
 	// send query to the database
-	err = e.client.Table(constants.TableDashboard).Save(dashboard).Error
+	err = e.client.
+		WithContext(ctx).
+		Table(constants.TableDashboard).
+		Save(dashboard).Error
 	if err != nil {
 		return nil, err
 	}

database/deployment/count.go

@@ -17,6 +17,7 @@ func (e *engine) CountDeployments(ctx context.Context) (int64, error) {
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Count(&d).
 		Error

database/deployment/count_repo.go

@@ -23,6 +23,7 @@ func (e *engine) CountDeploymentsForRepo(ctx context.Context, r *api.Repo) (int6
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Where("repo_id = ?", r.GetID()).
 		Count(&d).

database/deployment/create.go

@@ -27,7 +27,10 @@ func (e *engine) CreateDeployment(ctx context.Context, d *library.Deployment) (*
 		return nil, err
 	}
 
-	result := e.client.Table(constants.TableDeployment).Create(deployment)
+	result := e.client.
+		WithContext(ctx).
+		Table(constants.TableDeployment).
+		Create(deployment)
 
 	// send query to the database
 	return deployment.ToLibrary(d.Builds), result.Error

database/deployment/delete.go

@@ -23,6 +23,7 @@ func (e *engine) DeleteDeployment(ctx context.Context, d *library.Deployment) er
 
 	// send query to the database
 	return e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Delete(deployment).
 		Error

database/deployment/get.go

@@ -20,6 +20,7 @@ func (e *engine) GetDeployment(ctx context.Context, id int64) (*library.Deployme
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Where("id = ?", id).
 		Take(d).
@@ -40,6 +41,7 @@ func (e *engine) GetDeployment(ctx context.Context, id int64) (*library.Deployme
 
 		// send query to the database and store result in variable
 		err = e.client.
+			WithContext(ctx).
 			Table(constants.TableBuild).
 			Where("id = ?", bID).
 			Take(b).

database/deployment/get_repo.go

@@ -27,6 +27,7 @@ func (e *engine) GetDeploymentForRepo(ctx context.Context, r *api.Repo, number i
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Where("repo_id = ?", r.GetID()).
 		Where("number = ?", number).
@@ -48,6 +49,7 @@ func (e *engine) GetDeploymentForRepo(ctx context.Context, r *api.Repo, number i
 
 		// send query to the database and store result in variable
 		err = e.client.
+			WithContext(ctx).
 			Table(constants.TableBuild).
 			Where("id = ?", bID).
 			Take(b).

database/deployment/index.go

@@ -20,5 +20,7 @@ func (e *engine) CreateDeploymentIndexes(ctx context.Context) error {
 	e.logger.Tracef("creating indexes for deployments table")
 
 	// create the repo_id column index for the deployments table
-	return e.client.Exec(CreateRepoIDIndex).Error
+	return e.client.
+		WithContext(ctx).
+		Exec(CreateRepoIDIndex).Error
 }

database/deployment/list.go

@@ -21,6 +21,7 @@ func (e *engine) ListDeployments(ctx context.Context) ([]*library.Deployment, er
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Find(&d).
 		Error
@@ -45,6 +46,7 @@ func (e *engine) ListDeployments(ctx context.Context) ([]*library.Deployment, er
 
 			// send query to the database and store result in variable
 			err = e.client.
+				WithContext(ctx).
 				Table(constants.TableBuild).
 				Where("id = ?", bID).
 				Take(b).

database/deployment/list_repo.go

@@ -30,6 +30,7 @@ func (e *engine) ListDeploymentsForRepo(ctx context.Context, r *api.Repo, page,
 
 	// send query to the database and store result in variable
 	err := e.client.
+		WithContext(ctx).
 		Table(constants.TableDeployment).
 		Where("repo_id = ?", r.GetID()).
 		Order("number DESC").
@@ -58,6 +59,7 @@ func (e *engine) ListDeploymentsForRepo(ctx context.Context, r *api.Repo, page,
 
 			// send query to the database and store result in variable
 			err = e.client.
+				WithContext(ctx).
 				Table(constants.TableBuild).
 				Where("id = ?", bID).
 				Take(b).

database/deployment/table.go

@@ -62,11 +62,15 @@ func (e *engine) CreateDeploymentTable(ctx context.Context, driver string) error
 	switch driver {
 	case constants.DriverPostgres:
 		// create the deployments table for Postgres
-		return e.client.Exec(CreatePostgresTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreatePostgresTable).Error
 	case constants.DriverSqlite:
 		fallthrough
 	default:
 		// create the deployments table for Sqlite
-		return e.client.Exec(CreateSqliteTable).Error
+		return e.client.
+			WithContext(ctx).
+			Exec(CreateSqliteTable).Error
 	}
 }