mirror of
https://github.com/coder/coder.git
synced 2025-07-18 14:17:22 +00:00
feat: persist AI task state in template imports & workspace builds (#18449)
This commit is contained in:
@ -2825,6 +2825,13 @@ func (m queryMetricsStore) UpdateTemplateScheduleByID(ctx context.Context, arg d
|
||||
return err
|
||||
}
|
||||
|
||||
func (m queryMetricsStore) UpdateTemplateVersionAITaskByJobID(ctx context.Context, arg database.UpdateTemplateVersionAITaskByJobIDParams) error {
|
||||
start := time.Now()
|
||||
r0 := m.s.UpdateTemplateVersionAITaskByJobID(ctx, arg)
|
||||
m.queryLatencies.WithLabelValues("UpdateTemplateVersionAITaskByJobID").Observe(time.Since(start).Seconds())
|
||||
return r0
|
||||
}
|
||||
|
||||
func (m queryMetricsStore) UpdateTemplateVersionByID(ctx context.Context, arg database.UpdateTemplateVersionByIDParams) error {
|
||||
start := time.Now()
|
||||
err := m.s.UpdateTemplateVersionByID(ctx, arg)
|
||||
@ -3028,6 +3035,13 @@ func (m queryMetricsStore) UpdateWorkspaceAutostart(ctx context.Context, arg dat
|
||||
return err
|
||||
}
|
||||
|
||||
func (m queryMetricsStore) UpdateWorkspaceBuildAITaskByID(ctx context.Context, arg database.UpdateWorkspaceBuildAITaskByIDParams) error {
|
||||
start := time.Now()
|
||||
r0 := m.s.UpdateWorkspaceBuildAITaskByID(ctx, arg)
|
||||
m.queryLatencies.WithLabelValues("UpdateWorkspaceBuildAITaskByID").Observe(time.Since(start).Seconds())
|
||||
return r0
|
||||
}
|
||||
|
||||
func (m queryMetricsStore) UpdateWorkspaceBuildCostByID(ctx context.Context, arg database.UpdateWorkspaceBuildCostByIDParams) error {
|
||||
start := time.Now()
|
||||
err := m.s.UpdateWorkspaceBuildCostByID(ctx, arg)
|
||||
|
Reference in New Issue
Block a user