You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.6 KiB
62 lines
1.6 KiB
package pool
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"time"
|
|
|
|
xtime "github.com/bilibili/kratos/pkg/time"
|
|
)
|
|
|
|
var (
|
|
// ErrPoolExhausted connections are exhausted.
|
|
ErrPoolExhausted = errors.New("container/pool exhausted")
|
|
// ErrPoolClosed connection pool is closed.
|
|
ErrPoolClosed = errors.New("container/pool closed")
|
|
|
|
// nowFunc returns the current time; it's overridden in tests.
|
|
nowFunc = time.Now
|
|
)
|
|
|
|
// Config is the pool configuration struct.
|
|
type Config struct {
|
|
// Active number of items allocated by the pool at a given time.
|
|
// When zero, there is no limit on the number of items in the pool.
|
|
Active int
|
|
// Idle number of idle items in the pool.
|
|
Idle int
|
|
// Close items after remaining item for this duration. If the value
|
|
// is zero, then item items are not closed. Applications should set
|
|
// the timeout to a value less than the server's timeout.
|
|
IdleTimeout xtime.Duration
|
|
// If WaitTimeout is set and the pool is at the Active limit, then Get() waits WatiTimeout
|
|
// until a item to be returned to the pool before returning.
|
|
WaitTimeout xtime.Duration
|
|
// If WaitTimeout is not set, then Wait effects.
|
|
// if Wait is set true, then wait until ctx timeout, or default flase and return directly.
|
|
Wait bool
|
|
}
|
|
|
|
type item struct {
|
|
createdAt time.Time
|
|
c io.Closer
|
|
}
|
|
|
|
func (i *item) expired(timeout time.Duration) bool {
|
|
if timeout <= 0 {
|
|
return false
|
|
}
|
|
return i.createdAt.Add(timeout).Before(nowFunc())
|
|
}
|
|
|
|
func (i *item) close() error {
|
|
return i.c.Close()
|
|
}
|
|
|
|
// Pool interface.
|
|
type Pool interface {
|
|
Get(ctx context.Context) (io.Closer, error)
|
|
Put(ctx context.Context, c io.Closer, forceClose bool) error
|
|
Close() error
|
|
}
|
|
|