mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
87 lines
1.5 KiB
Go
87 lines
1.5 KiB
Go
|
package actions
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"time"
|
||
|
|
||
|
"github.com/dop251/goja"
|
||
|
"github.com/dop251/goja_nodejs/require"
|
||
|
"github.com/zitadel/logging"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
maxPrepareTimeout = 5 * time.Second
|
||
|
)
|
||
|
|
||
|
type Option func(*runConfig)
|
||
|
|
||
|
func WithAllowedToFail() Option {
|
||
|
return func(c *runConfig) {
|
||
|
c.allowedToFail = true
|
||
|
}
|
||
|
}
|
||
|
|
||
|
type runConfig struct {
|
||
|
allowedToFail bool
|
||
|
timeout,
|
||
|
prepareTimeout time.Duration
|
||
|
modules map[string]require.ModuleLoader
|
||
|
|
||
|
vm *goja.Runtime
|
||
|
ctxParam *ctxConfig
|
||
|
apiParam *apiConfig
|
||
|
}
|
||
|
|
||
|
func newRunConfig(ctx context.Context, opts ...Option) *runConfig {
|
||
|
deadline, ok := ctx.Deadline()
|
||
|
if !ok {
|
||
|
logging.Warn("no timeout set on action run")
|
||
|
}
|
||
|
|
||
|
vm := goja.New()
|
||
|
vm.SetFieldNameMapper(goja.UncapFieldNameMapper())
|
||
|
|
||
|
config := &runConfig{
|
||
|
timeout: time.Until(deadline),
|
||
|
prepareTimeout: maxPrepareTimeout,
|
||
|
modules: map[string]require.ModuleLoader{},
|
||
|
vm: vm,
|
||
|
ctxParam: &ctxConfig{
|
||
|
FieldConfig: FieldConfig{
|
||
|
Runtime: vm,
|
||
|
fields: fields{},
|
||
|
},
|
||
|
},
|
||
|
apiParam: &apiConfig{
|
||
|
FieldConfig: FieldConfig{
|
||
|
Runtime: vm,
|
||
|
fields: fields{},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, opt := range opts {
|
||
|
opt(config)
|
||
|
}
|
||
|
|
||
|
if config.prepareTimeout > config.timeout {
|
||
|
config.prepareTimeout = config.timeout
|
||
|
}
|
||
|
|
||
|
return config
|
||
|
}
|
||
|
|
||
|
func (c *runConfig) Start() *time.Timer {
|
||
|
c.vm.ClearInterrupt()
|
||
|
return time.AfterFunc(c.timeout, func() {
|
||
|
c.vm.Interrupt(ErrHalt)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *runConfig) Prepare() *time.Timer {
|
||
|
c.vm.ClearInterrupt()
|
||
|
return time.AfterFunc(c.prepareTimeout, func() {
|
||
|
c.vm.Interrupt(ErrHalt)
|
||
|
})
|
||
|
}
|