engine: resources, lang: funcs, parser: Add panic magic

It's valuable to check your runtime values and to shut down the entire
engine in case something doesn't match. This patch adds some magic
plumbing to support a "panic" mechanism.

A new "panic" statement gets transparently converted into a panic
function and panic resource. The former errors if the input is not
empty. The latter must be present to consume the value, but doesn't
actually do anything.
This commit is contained in:
James Shubin
2023-11-28 13:49:31 -05:00
parent 64e6e686e0
commit 2cbce963b7
12 changed files with 328 additions and 17 deletions

114
engine/resources/panic.go Normal file
View File

@@ -0,0 +1,114 @@
// Mgmt
// Copyright (C) 2013-2023+ James Shubin and the project contributors
// Written by James Shubin <james@shubin.ca> and the project contributors
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package resources
import (
"context"
"fmt"
"github.com/purpleidea/mgmt/engine"
"github.com/purpleidea/mgmt/engine/traits"
"github.com/purpleidea/mgmt/lang/interfaces"
)
func init() {
// It starts with an underscore so a user can't add it manually.
engine.RegisterResource(interfaces.PanicResKind, func() engine.Res { return &PanicRes{} })
}
// PanicRes is a no-op resource that does nothing as quietly as possible. One of
// these will be added the graph if you use the panic function. (Even when it is
// in a non-panic mode.) This is possibly the simplest resource that exists, and
// in fact, everytime it is used, it will always have the same "name" value. It
// is only used so that there is a valid destination for the panic function.
type PanicRes struct {
traits.Base // add the base methods without re-implementation
init *engine.Init
}
// Default returns some sensible defaults for this resource.
func (obj *PanicRes) Default() engine.Res {
return &PanicRes{}
}
// Validate if the params passed in are valid data.
func (obj *PanicRes) Validate() error {
return nil
}
// Init runs some startup code for this resource.
func (obj *PanicRes) Init(init *engine.Init) error {
obj.init = init // save for later
return nil
}
// Cleanup is run by the engine to clean up after the resource is done.
func (obj *PanicRes) Cleanup() error {
return nil
}
// Watch is the primary listener for this resource and it outputs events.
func (obj *PanicRes) Watch(ctx context.Context) error {
obj.init.Running() // when started, notify engine that we're running
select {
case <-ctx.Done(): // closed by the engine to signal shutdown
}
//obj.init.Event() // notify engine of an event (this can block)
return nil
}
// CheckApply method for Panic resource. Does nothing, returns happy!
func (obj *PanicRes) CheckApply(context.Context, bool) (bool, error) {
return true, nil // state is always okay
}
// Cmp compares two resources and returns an error if they are not equivalent.
func (obj *PanicRes) Cmp(r engine.Res) error {
// we can only compare PanicRes to others of the same resource kind
_, ok := r.(*PanicRes)
if !ok {
return fmt.Errorf("not a %s", obj.Kind())
}
return nil
}
// UnmarshalYAML is the custom unmarshal handler for this struct. It is
// primarily useful for setting the defaults.
func (obj *PanicRes) UnmarshalYAML(unmarshal func(interface{}) error) error {
type rawRes PanicRes // indirection to avoid infinite recursion
def := obj.Default() // get the default
res, ok := def.(*PanicRes) // put in the right format
if !ok {
return fmt.Errorf("could not convert to PanicRes")
}
raw := rawRes(*res) // convert; the defaults go here
if err := unmarshal(&raw); err != nil {
return err
}
*obj = PanicRes(raw) // restore from indirection with type conversion!
return nil
}