grim/convey
Clone
Summary
Browse
Changes
Graph
Missed a bunch of updates on unit tests for the options update
2018-01-21, Gary Kramlich
2909e17deafe
Missed a bunch of updates on unit tests for the options update
// Convey
// Copyright 2016-2018 Gary Kramlich <grim@reaperworld.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
// Package plans contains the plans structure.
package
plans
import
(
"errors"
"fmt"
"time"
"github.com/aphistic/gomol"
"bitbucket.org/rw_grim/convey/docker"
"bitbucket.org/rw_grim/convey/environment"
"bitbucket.org/rw_grim/convey/logging"
"bitbucket.org/rw_grim/convey/stages"
"bitbucket.org/rw_grim/convey/state"
"bitbucket.org/rw_grim/convey/tasks"
"bitbucket.org/rw_grim/convey/yaml"
)
var
(
// ErrNoStages is an error that can be returned from Plan.Valid.
ErrNoStages
=
errors
.
New
(
"no stages found"
)
)
// Plan is the representation of a Plan.
type
Plan
struct
{
Environment
yaml
.
StringOrSlice
`yaml:"environment"`
Stages
[]
stages
.
Stage
`yaml:"stages"`
Merge
bool
`yaml:"merge"`
}
func
(
p
*
Plan
)
setup
(
logger
*
gomol
.
LogAdapter
,
st
*
state
.
State
)
error
{
// create the network
logger
.
Info
(
"creating network..."
)
network
,
err
:=
docker
.
NewNetwork
(
st
)
if
err
!=
nil
{
logger
.
Fatal
(
"failed to create network"
)
return
err
}
st
.
Network
=
network
logger
.
Infof
(
"created network %s"
,
st
.
Network
.
Name
())
st
.
Cleanup
(
func
(
logger
*
gomol
.
LogAdapter
)
{
logger
.
Infof
(
"removing network %s"
,
st
.
Network
.
Name
())
if
err
:=
st
.
Network
.
Destroy
();
err
!=
nil
{
logger
.
Fatalf
(
"failed to remove network %s"
,
st
.
Network
.
Name
())
}
else
{
logger
.
Infof
(
"removed network %s"
,
st
.
Network
.
Name
())
}
})
// create the workspace
logger
.
Info
(
"creating workspace..."
)
ws
,
err
:=
docker
.
NewWorkspace
(
st
)
if
err
!=
nil
{
logger
.
Fatal
(
"failed to create workspace"
)
return
err
}
st
.
Workspace
=
ws
logger
.
Infof
(
"created workspace %s"
,
st
.
Workspace
.
Name
())
return
nil
}
// Execute runs the plan.
func
(
p
*
Plan
)
Execute
(
path
string
,
tasks
map
[
string
]
tasks
.
Task
,
env
[]
string
,
st
*
state
.
State
)
error
{
logger
:=
logging
.
NewAdapter
(
path
)
planEnv
:=
environment
.
Merge
(
env
,
p
.
Environment
)
if
err
:=
p
.
setup
(
logger
,
st
);
err
!=
nil
{
return
err
}
// set a flag on whether or not we've failed
var
planErr
error
for
_
,
stage
:=
range
p
.
Stages
{
start
:=
time
.
Now
()
absStageName
:=
fmt
.
Sprintf
(
"%s/%s"
,
path
,
stage
.
Name
)
stageLogger
:=
logging
.
NewAdapter
(
absStageName
)
if
stage
.
ShouldRun
(
planErr
)
{
stageLogger
.
Info
(
"stage starting"
)
err
:=
stage
.
Execute
(
absStageName
,
stageLogger
,
tasks
,
planEnv
,
st
)
if
err
!=
nil
{
stageLogger
.
Fatal
(
"stage failed"
)
planErr
=
err
continue
}
elapsed
:=
time
.
Since
(
start
)
st
.
PlanTimeout
-=
elapsed
logger
.
Debugf
(
"remaining plantime: %s"
,
st
.
PlanTimeout
)
stageLogger
.
Infof
(
"stage finished [%s]"
,
elapsed
.
Round
(
10
*
time
.
Millisecond
))
}
}
return
planErr
}
// Valid validates the plan.
func
(
p
*
Plan
)
Valid
()
error
{
if
len
(
p
.
Stages
)
==
0
{
return
ErrNoStages
}
return
nil
}