Skip to content

Commit

Permalink
Improve naming and consistence
Browse files Browse the repository at this point in the history
  • Loading branch information
Albert Manya committed Jul 13, 2017
1 parent e404634 commit 6a5df5c
Show file tree
Hide file tree
Showing 7 changed files with 57 additions and 57 deletions.
14 changes: 7 additions & 7 deletions cli.go
Expand Up @@ -165,13 +165,13 @@ func (i *CLI) Setup() (bool, error) {
Hostname: i.Config.Hostname,
Context: ctx,

HardTimeout: i.Config.HardTimeout,
InitialSleep: i.Config.InitialSleep,
LogTimeout: i.Config.LogTimeout,
MaxLogLength: i.Config.MaxLogLength,
ScriptUploadTimeout: i.Config.ScriptUploadTimeout,
StartupTimeout: i.Config.StartupTimeout,
PayloadFilterScript: i.Config.FilterJobJsonScript,
HardTimeout: i.Config.HardTimeout,
InitialSleep: i.Config.InitialSleep,
LogTimeout: i.Config.LogTimeout,
MaxLogLength: i.Config.MaxLogLength,
ScriptUploadTimeout: i.Config.ScriptUploadTimeout,
StartupTimeout: i.Config.StartupTimeout,
PayloadFilterExecutable: i.Config.PayloadFilterExecutable,
}

pool := NewProcessorPool(ppc, i.BackendProvider, i.BuildScriptGenerator, i.CancellationBroadcaster)
Expand Down
6 changes: 3 additions & 3 deletions config/config.go
Expand Up @@ -166,8 +166,8 @@ var (
NewConfigDef("BuildCacheS3SecretAccessKey", &cli.StringFlag{}),

// non-config and special case flags
NewConfigDef("FilterJobJsonScript", &cli.StringFlag{
Usage: "External script which will be called to filter the json to be sent to the build script generator",
NewConfigDef("PayloadFilterExecutable", &cli.StringFlag{
Usage: "External executable which will be called to filter the json to be sent to the build script generator",
}),
NewConfigDef("SkipShutdownOnLogTimeout", &cli.BoolFlag{
Usage: "Special-case mode to aid with debugging timed out jobs",
Expand Down Expand Up @@ -347,7 +347,7 @@ type Config struct {
BuildCacheS3AccessKeyID string `config:"build-cache-s3-access-key-id"`
BuildCacheS3SecretAccessKey string `config:"build-cache-s3-secret-access-key"`

FilterJobJsonScript string `config:"filter-job-json-script"`
PayloadFilterExecutable string `config:"payload-filter-executable"`

ProviderConfig *ProviderConfig
}
Expand Down
44 changes: 22 additions & 22 deletions processor.go
Expand Up @@ -18,13 +18,13 @@ type Processor struct {
ID uuid.UUID
hostname string

hardTimeout time.Duration
initialSleep time.Duration
logTimeout time.Duration
maxLogLength int
scriptUploadTimeout time.Duration
startupTimeout time.Duration
payloadFilterScript string
hardTimeout time.Duration
initialSleep time.Duration
logTimeout time.Duration
maxLogLength int
scriptUploadTimeout time.Duration
startupTimeout time.Duration
payloadFilterExecutable string

ctx gocontext.Context
buildJobsChan <-chan Job
Expand All @@ -51,13 +51,13 @@ type Processor struct {
}

type ProcessorConfig struct {
HardTimeout time.Duration
InitialSleep time.Duration
LogTimeout time.Duration
MaxLogLength int
ScriptUploadTimeout time.Duration
StartupTimeout time.Duration
PayloadFilterScript string
HardTimeout time.Duration
InitialSleep time.Duration
LogTimeout time.Duration
MaxLogLength int
ScriptUploadTimeout time.Duration
StartupTimeout time.Duration
PayloadFilterExecutable string
}

// NewProcessor creates a new processor that will run the build jobs on the
Expand All @@ -83,13 +83,13 @@ func NewProcessor(ctx gocontext.Context, hostname string, queue JobQueue,
ID: processorUUID,
hostname: hostname,

initialSleep: config.InitialSleep,
hardTimeout: config.HardTimeout,
logTimeout: config.LogTimeout,
scriptUploadTimeout: config.ScriptUploadTimeout,
startupTimeout: config.StartupTimeout,
maxLogLength: config.MaxLogLength,
payloadFilterScript: config.PayloadFilterScript,
initialSleep: config.InitialSleep,
hardTimeout: config.HardTimeout,
logTimeout: config.LogTimeout,
scriptUploadTimeout: config.ScriptUploadTimeout,
startupTimeout: config.StartupTimeout,
maxLogLength: config.MaxLogLength,
payloadFilterExecutable: config.PayloadFilterExecutable,

ctx: ctx,
buildJobsChan: buildJobsChan,
Expand Down Expand Up @@ -198,7 +198,7 @@ func (p *Processor) process(ctx gocontext.Context, buildJob Job) {
cancellationBroadcaster: p.cancellationBroadcaster,
},
&stepTransformBuildJSON{
payloadFilterScript: p.payloadFilterScript,
payloadFilterExecutable: p.payloadFilterExecutable,
},
&stepGenerateScript{
generator: p.generator,
Expand Down
20 changes: 10 additions & 10 deletions processor_pool.go
Expand Up @@ -25,7 +25,7 @@ type ProcessorPool struct {
HardTimeout, InitialSleep, LogTimeout, ScriptUploadTimeout, StartupTimeout time.Duration
MaxLogLength int

PayloadFilterScript string
PayloadFilterExecutable string

SkipShutdownOnLogTimeout bool

Expand All @@ -44,7 +44,7 @@ type ProcessorPoolConfig struct {
HardTimeout, InitialSleep, LogTimeout, ScriptUploadTimeout, StartupTimeout time.Duration
MaxLogLength int

PayloadFilterScript string
PayloadFilterExecutable string
}

// NewProcessorPool creates a new processor pool using the given arguments.
Expand All @@ -66,7 +66,7 @@ func NewProcessorPool(ppc *ProcessorPoolConfig,
Provider: provider,
Generator: generator,
CancellationBroadcaster: cancellationBroadcaster,
PayloadFilterScript: ppc.PayloadFilterScript,
PayloadFilterExecutable: ppc.PayloadFilterExecutable,
}
}

Expand Down Expand Up @@ -184,13 +184,13 @@ func (p *ProcessorPool) runProcessor(queue JobQueue) error {
proc, err := NewProcessor(ctx, p.Hostname,
queue, p.Provider, p.Generator, p.CancellationBroadcaster,
ProcessorConfig{
HardTimeout: p.HardTimeout,
InitialSleep: p.InitialSleep,
LogTimeout: p.LogTimeout,
MaxLogLength: p.MaxLogLength,
ScriptUploadTimeout: p.ScriptUploadTimeout,
StartupTimeout: p.StartupTimeout,
PayloadFilterScript: p.PayloadFilterScript,
HardTimeout: p.HardTimeout,
InitialSleep: p.InitialSleep,
LogTimeout: p.LogTimeout,
MaxLogLength: p.MaxLogLength,
ScriptUploadTimeout: p.ScriptUploadTimeout,
StartupTimeout: p.StartupTimeout,
PayloadFilterExecutable: p.PayloadFilterExecutable,
})

if err != nil {
Expand Down
12 changes: 6 additions & 6 deletions processor_test.go
Expand Up @@ -39,12 +39,12 @@ func TestProcessor(t *testing.T) {
cancellationBroadcaster := NewCancellationBroadcaster()

processor, err := NewProcessor(ctx, "test-hostname", jobQueue, provider, generator, cancellationBroadcaster, ProcessorConfig{
HardTimeout: 2 * time.Second,
LogTimeout: time.Second,
ScriptUploadTimeout: 3 * time.Second,
StartupTimeout: 4 * time.Second,
MaxLogLength: 4500000,
PayloadFilterScript: "filter.py",
HardTimeout: 2 * time.Second,
LogTimeout: time.Second,
ScriptUploadTimeout: 3 * time.Second,
StartupTimeout: 4 * time.Second,
MaxLogLength: 4500000,
PayloadFilterExecutable: "filter.py",
})
if err != nil {
t.Error(err)
Expand Down
12 changes: 6 additions & 6 deletions step_transform_build_json.go
Expand Up @@ -12,7 +12,7 @@ import (
)

type stepTransformBuildJSON struct {
payloadFilterScript string
payloadFilterExecutable string
}

type EnvVar struct {
Expand All @@ -25,16 +25,16 @@ func (s *stepTransformBuildJSON) Run(state multistep.StateBag) multistep.StepAct
buildJob := state.Get("buildJob").(Job)
ctx := state.Get("ctx").(gocontext.Context)

if s.payloadFilterScript == "" {
context.LoggerFromContext(ctx).Info("skipping json transformation, no filter script defined")
if s.payloadFilterExecutable == "" {
context.LoggerFromContext(ctx).Info("skipping json transformation, no filter executable defined")
return multistep.ActionContinue
}

context.LoggerFromContext(ctx).Info(fmt.Sprintf("calling filter script: %s", s.payloadFilterScript))
context.LoggerFromContext(ctx).Info(fmt.Sprintf("calling filter executable: %s", s.payloadFilterExecutable))

payload := buildJob.RawPayload()

cmd := exec.Command(s.payloadFilterScript)
cmd := exec.Command(s.payloadFilterExecutable)
rawJson, err := payload.MarshalJSON()
if err != nil {
context.LoggerFromContext(ctx).Info(fmt.Sprintf("failed to marshal json: %v", err))
Expand All @@ -48,7 +48,7 @@ func (s *stepTransformBuildJSON) Run(state multistep.StateBag) multistep.StepAct

err = cmd.Run()
if err != nil {
context.LoggerFromContext(ctx).Info(fmt.Sprintf("failed to execute filter script: %v", err))
context.LoggerFromContext(ctx).Info(fmt.Sprintf("failed to run filter executable: %v", err))
return multistep.ActionContinue
}

Expand Down
6 changes: 3 additions & 3 deletions step_transform_build_json_test.go
Expand Up @@ -37,18 +37,18 @@ func setupStepTransformBuildJSON(cfg *config.ProviderConfig) (*stepTransformBuil
func TestStepTransformBuildJSON_Run(t *testing.T) {

cfg := config.ProviderConfigFromMap(map[string]string{
"FILTER_JOB_JSON_SCRIPT": "",
"PAYLOAD_FILTER_EXECUTABLE": "",
})

s, state := setupStepTransformBuildJSON(cfg)
action := s.Run(state)
assert.Equal(t, multistep.ActionContinue, action)
}

func TestStepTransformBuildJSON_RunWithScriptConfigured(t *testing.T) {
func TestStepTransformBuildJSON_RunWithExecutableConfigured(t *testing.T) {

cfg := config.ProviderConfigFromMap(map[string]string{
"FILTER_JOB_JSON_SCRIPT": "/usr/local/bin/filter.py",
"PAYLOAD_FILTER_EXECUTABLE": "/usr/local/bin/filter.py",
})

s, state := setupStepTransformBuildJSON(cfg)
Expand Down

0 comments on commit 6a5df5c

Please sign in to comment.