mirror of
https://github.com/aptly-dev/aptly.git
synced 2026-05-06 22:18:28 +00:00
Andrey Smirnov
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
//Copyright 2013 GoGraphviz Authors
|
|
//
|
|
//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 gographviz
|
|
|
|
import (
|
|
"sort"
|
|
)
|
|
|
|
//Represents a Node.
|
|
type Node struct {
|
|
Name string
|
|
Attrs Attrs
|
|
}
|
|
|
|
//Represents a set of Nodes.
|
|
type Nodes struct {
|
|
Lookup map[string]*Node
|
|
Nodes []*Node
|
|
}
|
|
|
|
//Creates a new set of Nodes.
|
|
func NewNodes() *Nodes {
|
|
return &Nodes{make(map[string]*Node), make([]*Node, 0)}
|
|
}
|
|
|
|
//Adds a Node to the set of Nodes, ammending the attributes of an already existing node.
|
|
func (this *Nodes) Add(node *Node) {
|
|
n, ok := this.Lookup[node.Name]
|
|
if ok {
|
|
n.Attrs.Ammend(node.Attrs)
|
|
return
|
|
}
|
|
this.Lookup[node.Name] = node
|
|
this.Nodes = append(this.Nodes, node)
|
|
}
|
|
|
|
//Returns a sorted list of nodes.
|
|
func (this Nodes) Sorted() []*Node {
|
|
keys := make([]string, 0, len(this.Lookup))
|
|
for key := range this.Lookup {
|
|
keys = append(keys, key)
|
|
}
|
|
sort.Strings(keys)
|
|
nodes := make([]*Node, len(keys))
|
|
for i := range keys {
|
|
nodes[i] = this.Lookup[keys[i]]
|
|
}
|
|
return nodes
|
|
}
|