-
Notifications
You must be signed in to change notification settings - Fork 113
/
client.go
96 lines (82 loc) · 2.39 KB
/
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
/*
Copyright 2019 The OpenEBS Authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package kubernetes
import (
"context"
"github.com/golang/glog"
"github.com/openebs/node-disk-manager/pkg/apis"
"github.com/openebs/node-disk-manager/pkg/apis/openebs/v1alpha1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/rest"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/manager"
)
// Client is the wrapper over the k8s client that will be used by
// NDM to interface with etcd
type Client struct {
cfg *rest.Config
client client.Client
}
// New creates a new client object using the given config
func New(config *rest.Config) (Client, error) {
c := Client{
cfg: config,
}
err := c.Set()
if err != nil {
return c, err
}
return c, nil
}
// Set sets the client using the config
func (cl *Client) Set() error {
c, err := client.New(cl.cfg, client.Options{})
if err != nil {
return err
}
cl.client = c
return nil
}
// RegisterAPI registers the API scheme in the client using the manager.
// This function needs to be called only once a client object
func (cl *Client) RegisterAPI() error {
mgr, err := manager.New(cl.cfg, manager.Options{})
if err != nil {
return err
}
// Setup Scheme for all resources
if err := apis.AddToScheme(mgr.GetScheme()); err != nil {
return err
}
return nil
}
// ListBlockDevice lists the block device from etcd based on
// the filters used
func (cl *Client) ListBlockDevice(filters ...string) ([]v1alpha1.BlockDevice, error) {
bdList := &v1alpha1.BlockDeviceList{
TypeMeta: metav1.TypeMeta{
Kind: "BlockDevice",
APIVersion: "openebs.io/v1alpha1",
},
}
listOptions := &client.ListOptions{}
for _, filter := range filters {
listOptions.SetLabelSelector(filter)
}
err := cl.client.List(context.TODO(), listOptions, bdList)
if err != nil {
glog.Error("error in listing BDs. ", err)
return nil, err
}
return bdList.Items, nil
}