createPoolHook creates a pool hook with this manager already set.
(baseDialer func(context.Context, string, string) (net.Conn, error))
| 377 | |
| 378 | // createPoolHook creates a pool hook with this manager already set. |
| 379 | func (hm *Manager) createPoolHook(baseDialer func(context.Context, string, string) (net.Conn, error)) *PoolHook { |
| 380 | if hm.poolHooksRef != nil { |
| 381 | return hm.poolHooksRef |
| 382 | } |
| 383 | // Get pool size from client options for better worker defaults |
| 384 | poolSize := 0 |
| 385 | if hm.options != nil { |
| 386 | poolSize = hm.options.GetPoolSize() |
| 387 | } |
| 388 | |
| 389 | hm.poolHooksRef = NewPoolHookWithPoolSize(baseDialer, hm.options.GetNetwork(), hm.config, hm, poolSize) |
| 390 | hm.poolHooksRef.SetPool(hm.pool) |
| 391 | |
| 392 | return hm.poolHooksRef |
| 393 | } |
| 394 | |
| 395 | func (hm *Manager) AddNotificationHook(notificationHook NotificationHook) { |
| 396 | hm.hooksMu.Lock() |
no test coverage detected