/
types.go
92 lines (81 loc) · 2.7 KB
/
types.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
/*
* Tencent is pleased to support the open source community by making TKEStack
* available.
*
* Copyright (C) 2012-2019 Tencent. 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
*
* https://opensource.org/licenses/Apache-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 OF ANY KIND, either express or implied. See the License for the
* specific language governing permissions and limitations under the License.
*/
package kubeadm
import (
"bytes"
"reflect"
"github.com/pkg/errors"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/serializer"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
kubeadmv1beta2 "tkestack.io/tke/pkg/platform/provider/baremetal/apis/kubeadm/v1beta2"
kubeletv1beta1 "tkestack.io/tke/pkg/platform/provider/baremetal/apis/kubelet/config/v1beta1"
kubeproxyv1alpha1 "tkestack.io/tke/pkg/platform/provider/baremetal/apis/kubeproxy/config/v1alpha1"
)
var Scheme = runtime.NewScheme()
var Codecs = serializer.NewCodecFactory(Scheme)
var localSchemeBuilder = runtime.SchemeBuilder{
kubeadmv1beta2.AddToScheme,
kubeletv1beta1.AddToScheme,
kubeproxyv1alpha1.AddToScheme,
}
var AddToScheme = localSchemeBuilder.AddToScheme
func init() {
utilruntime.Must(AddToScheme(Scheme))
}
type InitConfig struct {
InitConfiguration *kubeadmv1beta2.InitConfiguration
ClusterConfiguration *kubeadmv1beta2.ClusterConfiguration
KubeletConfiguration *kubeletv1beta1.KubeletConfiguration
KubeProxyConfiguration *kubeproxyv1alpha1.KubeProxyConfiguration
}
func (c *InitConfig) Marshal() ([]byte, error) {
buf := new(bytes.Buffer)
v := reflect.ValueOf(*c)
for i := 0; i < v.NumField(); i++ {
if v.Field(i).IsNil() {
continue
}
obj, ok := v.Field(i).Interface().(runtime.Object)
if !ok {
panic("no runtime.Object")
}
yamlData, err := MarshalToYAML(obj)
if err != nil {
return nil, err
}
buf.WriteString("---\n")
buf.Write(yamlData)
}
return buf.Bytes(), nil
}
// MarshalToYaml marshals an object into yaml.
func MarshalToYAML(obj runtime.Object) ([]byte, error) {
gvks, _, err := Scheme.ObjectKinds(obj)
if err != nil {
return nil, err
}
gv := gvks[0].GroupVersion()
const mediaType = runtime.ContentTypeYAML
info, ok := runtime.SerializerInfoForMediaType(Codecs.SupportedMediaTypes(), mediaType)
if !ok {
return []byte{}, errors.Errorf("unsupported media type %q", mediaType)
}
encoder := Codecs.EncoderForVersion(info.Serializer, gv)
return runtime.Encode(encoder, obj)
}