forked from TrueCloudLab/frostfs-node
36 lines
642 B
Go
36 lines
642 B
Go
|
package module
|
||
|
|
|
||
|
|
import (
|
||
|
|
"go.uber.org/dig"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
// Module type
|
||
|
|
Module []*Provider
|
||
|
|
|
||
|
|
// Provider struct
|
||
|
|
Provider struct {
|
||
|
|
Constructor interface{}
|
||
|
|
Options []dig.ProvideOption
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
// Append module to target module and return new module
|
||
|
|
func (m Module) Append(mods ...Module) Module {
|
||
|
|
var result = m
|
||
|
|
for _, mod := range mods {
|
||
|
|
result = append(result, mod...)
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
// Provide set providers functions to DI container
|
||
|
|
func Provide(dic *dig.Container, providers Module) error {
|
||
|
|
for _, p := range providers {
|
||
|
|
if err := dic.Provide(p.Constructor, p.Options...); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|