149 lines
3.3 KiB
Go
149 lines
3.3 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes 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 tiller
|
|
|
|
import (
|
|
"sort"
|
|
)
|
|
|
|
// SortOrder is an ordering of Kinds.
|
|
type SortOrder []string
|
|
|
|
// InstallOrder is the order in which manifests should be installed (by Kind).
|
|
//
|
|
// Those occurring earlier in the list get installed before those occurring later in the list.
|
|
var InstallOrder SortOrder = []string{
|
|
"Namespace",
|
|
"ResourceQuota",
|
|
"LimitRange",
|
|
"Secret",
|
|
"ConfigMap",
|
|
"StorageClass",
|
|
"PersistentVolume",
|
|
"PersistentVolumeClaim",
|
|
"ServiceAccount",
|
|
"CustomResourceDefinition",
|
|
"ClusterRole",
|
|
"ClusterRoleBinding",
|
|
"Role",
|
|
"RoleBinding",
|
|
"Service",
|
|
"DaemonSet",
|
|
"Pod",
|
|
"ReplicationController",
|
|
"ReplicaSet",
|
|
"Deployment",
|
|
"StatefulSet",
|
|
"Job",
|
|
"CronJob",
|
|
"Ingress",
|
|
"APIService",
|
|
}
|
|
|
|
// UninstallOrder is the order in which manifests should be uninstalled (by Kind).
|
|
//
|
|
// Those occurring earlier in the list get uninstalled before those occurring later in the list.
|
|
var UninstallOrder SortOrder = []string{
|
|
"APIService",
|
|
"Ingress",
|
|
"Service",
|
|
"CronJob",
|
|
"Job",
|
|
"StatefulSet",
|
|
"Deployment",
|
|
"ReplicaSet",
|
|
"ReplicationController",
|
|
"Pod",
|
|
"DaemonSet",
|
|
"RoleBinding",
|
|
"Role",
|
|
"ClusterRoleBinding",
|
|
"ClusterRole",
|
|
"CustomResourceDefinition",
|
|
"ServiceAccount",
|
|
"PersistentVolumeClaim",
|
|
"PersistentVolume",
|
|
"StorageClass",
|
|
"ConfigMap",
|
|
"Secret",
|
|
"LimitRange",
|
|
"ResourceQuota",
|
|
"Namespace",
|
|
}
|
|
|
|
// sortByKind does an in-place sort of manifests by Kind.
|
|
//
|
|
// Results are sorted by 'ordering'
|
|
func sortByKind(manifests []Manifest, ordering SortOrder) []Manifest {
|
|
ks := newKindSorter(manifests, ordering)
|
|
sort.Sort(ks)
|
|
return ks.manifests
|
|
}
|
|
|
|
type kindSorter struct {
|
|
ordering map[string]int
|
|
manifests []Manifest
|
|
}
|
|
|
|
func newKindSorter(m []Manifest, s SortOrder) *kindSorter {
|
|
o := make(map[string]int, len(s))
|
|
for v, k := range s {
|
|
o[k] = v
|
|
}
|
|
|
|
return &kindSorter{
|
|
manifests: m,
|
|
ordering: o,
|
|
}
|
|
}
|
|
|
|
func (k *kindSorter) Len() int { return len(k.manifests) }
|
|
|
|
func (k *kindSorter) Swap(i, j int) { k.manifests[i], k.manifests[j] = k.manifests[j], k.manifests[i] }
|
|
|
|
func (k *kindSorter) Less(i, j int) bool {
|
|
a := k.manifests[i]
|
|
b := k.manifests[j]
|
|
first, aok := k.ordering[a.Head.Kind]
|
|
second, bok := k.ordering[b.Head.Kind]
|
|
// if same kind (including unknown) sub sort alphanumeric
|
|
if first == second {
|
|
// if both are unknown and of different kind sort by kind alphabetically
|
|
if !aok && !bok && a.Head.Kind != b.Head.Kind {
|
|
return a.Head.Kind < b.Head.Kind
|
|
}
|
|
return a.Name < b.Name
|
|
}
|
|
// unknown kind is last
|
|
if !aok {
|
|
return false
|
|
}
|
|
if !bok {
|
|
return true
|
|
}
|
|
// sort different kinds
|
|
return first < second
|
|
}
|
|
|
|
// SortByKind sorts manifests in InstallOrder
|
|
func SortByKind(manifests []Manifest) []Manifest {
|
|
ordering := InstallOrder
|
|
ks := newKindSorter(manifests, ordering)
|
|
sort.Sort(ks)
|
|
return ks.manifests
|
|
}
|