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 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.
package state
import (
"strings"
"labix.org/v2/mgo/bson"
"labix.org/v2/mgo/txn"
"launchpad.net/juju-core/instance"
)
// machineContainers holds the machine ids of all the containers belonging to a parent machine.
// All machines have an associated container ref doc, regardless of whether they host any containers.
type machineContainers struct {
Id string `bson:"_id"`
Children []string `bson:",omitempty"`
}
func (st *State) addChildToContainerRefOp(parentId string, childId string) txn.Op {
return txn.Op{
C: st.containerRefs.Name,
Id: parentId,
Assert: txn.DocExists,
Update: bson.D{{"$addToSet", bson.D{{"children", childId}}}},
}
}
func (st *State) insertNewContainerRefOp(machineId string, children ...string) txn.Op {
return txn.Op{
C: st.containerRefs.Name,
Id: machineId,
Assert: txn.DocMissing,
Insert: &machineContainers{
Id: machineId,
Children: children,
},
}
}
// removeContainerRefOps returns the txn.Op's necessary to remove a machine container record.
// These include removing the record itself and updating the host machine's children property.
func removeContainerRefOps(st *State, machineId string) []txn.Op {
removeRefOp := txn.Op{
C: st.containerRefs.Name,
Id: machineId,
Assert: txn.DocExists,
Remove: true,
}
// If the machine is a container, figure out its parent host.
parentId := ParentId(machineId)
if parentId == "" {
return []txn.Op{removeRefOp}
}
removeParentRefOp := txn.Op{
C: st.containerRefs.Name,
Id: parentId,
Assert: txn.DocExists,
Update: bson.D{{"$pull", bson.D{{"children", machineId}}}},
}
return []txn.Op{removeRefOp, removeParentRefOp}
}
// ParentId returns the id of the host machine if machineId a container id, or ""
// if machineId is not for a container.
func ParentId(machineId string) string {
idParts := strings.Split(machineId, "/")
if len(idParts) < 3 {
return ""
}
return strings.Join(idParts[:len(idParts)-2], "/")
}
// ContainerTypeFromId returns the container type if machineId is a container id, or ""
// if machineId is not for a container.
func ContainerTypeFromId(machineId string) instance.ContainerType {
idParts := strings.Split(machineId, "/")
if len(idParts) < 3 {
return instance.ContainerType("")
}
return instance.ContainerType(idParts[len(idParts)-2])
}
// NestingLevel returns how many levels of nesting exist for a machine id.
func NestingLevel(machineId string) int {
idParts := strings.Split(machineId, "/")
return (len(idParts) - 1) / 2
}
// TopParentId returns the id of the top level host machine for a container id.
func TopParentId(machineId string) string {
idParts := strings.Split(machineId, "/")
return idParts[0]
}
|