dragonheim/gagent
dragonheim/gagent
1
0
Fork 0
mirror of https://github.com/dragonheim/gagent.git synced 2025-01-18 02:36:27 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: cleaning up some old issues and stndardizing.

Browse source
This commit is contained in:
James Wells 2023-03-20 16:00:15 -07:00
parent b95d3950c5
commit 8640d42132
Signed by: jwells
GPG key ID: 73196D10B8E65666
6 changed files with 190 additions and 113 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cmd/gagent/main.go
View file

@ -1,10 +1,10 @@
package main
import (
fmt "fmt"
log "log"
http "net/http"
os "os"
strconv "strconv"
sync "sync"
autorestart "github.com/slayer/autorestart"
@ -290,7 +290,7 @@ func init() {
if config.MonitorPort != 0 {
go func() {
log.Printf("[INFO] Starting Prometheus metrics exporter on port %d\n", config.MonitorPort)
log.Fatal(http.ListenAndServe(fmt.Sprintf("%s:%d", config.ListenAddr, config.MonitorPort), nil))
log.Fatal(http.ListenAndServe(string(config.ListenAddr)+strconv.FormatInt(config.MonitorPort, 10), nil))
}()
}
}

2
internal/gstructs/gstructs.go
View file

@ -7,8 +7,8 @@ type GagentConfig struct {
Name string `hcl:"name,optional"`
Mode string `hcl:"mode,attr"`
UUID string `hcl:"uuid,optional"`
MonitorPort int `hcl:"monitorport,optional"`
ListenAddr string `hcl:"listenaddr,optional"`
MonitorPort int64 `hcl:"monitorport,optional"`
ClientPort int64 `hcl:"clientport,optional"`
RouterPort int64 `hcl:"routerport,optional"`
WorkerPort int64 `hcl:"workerport,optional"`

6
internal/worker/worker.go
View file

@ -1,8 +1,8 @@
package worker
import (
fmt "fmt"
log "log"
strconv "strconv"
sync "sync"
gstructs "github.com/dragonheim/gagent/internal/gstructs"
@ -43,13 +43,13 @@ func Main(wg *sync.WaitGroup, config gstructs.GagentConfig) {
/*
* Generate connect string for this router.
*/
connectString := fmt.Sprintf("tcp://%s:%d", config.Routers[key].RouterAddr, rport)
connectString := "tcp://" + config.Routers[key].RouterAddr + ":" + strconv.FormatInt(rport, 10)
wg.Add(1)
go getAgent(wg, config.UUID, connectString)
}
/*
* workerListener := fmt.Sprintf("tcp://%s:%d", config.ListenAddr, config.WorkerPort)
* workerListener := "tcp://" + config.ListenAddr + ":" + strconv.FormatInt(config.WorkerPort, 10)
*/
}

131
pkg/picol/commands.go
View file

@ -7,13 +7,18 @@ import (
"testing"
)
func arityErr(i *Interp, name string, argv []string) error {
/*
* incorrectArgCountError returns an error message indicating the incorrect
* number of arguments provided for a given function. It takes an interpreter
* instance 'i', the function name 'name', and a slice of argument values 'argv'.
*/
func incorrectArgCountError(i *Interpreter, name string, argv []string) error {
return fmt.Errorf("wrong number of args for %s %s", name, argv)
}
/*
needleInHaystack returns true if the string is in a slice
*/
* needleInHaystack returns true if the string is in a slice
*/
func needleInHaystack(needle string, haystack []string) bool {
for _, haystackMember := range haystack {
if haystackMember == needle {
@ -24,9 +29,9 @@ func needleInHaystack(needle string, haystack []string) bool {
}
/*
TestneedleInHaystack tests the return value of needleInHaystack
*/
func TestneedleInHaystack(t *testing.T) {
* Test_needleInHaystack tests the return value of needleInHaystack
*/
func Test_needleInHaystack(t *testing.T) {
var haystack = []string{"a", "b", "c"}
var needle = "a"
if !needleInHaystack(needle, haystack) {
@ -44,10 +49,12 @@ func TestneedleInHaystack(t *testing.T) {
}
}
// CommandMath is the math command for TCL
func CommandMath(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandMath is the math command for TCL
*/
func CommandMath(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 3 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
a, _ := strconv.Atoi(argv[1])
b, _ := strconv.Atoi(argv[2])
@ -91,28 +98,34 @@ func CommandMath(i *Interp, argv []string, pd interface{}) (string, error) {
return fmt.Sprintf("%d", c), nil
}
// CommandSet is the set command for TCL
func CommandSet(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandSet is the set command for TCL
*/
func CommandSet(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 3 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
i.SetVar(argv[1], argv[2])
i.SetVariable(argv[1], argv[2])
return argv[2], nil
}
// CommandUnset is the unset command for TCL
func CommandUnset(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandUnset is the unset command for TCL
*/
func CommandUnset(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 2 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
i.UnsetVar(argv[1])
i.UnsetVariable(argv[1])
return "", nil
}
// CommandIf is the if command for TCL
func CommandIf(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandIf is the if command for TCL
*/
func CommandIf(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 3 && len(argv) != 5 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
result, err := i.Eval(argv[1])
@ -129,10 +142,12 @@ func CommandIf(i *Interp, argv []string, pd interface{}) (string, error) {
return result, nil
}
// CommandWhile is the while command for TCL
func CommandWhile(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandWhile is the while command for TCL
*/
func CommandWhile(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 3 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
for {
@ -143,9 +158,11 @@ func CommandWhile(i *Interp, argv []string, pd interface{}) (string, error) {
if r, _ := strconv.Atoi(result); r != 0 {
result, err := i.Eval(argv[2])
switch err {
case errContinue, nil:
//pass
case errBreak:
case ErrContinue, nil:
/*
* pass
*/
case ErrBreak:
return result, nil
default:
return result, err
@ -156,22 +173,26 @@ func CommandWhile(i *Interp, argv []string, pd interface{}) (string, error) {
}
}
// CommandRetCodes is a function to get the return codes for TCL
func CommandRetCodes(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandRetCodes is a function to get the return codes for TCL
*/
func CommandRetCodes(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 1 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
switch argv[0] {
case "break":
return "", errBreak
return "", ErrBreak
case "continue":
return "", errContinue
return "", ErrContinue
}
return "", nil
}
// CommandCallProc is a function to call proc commands for TCL
func CommandCallProc(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandCallProc is a function to call proc commands for TCL
*/
func CommandCallProc(i *Interpreter, argv []string, pd interface{}) (string, error) {
var x []string
if pd, ok := pd.([]string); ok {
@ -180,7 +201,7 @@ func CommandCallProc(i *Interp, argv []string, pd interface{}) (string, error) {
return "", nil
}
i.callframe = &CallFrame{vars: make(map[string]Var), parent: i.callframe}
i.callframe = &CallFrame{vars: make(map[string]Variable), parent: i.callframe}
defer func() { i.callframe = i.callframe.parent }() // remove the called proc callframe
arity := 0
@ -189,7 +210,7 @@ func CommandCallProc(i *Interp, argv []string, pd interface{}) (string, error) {
continue
}
arity++
i.SetVar(arg, argv[arity])
i.SetVariable(arg, argv[arity])
}
if arity != len(argv)-1 {
@ -198,42 +219,50 @@ func CommandCallProc(i *Interp, argv []string, pd interface{}) (string, error) {
body := x[1]
result, err := i.Eval(body)
if err == errReturn {
if err == ErrReturn {
err = nil
}
return result, err
}
// CommandProc is a function to register proc commands for TCL
func CommandProc(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandProc is a function to register proc commands for TCL
*/
func CommandProc(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 4 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
return "", i.RegisterCommand(argv[1], CommandCallProc, []string{argv[2], argv[3]})
}
// CommandReturn is a function to register return codes for commands for TCL
func CommandReturn(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandReturn is a function to register return codes for commands for TCL
*/
func CommandReturn(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 1 && len(argv) != 2 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
var r string
if len(argv) == 2 {
r = argv[1]
}
return r, errReturn
return r, ErrReturn
}
// CommandError is a function to return error codes for commands for TCL
func CommandError(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandError is a function to return error codes for commands for TCL
*/
func CommandError(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 1 && len(argv) != 2 {
return "", arityErr(i, argv[0], argv)
return "", incorrectArgCountError(i, argv[0], argv)
}
return "", fmt.Errorf(argv[1])
}
// CommandPuts is a function to print strings for TCL
func CommandPuts(i *Interp, argv []string, pd interface{}) (string, error) {
/*
* CommandPuts is a function to print strings for TCL
*/
func CommandPuts(i *Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 2 {
return "", fmt.Errorf("wrong number of args for %s %s", argv[0], argv)
}
@ -241,8 +270,10 @@ func CommandPuts(i *Interp, argv []string, pd interface{}) (string, error) {
return "", nil
}
// RegisterCoreCommands is a callable to register TCL commands.
func (i *Interp) RegisterCoreCommands() {
/*
* RegisterCoreCommands is a callable to register TCL commands.
*/
func (i *Interpreter) RegisterCoreCommands() {
name := [...]string{"+", "-", "*", "/", ">", ">=", "<", "<=", "==", "!="}
for _, n := range name {
_ = i.RegisterCommand(n, CommandMath, nil)

154
pkg/picol/picol.go
View file

@ -6,38 +6,66 @@ import (
"strings"
)
/*
* Error variables
*/
var (
errReturn = errors.New("RETURN")
errBreak = errors.New("BREAK")
errContinue = errors.New("CONTINUE")
ErrReturn = errors.New("RETURN")
ErrBreak = errors.New("BREAK")
ErrContinue = errors.New("CONTINUE")
)
type Var string
type CmdFunc func(i *Interp, argv []string, privdata interface{}) (string, error)
type Cmd struct {
fn CmdFunc
/*
* Variable type
*/
type Variable string
/*
* CommandFunc type
*/
type CommandFunc func(interp *Interpreter, argv []string, privdata interface{}) (string, error)
/*
* Command structure
*/
type Command struct {
fn CommandFunc
privdata interface{}
}
/*
* CallFrame structure
*/
type CallFrame struct {
vars map[string]Var
vars map[string]Variable
parent *CallFrame
}
type Interp struct {
/*
* Interpreter structure
*/
type Interpreter struct {
level int
callframe *CallFrame
commands map[string]Cmd
commands map[string]Command
}
func InitInterp() *Interp {
return &Interp{
/*
* NewInterpreter initializes a new Interpreter
*/
func NewInterpreter() *Interpreter {
return &Interpreter{
level: 0,
callframe: &CallFrame{vars: make(map[string]Var)},
commands: make(map[string]Cmd),
callframe: &CallFrame{vars: make(map[string]Variable)},
commands: make(map[string]Command),
}
}
func (i *Interp) Var(name string) (Var, bool) {
for frame := i.callframe; frame != nil; frame = frame.parent {
/*
* Variable retrieves a variable's value
*/
func (interp *Interpreter) Variable(name string) (Variable, bool) {
for frame := interp.callframe; frame != nil; frame = frame.parent {
v, ok := frame.vars[name]
if ok {
return v, ok
@ -45,94 +73,112 @@ func (i *Interp) Var(name string) (Var, bool) {
}
return "", false
}
func (i *Interp) SetVar(name, val string) {
i.callframe.vars[name] = Var(val)
/*
* SetVariable sets a variable's value
*/
func (interp *Interpreter) SetVariable(name, val string) {
interp.callframe.vars[name] = Variable(val)
}
func (i *Interp) UnsetVar(name string) {
delete(i.callframe.vars, name)
/*
* UnsetVariable removes a variable
*/
func (interp *Interpreter) UnsetVariable(name string) {
delete(interp.callframe.vars, name)
}
func (i *Interp) Command(name string) *Cmd {
v, ok := i.commands[name]
/*
* Command retrieves a command
*/
func (interp *Interpreter) Command(name string) *Command {
v, ok := interp.commands[name]
if !ok {
return nil
}
return &v
}
func (i *Interp) RegisterCommand(name string, fn CmdFunc, privdata interface{}) error {
c := i.Command(name)
if c != nil {
/*
* RegisterCommand registers a new command
*/
func (interp *Interpreter) RegisterCommand(name string, fn CommandFunc, privdata interface{}) error {
cmd := interp.Command(name)
if cmd != nil {
return fmt.Errorf("Command '%s' already defined", name)
}
i.commands[name] = Cmd{fn, privdata}
interp.commands[name] = Command{fn, privdata}
return nil
}
/* EVAL! */
func (i *Interp) Eval(t string) (string, error) {
p := initParser(t)
/*
* Eval evaluates a script
*/
func (interp *Interpreter) Eval(script string) (string, error) {
parser := initParser(script)
var result string
var err error
argv := []string{}
for {
prevtype := p.Type
// XXX
t = p.GetToken()
if p.Type == ptEOF {
prevType := parser.Type
token := parser.GetToken()
if parser.Type == ptEOF {
break
}
switch p.Type {
switch parser.Type {
case ptVAR:
v, ok := i.Var(t)
v, ok := interp.Variable(token)
if !ok {
return "", fmt.Errorf("no such variable '%s'", t)
return "", fmt.Errorf("no such variable '%s'", token)
}
t = string(v)
token = string(v)
case ptCMD:
result, err = i.Eval(t)
result, err = interp.Eval(token)
if err != nil {
return result, err
} else {
t = result
}
token = result
case ptESC:
// XXX: escape handling missing!
/*
* TODO: escape handling missing!
*/
case ptSEP:
prevtype = p.Type
prevType = parser.Type
continue
}
// We have a complete command + args. Call it!
if p.Type == ptEOL {
prevtype = p.Type
if parser.Type == ptEOL {
prevType = parser.Type
if len(argv) != 0 {
c := i.Command(argv[0])
if c == nil {
cmd := interp.Command(argv[0])
if cmd == nil {
return "", fmt.Errorf("no such command '%s'", argv[0])
}
result, err = c.fn(i, argv, c.privdata)
result, err = cmd.fn(interp, argv, cmd.privdata)
if err != nil {
return result, err
}
}
// Prepare for the next command
/*
* Prepare for the next command
*/
argv = []string{}
continue
}
// We have a new token, append to the previous or as new arg?
if prevtype == ptSEP || prevtype == ptEOL {
argv = append(argv, t)
/*
* We have a new token, append to the previous or as new arg?
*/
if prevType == ptSEP || prevType == ptEOL {
argv = append(argv, token)
} else { // Interpolation
argv[len(argv)-1] = strings.Join([]string{argv[len(argv)-1], t}, "")
argv[len(argv)-1] = strings.Join([]string{argv[len(argv)-1], token}, "")
}
prevtype = p.Type
prevType = parser.Type
}
return result, nil
}

6
pkg/picol/picol_unused/main.go_unused
View file

@ -7,13 +7,13 @@ import (
"io/ioutil"
"os"
picol "github.com/dragonheim/gagent/src/picol"
picol "github.com/dragonheim/gagent/pkg/picol"
)
var fname = flag.String("f", "", "file name")
// CommandPuts is a simple version of the TCL puts function.
func CommandPuts(i *picol.Interp, argv []string, pd interface{}) (string, error) {
func CommandPuts(i *picol.Interpreter, argv []string, pd interface{}) (string, error) {
if len(argv) != 2 {
return "", fmt.Errorf("wrong number of args for %s %s", argv[0], argv)
}
@ -23,7 +23,7 @@ func CommandPuts(i *picol.Interp, argv []string, pd interface{}) (string, error)
func main() {
flag.Parse()
interp := picol.InitInterp()
interp := picol.NewInterpreter()
interp.RegisterCoreCommands()
interp.RegisterCommand("puts", CommandPuts, nil)