forked from awsdocs/aws-doc-sdk-examples
/
StartStopInstances.go
124 lines (108 loc) · 3.68 KB
/
StartStopInstances.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT-0
// snippet-start:[ec2.go.start_stop_instances]
package main
// snippet-start:[ec2.go.start_stop_instances.imports]
import (
"flag"
"fmt"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/awserr"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/aws/aws-sdk-go/service/ec2/ec2iface"
)
// snippet-end:[ec2.go.start_stop_instances.imports]
// StartInstance starts an Amazon EC2 instance.
// Inputs:
// svc is an Amazon EC2 service client
// instanceID is the ID of the instance
// Output:
// If success, nil
// Otherwise, an error from the call to StartInstances
func StartInstance(svc ec2iface.EC2API, instanceID *string) error {
// snippet-start:[ec2.go.start_stop_instances.start]
input := &ec2.StartInstancesInput{
InstanceIds: []*string{
instanceID,
},
DryRun: aws.Bool(true),
}
_, err := svc.StartInstances(input)
awsErr, ok := err.(awserr.Error)
if ok && awsErr.Code() == "DryRunOperation" {
// Set DryRun to be false to enable starting the instances
input.DryRun = aws.Bool(false)
_, err = svc.StartInstances(input)
// snippet-end:[ec2.go.start_stop_instances.start]
if err != nil {
return err
}
return nil
}
return err
}
// StopInstance stops an Amazon EC2 instance.
// Inputs:
// svc is an Amazon EC2 service client
// instance ID is the ID of the instance
// Output:
// If success, nil
// Otherwise, an error from the call to StopInstances
func StopInstance(svc ec2iface.EC2API, instanceID *string) error {
// snippet-start:[ec2.go.start_stop_instances.stop]
input := &ec2.StopInstancesInput{
InstanceIds: []*string{
instanceID,
},
DryRun: aws.Bool(true),
}
_, err := svc.StopInstances(input)
awsErr, ok := err.(awserr.Error)
if ok && awsErr.Code() == "DryRunOperation" {
input.DryRun = aws.Bool(false)
_, err = svc.StopInstances(input)
// snippet-end:[ec2.go.start_stop_instances.stop]
if err != nil {
return err
}
return nil
}
return err
}
func main() {
// snippet-start:[ec2.go.start_stop_instances.args]
instanceID := flag.String("i", "", "The ID of the instance to start or stop")
state := flag.String("s", "", "The state to put the instance in: START or STOP")
flag.Parse()
if (*state != "START" && *state != "STOP") || *instanceID == "" {
fmt.Println("You must supply a START or STOP state and an instance ID")
fmt.Println("(-s START | STOP -i INSTANCE-ID")
return
}
// snippet-end:[ec2.go.start_stop_instances.args]
// snippet-start:[ec2.go.start_stop_instances.session]
sess := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
svc := ec2.New(sess)
// snippet-end:[ec2.go.start_stop_instances.session]
if *state == "START" {
err := StartInstance(svc, instanceID)
if err != nil {
fmt.Println("Got an error starting instance")
fmt.Println(err)
return
}
fmt.Println("Started instance with ID " + *instanceID)
} else if *state == "STOP" {
err := StopInstance(svc, instanceID)
if err != nil {
fmt.Println("Got an error stopping the instance")
fmt.Println(err)
return
}
fmt.Println("Stopped instance with ID " + *instanceID)
}
}
// snippet-end:[ec2.go.start_stop_instances]