33 lines
668 B
Go
33 lines
668 B
Go
package redis
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/redis/go-redis/v9"
|
|
"github.com/rogeecn/atom/container"
|
|
"github.com/rogeecn/atom/utils/opt"
|
|
)
|
|
|
|
func Provide(opts ...opt.Option) error {
|
|
o := opt.New(opts...)
|
|
var conf Config
|
|
if err := o.UnmarshalConfig(&conf); err != nil {
|
|
return err
|
|
}
|
|
|
|
return container.Container.Provide(func() (*redis.Client, error) {
|
|
client := redis.NewClient(conf.ToRedisOptions())
|
|
|
|
_, err := client.Ping(context.Background()).Result()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to ping")
|
|
}
|
|
|
|
container.AddCloseAble(func() {
|
|
client.Close()
|
|
})
|
|
return client, nil
|
|
}, o.DiOptions()...)
|
|
}
|