@@ -16,6 +16,8 @@ import (
16
16
"github.com/aws/aws-sdk-go/aws/session"
17
17
"github.com/aws/aws-sdk-go/service/autoscaling"
18
18
"github.com/aws/aws-sdk-go/service/ec2"
19
+
20
+ nginx "github.com/nginxinc/nginx-plus-go-client/client"
19
21
)
20
22
21
23
var configFile = flag .String ("config_path" , "/etc/nginx/aws.yaml" , "Path to the config file" )
@@ -51,7 +53,7 @@ func main() {
51
53
}
52
54
53
55
httpClient := & http.Client {Timeout : connTimeoutInSecs * time .Second }
54
- nginx , err := NewNginxClient (httpClient , cfg .APIEndpoint )
56
+ nginxClient , err := nginx . NewNginxClient (httpClient , cfg .APIEndpoint )
55
57
56
58
if err != nil {
57
59
log .Printf ("Couldn't create NGINX client: %v" , err )
@@ -66,9 +68,9 @@ func main() {
66
68
67
69
for _ , ups := range cfg .Upstreams {
68
70
if ups .Kind == "http" {
69
- err = nginx .CheckIfUpstreamExists (ups .Name )
71
+ err = nginxClient .CheckIfUpstreamExists (ups .Name )
70
72
} else {
71
- err = nginx .CheckIfStreamUpstreamExists (ups .Name )
73
+ err = nginxClient .CheckIfStreamUpstreamExists (ups .Name )
72
74
}
73
75
74
76
if err != nil {
@@ -96,16 +98,17 @@ func main() {
96
98
}
97
99
98
100
if upstream .Kind == "http" {
99
- var upsServers []UpstreamServer
101
+
102
+ var upsServers []nginx.UpstreamServer
100
103
for _ , ip := range ips {
101
104
backend := fmt .Sprintf ("%v:%v" , ip , upstream .Port )
102
- upsServers = append (upsServers , UpstreamServer {
105
+ upsServers = append (upsServers , nginx. UpstreamServer {
103
106
Server : backend ,
104
107
MaxFails : 1 ,
105
108
})
106
109
}
107
110
108
- added , removed , err := nginx .UpdateHTTPServers (upstream .Name , upsServers )
111
+ added , removed , err := nginxClient .UpdateHTTPServers (upstream .Name , upsServers )
109
112
if err != nil {
110
113
log .Printf ("Couldn't update HTTP servers in NGINX: %v" , err )
111
114
continue
@@ -115,16 +118,16 @@ func main() {
115
118
log .Printf ("Updated HTTP servers of %v; Added: %v, Removed: %v" , upstream , added , removed )
116
119
}
117
120
} else {
118
- var upsServers []StreamUpstreamServer
121
+ var upsServers []nginx. StreamUpstreamServer
119
122
for _ , ip := range ips {
120
123
backend := fmt .Sprintf ("%v:%v" , ip , upstream .Port )
121
- upsServers = append (upsServers , StreamUpstreamServer {
124
+ upsServers = append (upsServers , nginx. StreamUpstreamServer {
122
125
Server : backend ,
123
126
MaxFails : 1 ,
124
127
})
125
128
}
126
129
127
- added , removed , err := nginx .UpdateStreamServers (upstream .Name , upsServers )
130
+ added , removed , err := nginxClient .UpdateStreamServers (upstream .Name , upsServers )
128
131
if err != nil {
129
132
log .Printf ("Couldn't update Steam servers in NGINX: %v" , err )
130
133
continue
0 commit comments