forked from rook/rook
-
Notifications
You must be signed in to change notification settings - Fork 0
/
k8sutil.go
87 lines (75 loc) · 3.03 KB
/
k8sutil.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
/*
Copyright 2016 The Rook Authors. All rights reserved.
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 k8sutil for Kubernetes helpers.
package k8sutil
import (
"crypto/md5"
"fmt"
"strings"
"github.com/coreos/pkg/capnslog"
"k8s.io/apimachinery/pkg/util/validation"
"k8s.io/client-go/kubernetes"
"k8s.io/kubernetes/pkg/util/version"
)
var logger = capnslog.NewPackageLogger("github.com/rook/rook", "op-k8sutil")
const (
// Namespace for rook
Namespace = "rook"
// DefaultNamespace for the cluster
DefaultNamespace = "default"
// DataDirVolume data dir volume
DataDirVolume = "rook-data"
// DataDir folder
DataDir = "/var/lib/rook"
// RookType for the CRD
RookType = "kubernetes.io/rook"
// PodNameEnvVar is the env variable for getting the pod name via downward api
PodNameEnvVar = "POD_NAME"
// PodNamespaceEnvVar is the env variable for getting the pod namespace via downward api
PodNamespaceEnvVar = "POD_NAMESPACE"
// NodeNameEnvVar is the env variable for getting the node via downward api
NodeNameEnvVar = "NODE_NAME"
)
// GetK8SVersion gets the version of the running K8S cluster
func GetK8SVersion(clientset kubernetes.Interface) (*version.Version, error) {
serverVersion, err := clientset.Discovery().ServerVersion()
if err != nil {
return nil, fmt.Errorf("error getting server version: %v", err)
}
// make sure the kubernetes version is parseable
index := strings.Index(serverVersion.GitVersion, "+")
if index != -1 {
newVersion := serverVersion.GitVersion[:index]
logger.Infof("returning version %s instead of %s", newVersion, serverVersion.GitVersion)
serverVersion.GitVersion = newVersion
}
return version.MustParseSemantic(serverVersion.GitVersion), nil
}
// Hash MD5 hash a given string
func Hash(s string) string {
return fmt.Sprintf("%x", md5.Sum([]byte(s)))
}
// TruncateNodeName hashes the nodeName in case it would case the name to be longer than 63 characters
// WARNING If your format and nodeName as a hash, are longer than 63 chars it won't be truncated!
// Your format alone should only be 31 chars at max because of MD5 hash being 32 chars.
// For more information, see the following resources:
// https://stackoverflow.com/a/50451893
// https://stackoverflow.com/a/32294443
func TruncateNodeName(format, nodeName string) string {
if len(nodeName)+len(fmt.Sprintf(format, "")) > validation.DNS1035LabelMaxLength {
hashed := Hash(nodeName)
logger.Infof("format and nodeName longer than %d chars, nodeName %s will be %s", validation.DNS1035LabelMaxLength, nodeName, hashed)
nodeName = hashed
}
return fmt.Sprintf(format, nodeName)
}