85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
|
// Copyright 2022 Listware
|
||
|
|
||
|
package module
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"strings"
|
||
|
"time"
|
||
|
|
||
|
"git.fg-tech.ru/listware/cmdb/pkg/cmdb/vertex"
|
||
|
"git.fg-tech.ru/listware/go-core/pkg/log"
|
||
|
"git.fg-tech.ru/listware/proto/sdk/pbtypes"
|
||
|
"github.com/apache/flink-statefun/statefun-sdk-go/v3/pkg/statefun"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
functionContextTypeName = statefun.MakeProtobufTypeWithTypeName(statefun.TypeNameFrom("type.googleapis.com/FunctionContext"))
|
||
|
)
|
||
|
|
||
|
// StatefulFunction function interface
|
||
|
type StatefulFunction func(Context) error
|
||
|
|
||
|
// functionMiddleware adapter implemets middleware
|
||
|
type functionMiddleware struct {
|
||
|
parent *module
|
||
|
f StatefulFunction
|
||
|
}
|
||
|
|
||
|
// Invoke statefun.StatefulFunction implementation
|
||
|
func (fm *functionMiddleware) Invoke(ctx statefun.Context, message statefun.Message) (err error) {
|
||
|
start := time.Now()
|
||
|
defer func() {
|
||
|
log.Debug(ctx.Self(), "took", time.Since(start))
|
||
|
}()
|
||
|
log.Debug(ctx.Self())
|
||
|
|
||
|
// if ctx.Caller().FunctionType != nil {
|
||
|
// log.Debug("caller", ctx.Caller())
|
||
|
// }
|
||
|
//
|
||
|
|
||
|
var functionContext pbtypes.FunctionContext
|
||
|
if err = message.As(functionContextTypeName, &functionContext); err != nil {
|
||
|
log.Debug(ctx.Self(), "proto", err)
|
||
|
// do not retry idempotent call
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
context := &contextAdapter{
|
||
|
Context: ctx,
|
||
|
message: functionContext.Value,
|
||
|
states: fm.parent.states[ctx.Self().FunctionType.String()],
|
||
|
}
|
||
|
|
||
|
collectionKey := strings.Split(ctx.Self().Id, "/")
|
||
|
if len(collectionKey) != 2 {
|
||
|
log.Debug(ctx.Self(), "bad id")
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
response, err := vertex.Read(ctx, collectionKey[1], collectionKey[0])
|
||
|
if err != nil {
|
||
|
log.Debug(err)
|
||
|
return nil
|
||
|
}
|
||
|
context.cmdbContext = response.GetPayload()
|
||
|
|
||
|
if context.cmdbContext == nil {
|
||
|
log.Debug(ctx.Self(), "bad context")
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if err = fm.f(context); err != nil {
|
||
|
log.Debug(err)
|
||
|
// print all errors
|
||
|
}
|
||
|
|
||
|
if err = errors.Unwrap(err); err != nil {
|
||
|
// retry only if wrapped error
|
||
|
return
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|