-
Notifications
You must be signed in to change notification settings - Fork 64
/
host.go
66 lines (60 loc) · 1.61 KB
/
host.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
// Copyright 2021 Google LLC
//
// 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 host
import (
"fmt"
tpb "github.com/openconfig/kne/proto/topo"
"github.com/openconfig/kne/topo/node"
)
func New(nodeImpl *node.Impl) (node.Node, error) {
if nodeImpl == nil {
return nil, fmt.Errorf("nodeImpl cannot be nil")
}
if nodeImpl.Proto == nil {
return nil, fmt.Errorf("nodeImpl.Proto cannot be nil")
}
cfg := defaults(nodeImpl.Proto)
nodeImpl.Proto = cfg
n := &Node{
Impl: nodeImpl,
}
return n, nil
}
type Node struct {
*node.Impl
}
func defaults(pb *tpb.Node) *tpb.Node {
if pb.Config == nil {
pb.Config = &tpb.Config{}
}
if len(pb.GetConfig().GetCommand()) == 0 {
pb.Config.Command = []string{"/bin/sh", "-c", "sleep 2000000000000"}
}
if pb.Config.EntryCommand == "" {
pb.Config.EntryCommand = fmt.Sprintf("kubectl exec -it %s -- sh", pb.Name)
}
if pb.Config.Image == "" {
pb.Config.Image = "alpine:latest"
}
if pb.Config.ConfigPath == "" {
pb.Config.ConfigPath = "/etc"
}
if pb.Config.ConfigFile == "" {
pb.Config.ConfigFile = "config"
}
return pb
}
func init() {
node.Vendor(tpb.Vendor_HOST, New)
}