mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-14 15:09:51 +00:00
1b65d97b65
* refactor script manager
* remove singleton GCTScriptConfig
* create constant for ".gct" extension
* move GctScriptManager into vm package
* reduce script manager global dependencies
* use manager struct to store runtime override values
* enable/disable scripting subsystem now doesn't store the setting in
config (aligned with other subsystems)
* setting max VMs via start option doesn't change config
* instantiate scriptmanager as part of creating a new Engine
* script manager config is now set during instantiation
* run script manager when enabled in conf or explicitly enabled
* use the Started() method to check if script manager is running
* in tests set script manager as running
* script manager adjustments
* create manager before attempting overrides
* check for nil config when creating script manager
* fix script manager waitgroup counter increased too late
* move autoload() function to autoload.go
* add tests to script manager
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package vm
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gofrs/uuid"
|
|
"github.com/thrasher-corp/gocryptotrader/gctscript/wrappers/validator"
|
|
"github.com/thrasher-corp/gocryptotrader/log"
|
|
)
|
|
|
|
// New returns a new instance of VM
|
|
func (g *GctScriptManager) New() *VM {
|
|
if VMSCount.Len() >= int32(g.GetMaxVirtualMachines()) {
|
|
if g.config.Verbose {
|
|
log.Warnf(log.GCTScriptMgr, "GCTScript MaxVirtualMachines (%v) hit, unable to start further instances",
|
|
g.GetMaxVirtualMachines())
|
|
}
|
|
return nil
|
|
}
|
|
VMSCount.add()
|
|
vm := g.NewVM()
|
|
if vm == nil {
|
|
VMSCount.remove()
|
|
} else {
|
|
AllVMSync.Store(vm.ID, vm)
|
|
}
|
|
return vm
|
|
}
|
|
|
|
// Validate will attempt to execute a script in a test/non-live environment
|
|
// to confirm it passes requirements for execution
|
|
func (g *GctScriptManager) Validate(file string) (err error) {
|
|
validator.IsTestExecution.Store(true)
|
|
defer validator.IsTestExecution.Store(false)
|
|
tempVM := g.NewVM()
|
|
err = tempVM.Load(file)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = tempVM.Compile()
|
|
if err != nil {
|
|
return
|
|
}
|
|
return tempVM.Run()
|
|
}
|
|
|
|
// ShutdownAll shutdown all
|
|
func (g *GctScriptManager) ShutdownAll() (err error) {
|
|
if g.config.Verbose {
|
|
log.Debugln(log.GCTScriptMgr, "Shutting down all Virtual Machines")
|
|
}
|
|
|
|
var errors []error
|
|
AllVMSync.Range(func(k, v interface{}) bool {
|
|
errShutdown := v.(*VM).Shutdown()
|
|
if err != nil {
|
|
errors = append(errors, errShutdown)
|
|
}
|
|
return true
|
|
})
|
|
|
|
if len(errors) > 0 {
|
|
err = fmt.Errorf("failed to shutdown the following Virtual Machines: %v", errors)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// RemoveVM remove VM from list
|
|
func (g *GctScriptManager) RemoveVM(id uuid.UUID) error {
|
|
if _, f := AllVMSync.Load(id); !f {
|
|
return fmt.Errorf(ErrNoVMFound, id.String())
|
|
}
|
|
|
|
AllVMSync.Delete(id)
|
|
VMSCount.remove()
|
|
if g.config.Verbose {
|
|
log.Debugf(log.GCTScriptMgr, "VM %v removed from AllVMs", id)
|
|
}
|
|
return nil
|
|
}
|