mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-08-04 16:38:34 +00:00
cb0aadf571
* feat: state rm * review feedback * fix conflict for pegomock generation code * adopt state command into allow-commands * fix conflicts * fix: state rm works on workspace * notify import/state rm discard plan file * fix lint * use repeat instead warning for re-plan * perl -pi -e 's!\* 🔁 plan file was discarded. to!🚮 A plan file was discarded. Re-plan would be required before applying.\n\n\* 🔁 To!g' server/**/* * follow main branch
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package events
|
|
|
|
import (
|
|
"github.com/runatlantis/atlantis/server/events/command"
|
|
"github.com/runatlantis/atlantis/server/logging"
|
|
"github.com/runatlantis/atlantis/server/metrics"
|
|
"github.com/uber-go/tally"
|
|
)
|
|
|
|
type InstrumentedProjectCommandBuilder struct {
|
|
ProjectCommandBuilder
|
|
Logger logging.SimpleLogging
|
|
scope tally.Scope
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) BuildApplyCommands(ctx *command.Context, comment *CommentCommand) ([]command.ProjectContext, error) {
|
|
return b.buildAndEmitStats(
|
|
"apply",
|
|
func() ([]command.ProjectContext, error) {
|
|
return b.ProjectCommandBuilder.BuildApplyCommands(ctx, comment)
|
|
},
|
|
)
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) BuildAutoplanCommands(ctx *command.Context) ([]command.ProjectContext, error) {
|
|
return b.buildAndEmitStats(
|
|
"auto plan",
|
|
func() ([]command.ProjectContext, error) {
|
|
return b.ProjectCommandBuilder.BuildAutoplanCommands(ctx)
|
|
},
|
|
)
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) BuildPlanCommands(ctx *command.Context, comment *CommentCommand) ([]command.ProjectContext, error) {
|
|
return b.buildAndEmitStats(
|
|
"plan",
|
|
func() ([]command.ProjectContext, error) {
|
|
return b.ProjectCommandBuilder.BuildPlanCommands(ctx, comment)
|
|
},
|
|
)
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) BuildImportCommands(ctx *command.Context, comment *CommentCommand) ([]command.ProjectContext, error) {
|
|
return b.buildAndEmitStats(
|
|
"import",
|
|
func() ([]command.ProjectContext, error) {
|
|
return b.ProjectCommandBuilder.BuildImportCommands(ctx, comment)
|
|
},
|
|
)
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) BuildStateRmCommands(ctx *command.Context, comment *CommentCommand) ([]command.ProjectContext, error) {
|
|
return b.buildAndEmitStats(
|
|
"state rm",
|
|
func() ([]command.ProjectContext, error) {
|
|
return b.ProjectCommandBuilder.BuildStateRmCommands(ctx, comment)
|
|
},
|
|
)
|
|
}
|
|
|
|
func (b *InstrumentedProjectCommandBuilder) buildAndEmitStats(
|
|
command string,
|
|
execute func() ([]command.ProjectContext, error),
|
|
) ([]command.ProjectContext, error) {
|
|
timer := b.scope.Timer(metrics.ExecutionTimeMetric).Start()
|
|
defer timer.Stop()
|
|
|
|
executionSuccess := b.scope.Counter(metrics.ExecutionSuccessMetric)
|
|
executionError := b.scope.Counter(metrics.ExecutionErrorMetric)
|
|
|
|
projectCmds, err := execute()
|
|
|
|
if err != nil {
|
|
executionError.Inc(1)
|
|
b.Logger.Err("Error building %s commands: %s", command, err)
|
|
} else {
|
|
executionSuccess.Inc(1)
|
|
}
|
|
|
|
return projectCmds, err
|
|
}
|