origin/util/blueprint/variables.go

package blueprint

import (
	"fmt"
	"strings"
)

const GetVariables = "GetVar"
const SetVariables = "SetVar"
const globalVariablesPrefix = "g_"

type GetVariablesNode struct {
	BaseExecNode
	nodeName string
	varName  string
}

type SetVariablesNode struct {
	BaseExecNode
	nodeName string
	varName  string
}

func (g *GetVariablesNode) GetName() string {
	return g.nodeName
}

func (g *GetVariablesNode) Exec() (int, error) {
	var port IPort
	if strings.HasPrefix(g.varName, globalVariablesPrefix) {
		port = g.gr.globalVariables[g.varName]
	} else {
		port = g.gr.variables[g.varName]
	}

	if port == nil {
		return -1, fmt.Errorf("variable %s not found,node name %s", g.varName, g.nodeName)
	}

	if !g.SetOutPort(0, port) {
		return -1, fmt.Errorf("set out port failed,node name %s", g.nodeName)
	}

	return 0, nil
}

func (g *GetVariablesNode) setVariableName(name string) bool {
	g.varName = name
	return true
}

func (g *SetVariablesNode) GetName() string {
	return g.nodeName
}

func (g *SetVariablesNode) Exec() (int, error) {
	port := g.GetInPort(1)
	if port == nil {
		return -1, fmt.Errorf("get in port failed,node name %s", g.nodeName)
	}

	varPort := port.Clone()
	varPort.SetValue(port)
	if strings.HasPrefix(g.varName, globalVariablesPrefix) {
		g.gr.globalVariables[g.varName] = varPort
	} else {
		g.gr.variables[g.varName] = varPort
	}

	if !g.SetOutPort(1, varPort) {
		return -1, fmt.Errorf("set out port failed,node name %s", g.nodeName)
	}

	return 0, nil
}

func (g *SetVariablesNode) setVariableName(name string) bool {
	g.varName = name
	return true
}