Skip to content

Commit 37314eb

Browse files
committed
Client's UpdateHTTPServers and UpdateStreamServers methods now attempt to update all servers and return a combined error
Previously these methods would return as soon as an error was encountered. Now they attempt to apply all the server updates and collect all errors encountered along the way. The reasoning behind this is that we want to apply as many of the desired changes as we possibly can and don't want any transitory errors that may affect a single server update to cause all the rest of the updates to be skipped.
1 parent b24ebb1 commit 37314eb

File tree

2 files changed

+165
-20
lines changed

2 files changed

+165
-20
lines changed

client/nginx.go

Lines changed: 42 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -808,6 +808,7 @@ func (client *NginxClient) DeleteHTTPServer(ctx context.Context, upstream string
808808
// Servers that are in the slice, but don't exist in NGINX will be added to NGINX.
809809
// Servers that aren't in the slice, but exist in NGINX, will be removed from NGINX.
810810
// Servers that are in the slice and exist in NGINX, but have different parameters, will be updated.
811+
// The client will attempt to update all servers, returning all the errors that occurred.
811812
func (client *NginxClient) UpdateHTTPServers(ctx context.Context, upstream string, servers []UpstreamServer) (added []UpstreamServer, deleted []UpstreamServer, updated []UpstreamServer, err error) {
812813
serversInNginx, err := client.GetHTTPServers(ctx, upstream)
813814
if err != nil {
@@ -824,27 +825,37 @@ func (client *NginxClient) UpdateHTTPServers(ctx context.Context, upstream strin
824825
toAdd, toDelete, toUpdate := determineUpdates(formattedServers, serversInNginx)
825826

826827
for _, server := range toAdd {
827-
err := client.AddHTTPServer(ctx, upstream, server)
828-
if err != nil {
829-
return nil, nil, nil, fmt.Errorf("failed to update servers of %v upstream: %w", upstream, err)
828+
addErr := client.AddHTTPServer(ctx, upstream, server)
829+
if addErr != nil {
830+
err = errors.Join(err, addErr)
831+
continue
830832
}
833+
added = append(added, server)
831834
}
832835

833836
for _, server := range toDelete {
834-
err := client.DeleteHTTPServer(ctx, upstream, server.Server)
835-
if err != nil {
836-
return nil, nil, nil, fmt.Errorf("failed to update servers of %v upstream: %w", upstream, err)
837+
deleteErr := client.DeleteHTTPServer(ctx, upstream, server.Server)
838+
if deleteErr != nil {
839+
err = errors.Join(err, deleteErr)
840+
continue
837841
}
842+
deleted = append(deleted, server)
838843
}
839844

840845
for _, server := range toUpdate {
841-
err := client.UpdateHTTPServer(ctx, upstream, server)
842-
if err != nil {
843-
return nil, nil, nil, fmt.Errorf("failed to update servers of %v upstream: %w", upstream, err)
846+
updateErr := client.UpdateHTTPServer(ctx, upstream, server)
847+
if updateErr != nil {
848+
err = errors.Join(err, updateErr)
849+
continue
844850
}
851+
updated = append(updated, server)
852+
}
853+
854+
if err != nil {
855+
err = fmt.Errorf("failed to update servers of %s upstream: %w", upstream, err)
845856
}
846857

847-
return toAdd, toDelete, toUpdate, nil
858+
return added, deleted, updated, err
848859
}
849860

850861
// haveSameParameters checks if a given server has the same parameters as a server already present in NGINX. Order matters.
@@ -1108,6 +1119,7 @@ func (client *NginxClient) DeleteStreamServer(ctx context.Context, upstream stri
11081119
// Servers that are in the slice, but don't exist in NGINX will be added to NGINX.
11091120
// Servers that aren't in the slice, but exist in NGINX, will be removed from NGINX.
11101121
// Servers that are in the slice and exist in NGINX, but have different parameters, will be updated.
1122+
// The client will attempt to update all servers, returning all the errors that occurred.
11111123
func (client *NginxClient) UpdateStreamServers(ctx context.Context, upstream string, servers []StreamUpstreamServer) (added []StreamUpstreamServer, deleted []StreamUpstreamServer, updated []StreamUpstreamServer, err error) {
11121124
serversInNginx, err := client.GetStreamServers(ctx, upstream)
11131125
if err != nil {
@@ -1123,27 +1135,37 @@ func (client *NginxClient) UpdateStreamServers(ctx context.Context, upstream str
11231135
toAdd, toDelete, toUpdate := determineStreamUpdates(formattedServers, serversInNginx)
11241136

11251137
for _, server := range toAdd {
1126-
err := client.AddStreamServer(ctx, upstream, server)
1127-
if err != nil {
1128-
return nil, nil, nil, fmt.Errorf("failed to update stream servers of %v upstream: %w", upstream, err)
1138+
addErr := client.AddStreamServer(ctx, upstream, server)
1139+
if addErr != nil {
1140+
err = errors.Join(err, addErr)
1141+
continue
11291142
}
1143+
added = append(added, server)
11301144
}
11311145

11321146
for _, server := range toDelete {
1133-
err := client.DeleteStreamServer(ctx, upstream, server.Server)
1134-
if err != nil {
1135-
return nil, nil, nil, fmt.Errorf("failed to update stream servers of %v upstream: %w", upstream, err)
1147+
deleteErr := client.DeleteStreamServer(ctx, upstream, server.Server)
1148+
if deleteErr != nil {
1149+
err = errors.Join(err, deleteErr)
1150+
continue
11361151
}
1152+
deleted = append(deleted, server)
11371153
}
11381154

11391155
for _, server := range toUpdate {
1140-
err := client.UpdateStreamServer(ctx, upstream, server)
1141-
if err != nil {
1142-
return nil, nil, nil, fmt.Errorf("failed to update stream servers of %v upstream: %w", upstream, err)
1156+
updateErr := client.UpdateStreamServer(ctx, upstream, server)
1157+
if updateErr != nil {
1158+
err = errors.Join(err, updateErr)
1159+
continue
11431160
}
1161+
updated = append(updated, server)
1162+
}
1163+
1164+
if err != nil {
1165+
err = fmt.Errorf("failed to update stream servers of %s upstream: %w", upstream, err)
11441166
}
11451167

1146-
return toAdd, toDelete, toUpdate, nil
1168+
return added, deleted, updated, err
11471169
}
11481170

11491171
func (client *NginxClient) getIDOfStreamServer(ctx context.Context, upstream string, name string) (int, error) {

tests/client_test.go

Lines changed: 123 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,67 @@ func TestStreamUpstreamServer(t *testing.T) {
304304
}
305305
}
306306

307+
func TestUpdateStreamUpstreamServersWithError(t *testing.T) {
308+
t.Parallel()
309+
c, err := client.NewNginxClient(helpers.GetAPIEndpoint())
310+
if err != nil {
311+
t.Fatalf("Error connecting to nginx: %v", err)
312+
}
313+
314+
maxFails := 64
315+
weight := 10
316+
maxConns := 321
317+
backup := true
318+
down := true
319+
320+
serversToUpdate := []client.StreamUpstreamServer{
321+
{
322+
ID: -1,
323+
Server: "127.0.0.1:3000",
324+
},
325+
{
326+
Server: "127.0.0.1:5000",
327+
MaxConns: &maxConns,
328+
MaxFails: &maxFails,
329+
FailTimeout: "21s",
330+
SlowStart: "12s",
331+
Weight: &weight,
332+
Backup: &backup,
333+
Down: &down,
334+
},
335+
}
336+
ctx := context.Background()
337+
338+
added, _, _, err := c.UpdateStreamServers(ctx, streamUpstream, serversToUpdate)
339+
if err == nil {
340+
t.Fatalf("Expected the client to return an error for adding a server with ID -1")
341+
}
342+
343+
if len(added) != 1 {
344+
t.Fatalf("Expected one server to be added, instead %d were added", len(added))
345+
}
346+
347+
servers, err := c.GetStreamServers(ctx, streamUpstream)
348+
if err != nil {
349+
t.Fatalf("Error getting stream servers: %v", err)
350+
}
351+
if len(servers) != 1 {
352+
t.Errorf("Too many servers")
353+
}
354+
// don't compare IDs
355+
servers[0].ID = 0
356+
357+
if !reflect.DeepEqual(serversToUpdate[1], servers[0]) {
358+
t.Errorf("Expected: %v Got: %v", serversToUpdate[1], servers[0])
359+
}
360+
361+
// remove stream upstream servers
362+
_, _, _, err = c.UpdateStreamServers(ctx, streamUpstream, []client.StreamUpstreamServer{})
363+
if err != nil {
364+
t.Errorf("Couldn't remove servers: %v", err)
365+
}
366+
}
367+
307368
//nolint:paralleltest
308369
func TestClient(t *testing.T) {
309370
c, err := client.NewNginxClient(helpers.GetAPIEndpoint())
@@ -577,6 +638,68 @@ func TestUpstreamServer(t *testing.T) {
577638
}
578639
}
579640

641+
//nolint:paralleltest
642+
func TestUpdateUpstreamServersWithError(t *testing.T) {
643+
c, err := client.NewNginxClient(helpers.GetAPIEndpoint())
644+
if err != nil {
645+
t.Fatalf("Error connecting to nginx: %v", err)
646+
}
647+
648+
maxFails := 64
649+
weight := 10
650+
maxConns := 321
651+
backup := true
652+
down := true
653+
654+
serversToUpdate := []client.UpstreamServer{
655+
{
656+
ID: -1,
657+
Server: "127.0.0.1:3000",
658+
},
659+
{
660+
Server: "127.0.0.1:2000",
661+
MaxConns: &maxConns,
662+
MaxFails: &maxFails,
663+
FailTimeout: "21s",
664+
SlowStart: "12s",
665+
Weight: &weight,
666+
Route: "test",
667+
Backup: &backup,
668+
Down: &down,
669+
},
670+
}
671+
672+
ctx := context.Background()
673+
added, _, _, err := c.UpdateHTTPServers(ctx, upstream, serversToUpdate)
674+
if err == nil {
675+
t.Fatal("Expected the client to return an error for adding a server with ID -1")
676+
}
677+
678+
if len(added) != 1 {
679+
t.Fatalf("Expected one server to be added, instead %d were added", len(added))
680+
}
681+
682+
servers, err := c.GetHTTPServers(ctx, upstream)
683+
if err != nil {
684+
t.Fatalf("Error getting HTTPServers: %v", err)
685+
}
686+
if len(servers) != 1 {
687+
t.Errorf("Too many servers")
688+
}
689+
// don't compare IDs
690+
servers[0].ID = 0
691+
692+
if !reflect.DeepEqual(serversToUpdate[1], servers[0]) {
693+
t.Errorf("Expected: %v Got: %v", serversToUpdate[1], servers[0])
694+
}
695+
696+
// remove upstream servers
697+
_, _, _, err = c.UpdateHTTPServers(ctx, upstream, []client.UpstreamServer{})
698+
if err != nil {
699+
t.Errorf("Couldn't remove servers: %v", err)
700+
}
701+
}
702+
580703
//nolint:paralleltest
581704
func TestStats(t *testing.T) {
582705
c, err := client.NewNginxClient(helpers.GetAPIEndpoint())

0 commit comments

Comments
 (0)