lang: Plumb through the unified state facility

This commit is contained in:
James Shubin
2024-07-01 14:47:04 -04:00
parent d326917432
commit 4e18c9c67a
6 changed files with 49 additions and 29 deletions

View File

@@ -59,6 +59,9 @@ type Init struct {
// overall cleaner unification algorithm in place.
Strategy map[string]string
// UnifiedState stores a common representation of our unification vars.
UnifiedState *types.UnifiedState
Debug bool
Logf func(format string, v ...interface{})
}

View File

@@ -861,10 +861,11 @@ func TestUnification1(t *testing.T) {
return
}
unifier := &unification.Unifier{
AST: xast,
Solver: solver,
Debug: debug,
Logf: logf,
AST: xast,
Solver: solver,
UnifiedState: types.NewUnifiedState(),
Debug: debug,
Logf: logf,
}
err = unifier.Unify(context.TODO())

View File

@@ -38,6 +38,7 @@ import (
"strings"
"github.com/purpleidea/mgmt/lang/interfaces"
"github.com/purpleidea/mgmt/lang/types"
)
// Unifier holds all the data that the Unify function will need for it to run.
@@ -52,6 +53,9 @@ type Unifier struct {
// overall cleaner unification algorithm in place.
Strategy map[string]string
// UnifiedState stores a common representation of our unification vars.
UnifiedState *types.UnifiedState
Debug bool
Logf func(format string, v ...interface{})
}
@@ -75,14 +79,18 @@ func (obj *Unifier) Unify(ctx context.Context) error {
if obj.Solver == nil {
return fmt.Errorf("the Solver is missing")
}
if obj.UnifiedState == nil {
return fmt.Errorf("the UnifiedState table is missing")
}
if obj.Logf == nil {
return fmt.Errorf("the Logf function is missing")
}
init := &Init{
Strategy: obj.Strategy,
Logf: obj.Logf,
Debug: obj.Debug,
Strategy: obj.Strategy,
UnifiedState: obj.UnifiedState,
Logf: obj.Logf,
Debug: obj.Debug,
}
if err := obj.Solver.Init(init); err != nil {
return err