Nov 13, 2024 · Nov 7, 2024 · Nov 7, 2024 · Nov 7, 2024 · Nov 7, 2024 · Nov 8, 2024
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
 return q.db.GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx, ownerID, prep)
 }

 func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.WorkspaceTable, error) {
 func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) {
 return q.db.GetWorkspacesEligibleForTransition(ctx, now)
 }

diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
 return q.GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx, ownerID, nil)
 }

 func (q *FakeQuerier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.WorkspaceTable, error) {
 func (q *FakeQuerier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) {
 q.mutex.RLock()
 defer q.mutex.RUnlock()

 workspaces := []database.WorkspaceTable{}
 workspaces := []database.GetWorkspacesEligibleForTransitionRow{}
 for _, workspace := range q.workspaces {
 build, err := q.getLatestWorkspaceBuildByWorkspaceIDNoLock(ctx, workspace.ID)
 if err != nil {
 return nil, err
 }

 if build.Transition == database.WorkspaceTransitionStart &&
 !build.Deadline.IsZero() &&
 build.Deadline.Before(now) &&
 !workspace.DormantAt.Valid {
 workspaces = append(workspaces, workspace)
 continue
 return nil, xerrors.Errorf("get workspace build by ID: %w", err)
 }

 if build.Transition == database.WorkspaceTransitionStop &&
 workspace.AutostartSchedule.Valid &&
 !workspace.DormantAt.Valid {
 workspaces = append(workspaces, workspace)
 continue
 user, err := q.getUserByIDNoLock(workspace.OwnerID)
 if err != nil {
 return nil, xerrors.Errorf("get user by ID: %w", err)
 }

 job, err := q.getProvisionerJobByIDNoLock(ctx, build.JobID)
 if err != nil {
 return nil, xerrors.Errorf("get provisioner job by ID: %w", err)
 }
 if codersdk.ProvisionerJobStatus(job.JobStatus) == codersdk.ProvisionerJobFailed {
 workspaces = append(workspaces, workspace)
 continue
 }

 template, err := q.getTemplateByIDNoLock(ctx, workspace.TemplateID)
 if err != nil {
 return nil, xerrors.Errorf("get template by ID: %w", err)
 }
 if !workspace.DormantAt.Valid && template.TimeTilDormant > 0 {
 workspaces = append(workspaces,workspace)

 ifworkspace.Deleted {
 continue
 }
 if workspace.DormantAt.Valid && template.TimeTilDormantAutoDelete > 0 {
 workspaces = append(workspaces, workspace)

 if job.JobStatus != database.ProvisionerJobStatusFailed &&
 !workspace.DormantAt.Valid &&
 build.Transition == database.WorkspaceTransitionStart &&
 (user.Status == database.UserStatusSuspended || (!build.Deadline.IsZero() && build.Deadline.Before(now))) {
 workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
 ID:   workspace.ID,
 Name: workspace.Name,
 })
 continue
 }

 user, err := q.getUserByIDNoLock(workspace.OwnerID)
 if err != nil {
 return nil, xerrors.Errorf("get user by ID: %w", err)
 if user.Status == database.UserStatusActive &&
 job.JobStatus != database.ProvisionerJobStatusFailed &&
 build.Transition == database.WorkspaceTransitionStop &&
 workspace.AutostartSchedule.Valid {
 workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
 ID:   workspace.ID,
 Name: workspace.Name,
 })
 continue
 }
 if user.Status == database.UserStatusSuspended && build.Transition == database.WorkspaceTransitionStart {
 workspaces = append(workspaces, workspace)

 if !workspace.DormantAt.Valid &&
 template.TimeTilDormant > 0 &&
 now.Sub(workspace.LastUsedAt) > time.Duration(template.TimeTilDormant) {
 workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
 ID:   workspace.ID,
 Name: workspace.Name,
 })
 continue
 }

 if workspace.DormantAt.Valid &&
 workspace.DeletingAt.Valid &&
 workspace.DeletingAt.Time.Before(now) &&
 template.TimeTilDormantAutoDelete > 0 {
 if build.Transition == database.WorkspaceTransitionDelete &&
 job.JobStatus == database.ProvisionerJobStatusFailed {
 if job.CanceledAt.Valid && now.Sub(job.CanceledAt.Time) <= 24*time.Hour {
 continue
 }

 if job.CompletedAt.Valid && now.Sub(job.CompletedAt.Time) <= 24*time.Hour {
 continue
 }
 }

 workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
 ID:   workspace.ID,
 Name: workspace.Name,
 })
 continue
 }

 if template.FailureTTL > 0 &&
 build.Transition == database.WorkspaceTransitionStart &&
 job.JobStatus == database.ProvisionerJobStatusFailed &&
 job.CompletedAt.Valid &&
 now.Sub(job.CompletedAt.Time) > time.Duration(template.FailureTTL) {
 workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
 ID:   workspace.ID,
 Name: workspace.Name,
 })
 continue
 }
 }
diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go
diff --git a/coderd/database/dbmock/dbmock.go b/coderd/database/dbmock/dbmock.go
diff --git a/coderd/database/querier.go b/coderd/database/querier.go
diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2801,7 +2801,7 @@ func (q *querier) GetWorkspacesAndAgentsByOwnerID(ctx context.Context, ownerID u
		return q.db.GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx, ownerID, prep)
		}

		func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.WorkspaceTable, error) {
		func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) {
		return q.db.GetWorkspacesEligibleForTransition(ctx, now)
		}

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -6868,60 +6868,99 @@ func (q *FakeQuerier) GetWorkspacesAndAgentsByOwnerID(ctx context.Context, owner
		return q.GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx, ownerID, nil)
		}

		func (q *FakeQuerier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.WorkspaceTable, error) {
		func (q *FakeQuerier) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) {
		q.mutex.RLock()
		defer q.mutex.RUnlock()

		workspaces := []database.WorkspaceTable{}
		workspaces := []database.GetWorkspacesEligibleForTransitionRow{}
		for _, workspace := range q.workspaces {
		build, err := q.getLatestWorkspaceBuildByWorkspaceIDNoLock(ctx, workspace.ID)
		if err != nil {
		return nil, err
		}

		if build.Transition == database.WorkspaceTransitionStart &&
		!build.Deadline.IsZero() &&
		build.Deadline.Before(now) &&
		!workspace.DormantAt.Valid {
		workspaces = append(workspaces, workspace)
		continue
		return nil, xerrors.Errorf("get workspace build by ID: %w", err)
		}

		if build.Transition == database.WorkspaceTransitionStop &&
		workspace.AutostartSchedule.Valid &&
		!workspace.DormantAt.Valid {
		workspaces = append(workspaces, workspace)
		continue
		user, err := q.getUserByIDNoLock(workspace.OwnerID)
		if err != nil {
		return nil, xerrors.Errorf("get user by ID: %w", err)
		}

		job, err := q.getProvisionerJobByIDNoLock(ctx, build.JobID)
		if err != nil {
		return nil, xerrors.Errorf("get provisioner job by ID: %w", err)
		}
		if codersdk.ProvisionerJobStatus(job.JobStatus) == codersdk.ProvisionerJobFailed {
		workspaces = append(workspaces, workspace)
		continue
		}

		template, err := q.getTemplateByIDNoLock(ctx, workspace.TemplateID)
		if err != nil {
		return nil, xerrors.Errorf("get template by ID: %w", err)
		}
		if !workspace.DormantAt.Valid && template.TimeTilDormant > 0 {
		workspaces = append(workspaces,workspace)

		ifworkspace.Deleted {
		continue
		}
		if workspace.DormantAt.Valid && template.TimeTilDormantAutoDelete > 0 {
		workspaces = append(workspaces, workspace)

		if job.JobStatus != database.ProvisionerJobStatusFailed &&
		!workspace.DormantAt.Valid &&
		build.Transition == database.WorkspaceTransitionStart &&
		(user.Status == database.UserStatusSuspended \|\| (!build.Deadline.IsZero() && build.Deadline.Before(now))) {
		workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
		ID: workspace.ID,
		Name: workspace.Name,
		})
		continue
		}

		user, err := q.getUserByIDNoLock(workspace.OwnerID)
		if err != nil {
		return nil, xerrors.Errorf("get user by ID: %w", err)
		if user.Status == database.UserStatusActive &&
		job.JobStatus != database.ProvisionerJobStatusFailed &&
		build.Transition == database.WorkspaceTransitionStop &&
		workspace.AutostartSchedule.Valid {
		workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
		ID: workspace.ID,
		Name: workspace.Name,
		})
		continue
		}
		if user.Status == database.UserStatusSuspended && build.Transition == database.WorkspaceTransitionStart {
		workspaces = append(workspaces, workspace)

		if !workspace.DormantAt.Valid &&
		template.TimeTilDormant > 0 &&
		now.Sub(workspace.LastUsedAt) > time.Duration(template.TimeTilDormant) {
		workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
		ID: workspace.ID,
		Name: workspace.Name,
		})
		continue
		}

		if workspace.DormantAt.Valid &&
		workspace.DeletingAt.Valid &&
		workspace.DeletingAt.Time.Before(now) &&
		template.TimeTilDormantAutoDelete > 0 {
		if build.Transition == database.WorkspaceTransitionDelete &&
		job.JobStatus == database.ProvisionerJobStatusFailed {
		if job.CanceledAt.Valid && now.Sub(job.CanceledAt.Time) <= 24*time.Hour {
		continue
		}

		if job.CompletedAt.Valid && now.Sub(job.CompletedAt.Time) <= 24*time.Hour {
		continue
		}
		}

		workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
		ID: workspace.ID,
		Name: workspace.Name,
		})
		continue
		}

		if template.FailureTTL > 0 &&
		build.Transition == database.WorkspaceTransitionStart &&
		job.JobStatus == database.ProvisionerJobStatusFailed &&
		job.CompletedAt.Valid &&
		now.Sub(job.CompletedAt.Time) > time.Duration(template.FailureTTL) {
		workspaces = append(workspaces, database.GetWorkspacesEligibleForTransitionRow{
		ID: workspace.ID,
		Name: workspace.Name,
		})
		continue
		}
		}
Expand Down