This repository was archived by the owner on Sep 13, 2024. It is now read-only.
Repository navigation
Expand file tree
/
Copy pathecs-init.go
More file actions
134 lines (119 loc) · 3.04 KB
/
Copy pathecs-init.go
File metadata and controls
134 lines (119 loc) · 3.04 KB
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
// Copyright 2015-2016 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"
"github.com/aws/amazon-ecs-init/ecs-init/version"
log "github.com/cihub/seelog"
)
// all supported commands
const (
VERSION = "version"
PRESTART = "pre-start"
START = "start"
PRESTOP = "pre-stop"
STOP = "stop"
POSTSTOP = "post-stop"
RECACHE = "reload-cache"
)
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, engine.DefaultInitErrorExitCode)
}
log.ReplaceLogger(logger)
if args[0] == VERSION {
err := version.PrintVersion()
if err != nil {
log.Errorf("failed print version info, err: %v", err)
}
return
}
init, err := engine.New()
if err != nil {
die(err, engine.DefaultInitErrorExitCode)
}
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 {
if err, ok := err.(*engine.TerminalError); ok {
die(err, engine.TerminalFailureAgentExitCode)
}
die(err, engine.DefaultInitErrorExitCode)
}
}
type action struct {
function func() error
description string
}
func actions(engine *engine.Engine) map[string]action {
return map[string]action{
PRESTART: 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",
},
// This is a deprecated command for stopping the agent
// when using upstart jobs
PRESTOP: action{
function: engine.PreStop,
description: "Stop the ECS Agent",
},
STOP: action{
function: engine.PreStop,
description: "Stop the ECS Agent",
},
RECACHE: action{
function: engine.ReloadCache,
description: "Reload the cached image of the ECS Agent into Docker",
},
POSTSTOP: action{
function: engine.PostStop,
description: "Cleanup procedure for the ECS Agent",
},
}
}
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, exitCode int) {
log.Error(err.Error())
log.Flush()
os.Exit(exitCode)
}