1234567891011121314151617181920212223242526272829303132333435363738394041424344454647 |
- package redis
- import (
- "crypto/tls"
- "io"
- red "github.com/go-redis/redis/v8"
- "github.com/zeromicro/go-zero/core/syncx"
- )
- const (
- defaultDatabase = 0
- maxRetries = 3
- idleConns = 8
- )
- var clientManager = syncx.NewResourceManager()
- func getClient(r *Redis) (*red.Client, error) {
- val, err := clientManager.GetResource(r.Addr, func() (io.Closer, error) {
- var tlsConfig *tls.Config
- if r.tls {
- tlsConfig = &tls.Config{
- InsecureSkipVerify: true,
- }
- }
- store := red.NewClient(&red.Options{
- Addr: r.Addr,
- Password: r.Pass,
- DB: defaultDatabase,
- MaxRetries: maxRetries,
- MinIdleConns: idleConns,
- TLSConfig: tlsConfig,
- })
- store.AddHook(durationHook)
- for _, hook := range r.hooks {
- store.AddHook(hook)
- }
- return store, nil
- })
- if err != nil {
- return nil, err
- }
- return val.(*red.Client), nil
- }
|