/
namespace.go
99 lines (87 loc) · 3.09 KB
/
namespace.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
/*
Copyright 2022 The SODA 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 util
import (
"context"
"fmt"
"time"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
corev1api "k8s.io/api/core/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
waitutil "k8s.io/apimachinery/pkg/util/wait"
clientset "k8s.io/client-go/kubernetes"
)
func NewNameSpace(c clientset.Interface, namespace string) *corev1api.Namespace {
return &corev1api.Namespace{
TypeMeta: metav1.TypeMeta{
APIVersion: corev1api.SchemeGroupVersion.String(),
Kind: "Namespace",
},
ObjectMeta: metav1.ObjectMeta{
Name: namespace,
},
}
}
func CreateNamespace(ctx context.Context, c clientset.Interface, name string) error {
ns := NewNameSpace(c, name)
_, err := c.CoreV1().Namespaces().Create(ctx, ns, metav1.CreateOptions{})
if apierrors.IsAlreadyExists(err) {
return nil
}
return err
}
func CreateNamespaceWithLabel(ctx context.Context, c clientset.Interface, namespace string, label map[string]string) error {
ns := NewNameSpace(c, namespace)
ns.Labels = label
_, err := c.CoreV1().Namespaces().Create(ctx, ns, metav1.CreateOptions{})
if apierrors.IsAlreadyExists(err) {
return nil
}
return err
}
func CreateNamespaceWithAnnotation(ctx context.Context, c clientset.Interface, namespace string, annotation map[string]string) error {
ns := NewNameSpace(c, namespace)
ns.ObjectMeta.Annotations = annotation
_, err := c.CoreV1().Namespaces().Create(ctx, ns, metav1.CreateOptions{})
if apierrors.IsAlreadyExists(err) {
return nil
}
return err
}
func GetNamespace(ctx context.Context, c clientset.Interface, namespace string) (*corev1api.Namespace, error) {
return c.CoreV1().Namespaces().Get(ctx, namespace, metav1.GetOptions{})
}
func DeleteNamespace(ctx context.Context, c clientset.Interface, namespace string, wait bool) error {
tenMinuteTimeout, cancel := context.WithTimeout(context.Background(), time.Minute*10)
defer cancel()
if err := c.CoreV1().Namespaces().Delete(ctx, namespace, metav1.DeleteOptions{}); err != nil {
return errors.Wrap(err, fmt.Sprintf("failed to delete the namespace %q", namespace))
}
if !wait {
return nil
}
return waitutil.PollImmediateInfinite(5*time.Second,
func() (bool, error) {
if _, err := c.CoreV1().Namespaces().Get(tenMinuteTimeout, namespace, metav1.GetOptions{}); err != nil {
if apierrors.IsNotFound(err) {
return true, nil
}
return false, err
}
fmt.Printf("namespace %q is still being deleted...\n", namespace)
logrus.Debugf("namespace %q is still being deleted...", namespace)
return false, nil
})
}