File: reset.go

package info (click to toggle)
deck 1.4.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,080 kB
  • sloc: makefile: 17; sh: 3
file content (101 lines) | stat: -rw-r--r-- 1,969 bytes parent folder | download | duplicates (2)
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
package reset

import (
	"context"

	"github.com/kong/deck/utils"
	"github.com/kong/go-kong/kong"
	"github.com/pkg/errors"
	"golang.org/x/sync/errgroup"
)

// Reset deletes all entities in Kong.
func Reset(state *utils.KongRawState, client *kong.Client) error {
	if state == nil {
		return errors.New("state cannot be empty")
	}

	group, ctx := errgroup.WithContext(context.Background())

	group.Go(func() error {
		// Delete routes before services
		for _, r := range state.Routes {
			err := client.Routes.Delete(ctx, r.ID)
			if err != nil {
				return err
			}
		}
		return nil
	})

	group.Go(func() error {
		for _, c := range state.Consumers {
			err := client.Consumers.Delete(ctx, c.ID)
			if err != nil {
				return err
			}
		}
		return nil
	})

	group.Go(func() error {
		// Upstreams also removes Targets
		for _, u := range state.Upstreams {
			err := client.Upstreams.Delete(ctx, u.ID)
			if err != nil {
				return err
			}
		}
		return nil
	})

	group.Go(func() error {
		for _, u := range state.CACertificates {
			err := client.CACertificates.Delete(nil, u.ID)
			if err != nil {
				return err
			}
		}
		return nil
	})

	group.Go(func() error {
		for _, p := range state.Plugins {
			// Delete global plugins explicitly since those will not
			// DELETE ON CASCADE
			if p.Consumer == nil && p.Service == nil &&
				p.Route == nil {
				err := client.Plugins.Delete(ctx, p.ID)
				if err != nil {
					return err
				}
			}
		}
		return nil
	})

	err := group.Wait()
	if err != nil {
		return err
	}

	// Routes must be delted before services can be deleted
	for _, s := range state.Services {
		err := client.Services.Delete(nil, s.ID)
		if err != nil {
			return err
		}
	}

	// Services must be deleted before certificates can be deleted
	// Certificates also removes SNIs
	for _, u := range state.Certificates {
		err := client.Certificates.Delete(nil, u.ID)
		if err != nil {
			return err
		}
	}

	// TODO handle custom entities
	return nil
}