-
Notifications
You must be signed in to change notification settings - Fork 119
/
ecs-init.go
100 lines (88 loc) · 2.17 KB
/
ecs-init.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// Copyright 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language governing
// permissions and limitations under the License.
package main
import (
"flag"
"fmt"
"os"
"github.com/aws/amazon-ecs-init/ecs-init/config"
"github.com/aws/amazon-ecs-init/ecs-init/engine"
log "github.com/cihub/seelog"
)
func main() {
defer log.Flush()
flag.Parse()
args := flag.Args()
if len(args) == 0 {
usage(actions(nil))
os.Exit(1)
}
logger, err := log.LoggerFromConfigAsString(config.Logger())
if err != nil {
die(err)
}
log.ReplaceLogger(logger)
init, err := engine.New()
if err != nil {
die(err)
}
log.Info(args[0])
actions := actions(init)
action, ok := actions[args[0]]
if !ok {
usage(actions)
os.Exit(1)
}
err = action.function()
if err != nil {
die(err)
}
}
type action struct {
function func() error
description string
}
func actions(engine *engine.Engine) map[string]action {
return map[string]action{
"pre-start": action{
function: engine.PreStart,
description: "Prepare the ECS Agent for starting",
},
"start": action{
function: engine.StartSupervised,
description: "Start the ECS Agent and wait for it to stop",
},
"pre-stop": action{
function: engine.PreStop,
description: "Stop the ECS Agent",
},
"reload-cache": action{
function: engine.ReloadCache,
description: "Reload the cached image of the ECS Agent into Docker",
},
}
}
func usage(actions map[string]action) {
fmt.Printf("Usage: %s ACTION\n", os.Args[0])
fmt.Println("")
fmt.Println(" Available actions:")
for command, action := range actions {
fmt.Printf(" %-15s %s\n", command, action.description)
}
fmt.Println("")
}
func die(err error) {
log.Error(err.Error())
log.Flush()
os.Exit(-1)
}