forked from cilium/cilium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cilium_client.go
141 lines (106 loc) · 3.59 KB
/
cilium_client.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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
package client
// This file was generated by the swagger tool.
// Editing this file might prove futile when you re-run the swagger generate command
import (
"github.com/go-openapi/runtime"
httptransport "github.com/go-openapi/runtime/client"
strfmt "github.com/go-openapi/strfmt"
"github.com/cilium/cilium/api/v1/client/daemon"
"github.com/cilium/cilium/api/v1/client/endpoint"
"github.com/cilium/cilium/api/v1/client/ipam"
"github.com/cilium/cilium/api/v1/client/policy"
"github.com/cilium/cilium/api/v1/client/service"
)
// Default cilium HTTP client.
var Default = NewHTTPClient(nil)
const (
// DefaultHost is the default Host
// found in Meta (info) section of spec file
DefaultHost string = "localhost"
// DefaultBasePath is the default BasePath
// found in Meta (info) section of spec file
DefaultBasePath string = "/v1beta"
)
// DefaultSchemes are the default schemes found in Meta (info) section of spec file
var DefaultSchemes = []string{"http"}
// NewHTTPClient creates a new cilium HTTP client.
func NewHTTPClient(formats strfmt.Registry) *Cilium {
return NewHTTPClientWithConfig(formats, nil)
}
// NewHTTPClientWithConfig creates a new cilium HTTP client,
// using a customizable transport config.
func NewHTTPClientWithConfig(formats strfmt.Registry, cfg *TransportConfig) *Cilium {
// ensure nullable parameters have default
if formats == nil {
formats = strfmt.Default
}
if cfg == nil {
cfg = DefaultTransportConfig()
}
// create transport and client
transport := httptransport.New(cfg.Host, cfg.BasePath, cfg.Schemes)
return New(transport, formats)
}
// New creates a new cilium client
func New(transport runtime.ClientTransport, formats strfmt.Registry) *Cilium {
cli := new(Cilium)
cli.Transport = transport
cli.Daemon = daemon.New(transport, formats)
cli.Endpoint = endpoint.New(transport, formats)
cli.IPAM = ipam.New(transport, formats)
cli.Policy = policy.New(transport, formats)
cli.Service = service.New(transport, formats)
return cli
}
// DefaultTransportConfig creates a TransportConfig with the
// default settings taken from the meta section of the spec file.
func DefaultTransportConfig() *TransportConfig {
return &TransportConfig{
Host: DefaultHost,
BasePath: DefaultBasePath,
Schemes: DefaultSchemes,
}
}
// TransportConfig contains the transport related info,
// found in the meta section of the spec file.
type TransportConfig struct {
Host string
BasePath string
Schemes []string
}
// WithHost overrides the default host,
// provided by the meta section of the spec file.
func (cfg *TransportConfig) WithHost(host string) *TransportConfig {
cfg.Host = host
return cfg
}
// WithBasePath overrides the default basePath,
// provided by the meta section of the spec file.
func (cfg *TransportConfig) WithBasePath(basePath string) *TransportConfig {
cfg.BasePath = basePath
return cfg
}
// WithSchemes overrides the default schemes,
// provided by the meta section of the spec file.
func (cfg *TransportConfig) WithSchemes(schemes []string) *TransportConfig {
cfg.Schemes = schemes
return cfg
}
// Cilium is a client for cilium
type Cilium struct {
Daemon *daemon.Client
Endpoint *endpoint.Client
IPAM *ipam.Client
Policy *policy.Client
Service *service.Client
Transport runtime.ClientTransport
}
// SetTransport changes the transport on the client and all its subresources
func (c *Cilium) SetTransport(transport runtime.ClientTransport) {
c.Transport = transport
c.Daemon.SetTransport(transport)
c.Endpoint.SetTransport(transport)
c.IPAM.SetTransport(transport)
c.Policy.SetTransport(transport)
c.Service.SetTransport(transport)
}