29 lines
827 B
Go
29 lines
827 B
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.front.kjuulh.io/kjuulh/curre"
|
||
|
|
"git.front.kjuulh.io/kjuulh/kraken/internal/serverdeps"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewStorageServer(logger *zap.Logger, deps *serverdeps.ServerDeps) curre.Component {
|
||
|
|
storage := deps.GetStorageService()
|
||
|
|
return curre.NewFunctionalComponent(&curre.FunctionalComponent{
|
||
|
|
InitFunc: func(_ *curre.FunctionalComponent, ctx context.Context) error {
|
||
|
|
logger.Debug("Initializing storage")
|
||
|
|
return storage.InitializeStorage(ctx)
|
||
|
|
},
|
||
|
|
StartFunc: func(fc *curre.FunctionalComponent, ctx context.Context) error {
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
StopFunc: func(_ *curre.FunctionalComponent, ctx context.Context) error {
|
||
|
|
logger.Debug("Cleaning up storage")
|
||
|
|
ctx, _ = context.WithTimeout(ctx, time.Second*10)
|
||
|
|
return storage.CleanupStorage(ctx)
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|