lourenco/mgmt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lang: funcs: Send empty values when appropriate

Browse Source 
I seem to have forgotten to differentiate between the empty string and
no data because the zero value for the stored result was the empty
string. This turns it into a pointer so that we don't block the function
engine if a template or one of the other patched functions sends an
empty string as the first value.
This commit is contained in:
James Shubin
2019-12-30 12:35:08 -05:00
parent ff20e67d07
commit 762c53fb8d
10 changed files with 39 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lang/funcs/core/deploy/abspath_func.go
View File

@@ -43,8 +43,8 @@ type AbsPathFunc struct {
data *interfaces.FuncData
last types.Value // last value received to use for diff
path string // the active path
result string // last calculated output
path string // the active path
result *string // last calculated output
closeChan chan struct{}
}
@@ -130,10 +130,10 @@ func (obj *AbsPathFunc) Stream() error {
}
result += obj.path
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -141,7 +141,7 @@ func (obj *AbsPathFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

10
lang/funcs/core/deploy/readfile_func.go
View File

@@ -40,8 +40,8 @@ type ReadFileFunc struct {
data *interfaces.FuncData
last types.Value // last value received to use for diff
filename string // the active filename
result string // last calculated output
filename string // the active filename
result *string // last calculated output
closeChan chan struct{}
}
@@ -139,10 +139,10 @@ func (obj *ReadFileFunc) Stream() error {
result := string(content) // convert to string
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -150,7 +150,7 @@ func (obj *ReadFileFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

10
lang/funcs/core/deploy/readfileabs_func.go
View File

@@ -40,8 +40,8 @@ type ReadFileAbsFunc struct {
data *interfaces.FuncData
last types.Value // last value received to use for diff
filename string // the active filename
result string // last calculated output
filename string // the active filename
result *string // last calculated output
closeChan chan struct{}
}
@@ -125,10 +125,10 @@ func (obj *ReadFileAbsFunc) Stream() error {
result := string(content) // convert to string
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -136,7 +136,7 @@ func (obj *ReadFileAbsFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

8
lang/funcs/core/example/vumeter_func.go
View File

@@ -45,7 +45,7 @@ type VUMeterFunc struct {
multiplier int64
peak float64
result string // last calculated output
result *string // last calculated output
closeChan chan struct{}
}
@@ -153,10 +153,10 @@ func (obj *VUMeterFunc) Stream() error {
return errwrap.Wrapf(err, "could not generate visual")
}
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -164,7 +164,7 @@ func (obj *VUMeterFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

8
lang/funcs/core/fmt/printf_func.go
View File

@@ -51,7 +51,7 @@ type PrintfFunc struct {
init *interfaces.Init
last types.Value // last value received to use for diff
result string // last calculated output
result *string // last calculated output
closeChan chan struct{}
}
@@ -224,10 +224,10 @@ func (obj *PrintfFunc) Stream() error {
return err // no errwrap needed b/c helper func
}
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -235,7 +235,7 @@ func (obj *PrintfFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

8
lang/funcs/core/os/readfile_func.go
View File

@@ -46,7 +46,7 @@ type ReadFileFunc struct {
events chan error // internal events
wg *sync.WaitGroup
result string // last calculated output
result *string // last calculated output
closeChan chan struct{}
}
@@ -195,10 +195,10 @@ func (obj *ReadFileFunc) Stream() error {
}
result := string(content) // convert to string
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -206,7 +206,7 @@ func (obj *ReadFileFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

7
lang/funcs/core/sys/cpucount_fact.go
View File

@@ -89,6 +89,8 @@ func (obj CPUCountFact) Stream() error {
closeChan := make(chan struct{}) // channel to unblock selects in goroutine
defer close(closeChan)
var once bool // did we send at least once?
// wait for kernel to poke us about new device changes on the system
wg.Add(1)
go func() {
@@ -121,6 +123,8 @@ func (obj CPUCountFact) Stream() error {
if err != nil {
obj.init.Logf("Could not get initial CPU count. Setting to zero.")
}
// TODO: would we rather error instead of sending zero?
case event, ok := <-eventChan:
if !ok {
continue
@@ -142,7 +146,7 @@ func (obj CPUCountFact) Stream() error {
return nil
}
if newCount == cpuCount {
if once && newCount == cpuCount {
continue
}
cpuCount = newCount
@@ -151,6 +155,7 @@ func (obj CPUCountFact) Stream() error {
case obj.init.Output <- &types.IntValue{
V: cpuCount,
}:
once = true
// send
case <-obj.closeChan:
return nil

8
lang/funcs/core/template_func.go
View File

@@ -68,7 +68,7 @@ type TemplateFunc struct {
init *interfaces.Init
last types.Value // last value received to use for diff
result string // last calculated output
result *string // last calculated output
closeChan chan struct{}
}
@@ -353,10 +353,10 @@ func (obj *TemplateFunc) Stream() error {
return err // no errwrap needed b/c helper func
}
if obj.result == result {
if obj.result != nil && *obj.result == result {
continue // result didn't change
}
obj.result = result // store new result
obj.result = &result // store new result
case <-obj.closeChan:
return nil
@@ -364,7 +364,7 @@ func (obj *TemplateFunc) Stream() error {
select {
case obj.init.Output <- &types.StrValue{
V: obj.result,
V: *obj.result,
}:
case <-obj.closeChan:
return nil

1
lang/funcs/simple/simple.go
View File

@@ -140,6 +140,7 @@ func (obj *WrappedFunc) Stream() error {
return errwrap.Wrapf(err, "simple function errored")
}
// TODO: do we want obj.result to be a pointer instead?
if obj.result == result {
continue // result didn't change
}

1
lang/funcs/simplepoly/simplepoly.go
View File

@@ -280,6 +280,7 @@ func (obj *WrappedFunc) Stream() error {
obj.init.Logf("Function returned with: %+v", values)
}
// TODO: do we want obj.result to be a pointer instead?
if obj.result == result {
continue // result didn't change
}