pulumi/pkg/resource/plan_graph.go
2017-05-18 14:51:52 -07:00

95 lines
3.1 KiB
Go

// Licensed to Pulumi Corporation ("Pulumi") under one or more
// contributor license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// Pulumi licenses this file to You 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 resource
import (
"github.com/pulumi/lumi/pkg/graph"
)
type planGraph struct {
objs []*planEdge
roots []graph.Edge
}
var _ graph.Graph = (*planGraph)(nil)
func newPlanGraph(objs []*planEdge) *planGraph {
roots := make([]graph.Edge, len(objs))
for i, root := range objs {
roots[i] = root
}
return &planGraph{
objs: objs,
roots: roots,
}
}
func (v *planGraph) Objs() []*planEdge { return v.objs }
func (v *planGraph) Roots() []graph.Edge { return v.roots }
type planVertex struct {
step *step // this vertex's step.
ins []graph.Edge // edges connecting from other vertices into this vertex.
insteps []*planEdge
outs []graph.Edge // edges connecting this vertex to other vertices.
outsteps []*planEdge
}
var _ graph.Vertex = (*planVertex)(nil)
func newPlanVertex(step *step) *planVertex {
return &planVertex{step: step}
}
func (v *planVertex) Step() *step { return v.step }
func (v *planVertex) Data() interface{} { return v.step }
func (v *planVertex) Label() string { return "" }
func (v *planVertex) Ins() []graph.Edge { return v.ins }
func (v *planVertex) InSteps() []*planEdge { return v.insteps }
func (v *planVertex) Outs() []graph.Edge { return v.outs }
func (v *planVertex) OutSteps() []*planEdge { return v.outsteps }
// connectTo creates an edge connecting the receiver vertex to the argument vertex.
func (v *planVertex) connectTo(to *planVertex) {
e := newPlanEdge(v, to)
v.outs = append(v.outs, e) // outgoing from this vertex to the other.
v.outsteps = append(v.outsteps, e)
to.ins = append(to.ins, e) // incoming from this vertex to the other.
to.insteps = append(to.insteps, e)
}
type planEdge struct {
to *planVertex // the vertex this edge connects to.
from *planVertex // the vertex this edge connects from.
}
var _ graph.Edge = (*planEdge)(nil)
func newPlanEdge(from *planVertex, to *planVertex) *planEdge {
return &planEdge{from: from, to: to}
}
func (e *planEdge) Data() interface{} { return nil }
func (e *planEdge) Label() string { return "" }
func (e *planEdge) To() graph.Vertex { return e.to }
func (e *planEdge) ToStep() *planVertex { return e.to }
func (e *planEdge) From() graph.Vertex {
if e.from == nil {
return nil
}
return e.from
}
func (e *planEdge) FromObj() *planVertex { return e.from }