114 lines
3.0 KiB
Go
114 lines
3.0 KiB
Go
package request
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.defalsify.org/vise.git/db"
|
|
"git.defalsify.org/vise.git/engine"
|
|
"git.defalsify.org/vise.git/persist"
|
|
"git.defalsify.org/vise.git/resource"
|
|
"git.grassecon.net/grassrootseconomics/visedriver/entry"
|
|
"git.grassecon.net/grassrootseconomics/visedriver/errors"
|
|
"git.grassecon.net/grassrootseconomics/visedriver/storage"
|
|
)
|
|
|
|
type EngineFunc func(engine.Config, resource.Resource, *persist.Persister) engine.Engine
|
|
|
|
type BaseRequestHandler struct {
|
|
cfgTemplate engine.Config
|
|
rp RequestParser
|
|
rs resource.Resource
|
|
hn entry.EntryHandler
|
|
provider storage.StorageProvider
|
|
engineFunc EngineFunc
|
|
}
|
|
|
|
func NewBaseRequestHandler(cfg engine.Config, rs resource.Resource, stateDb db.Db, userdataDb db.Db, rp RequestParser, hn entry.EntryHandler) *BaseRequestHandler {
|
|
h := &BaseRequestHandler{
|
|
cfgTemplate: cfg,
|
|
rs: rs,
|
|
hn: hn,
|
|
rp: rp,
|
|
provider: storage.NewSimpleStorageProvider(stateDb, userdataDb),
|
|
}
|
|
h.engineFunc = h.getDefaultEngine
|
|
return h
|
|
}
|
|
|
|
func (f *BaseRequestHandler) WithEngineFunc(fn EngineFunc) *BaseRequestHandler {
|
|
f.engineFunc = fn
|
|
return f
|
|
}
|
|
|
|
func (f *BaseRequestHandler) Shutdown(ctx context.Context) {
|
|
err := f.provider.Close(ctx)
|
|
if err != nil {
|
|
logg.Errorf("handler shutdown error", "err", err)
|
|
}
|
|
}
|
|
|
|
func (f *BaseRequestHandler) GetEngine(cfg engine.Config, rs resource.Resource, pr *persist.Persister) engine.Engine {
|
|
return f.engineFunc(cfg, rs, pr)
|
|
}
|
|
|
|
func (f *BaseRequestHandler) getDefaultEngine(cfg engine.Config, rs resource.Resource, pr *persist.Persister) engine.Engine {
|
|
en := engine.NewEngine(cfg, rs)
|
|
en = en.WithPersister(pr)
|
|
en = en.WithFirst(f.hn.Init)
|
|
if f.cfgTemplate.EngineDebug {
|
|
en = en.WithDebug(nil)
|
|
}
|
|
return en
|
|
}
|
|
|
|
func (f *BaseRequestHandler) Process(rqs RequestSession) (RequestSession, error) {
|
|
var r bool
|
|
var err error
|
|
|
|
logg.InfoCtxf(rqs.Ctx, "new request", "data", rqs)
|
|
|
|
rqs.Storage, err = f.provider.Get(rqs.Ctx, rqs.Config.SessionId)
|
|
if err != nil {
|
|
logg.ErrorCtxf(rqs.Ctx, "", "storage get error", err)
|
|
return rqs, errors.ErrStorage
|
|
}
|
|
|
|
f.hn.SetPersister(rqs.Storage.Persister)
|
|
defer func() {
|
|
f.hn.Exit()
|
|
}()
|
|
|
|
rqs.Engine = f.GetEngine(rqs.Config, f.rs, rqs.Storage.Persister)
|
|
r, err = rqs.Engine.Exec(rqs.Ctx, rqs.Input)
|
|
if err != nil {
|
|
perr := f.provider.Put(rqs.Ctx, rqs.Config.SessionId, rqs.Storage)
|
|
rqs.Storage = nil
|
|
if perr != nil {
|
|
logg.ErrorCtxf(rqs.Ctx, "", "storage put error", perr)
|
|
}
|
|
return rqs, err
|
|
}
|
|
|
|
rqs.Continue = r
|
|
return rqs, nil
|
|
}
|
|
|
|
func (f *BaseRequestHandler) Output(rqs RequestSession) (RequestSession, error) {
|
|
var err error
|
|
_, err = rqs.Engine.Flush(rqs.Ctx, rqs.Writer)
|
|
return rqs, err
|
|
}
|
|
|
|
func (f *BaseRequestHandler) Reset(ctx context.Context, rqs RequestSession) (RequestSession, error) {
|
|
defer f.provider.Put(ctx, rqs.Config.SessionId, rqs.Storage)
|
|
return rqs, rqs.Engine.Finish(ctx)
|
|
}
|
|
|
|
func (f *BaseRequestHandler) GetConfig() engine.Config {
|
|
return f.cfgTemplate
|
|
}
|
|
|
|
func (f *BaseRequestHandler) GetRequestParser() RequestParser {
|
|
return f.rp
|
|
}
|