2015-01-15 18:51:22 +03:00
|
|
|
package redis_test
|
|
|
|
|
|
|
|
import (
|
2019-07-04 11:18:06 +03:00
|
|
|
"context"
|
2016-10-02 15:44:01 +03:00
|
|
|
"time"
|
|
|
|
|
2015-01-15 18:51:22 +03:00
|
|
|
. "github.com/onsi/ginkgo"
|
|
|
|
. "github.com/onsi/gomega"
|
2021-09-08 16:00:52 +03:00
|
|
|
|
2022-06-04 17:39:21 +03:00
|
|
|
"github.com/go-redis/redis/v9"
|
2015-01-15 18:51:22 +03:00
|
|
|
)
|
|
|
|
|
2015-10-13 12:02:29 +03:00
|
|
|
var _ = Describe("pool", func() {
|
2015-01-15 18:51:22 +03:00
|
|
|
var client *redis.Client
|
2015-04-17 14:44:56 +03:00
|
|
|
|
2015-01-15 18:51:22 +03:00
|
|
|
BeforeEach(func() {
|
2018-08-12 10:08:21 +03:00
|
|
|
opt := redisOptions()
|
|
|
|
opt.MinIdleConns = 0
|
|
|
|
opt.MaxConnAge = 0
|
2021-03-01 11:45:35 +03:00
|
|
|
opt.IdleTimeout = time.Second
|
2018-08-12 10:08:21 +03:00
|
|
|
client = redis.NewClient(opt)
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
|
|
|
|
AfterEach(func() {
|
|
|
|
Expect(client.Close()).NotTo(HaveOccurred())
|
|
|
|
})
|
|
|
|
|
2016-10-02 15:44:01 +03:00
|
|
|
It("respects max size", func() {
|
2016-03-16 17:57:24 +03:00
|
|
|
perform(1000, func(id int) {
|
2020-03-11 17:26:42 +03:00
|
|
|
val, err := client.Ping(ctx).Result()
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(val).To(Equal("PONG"))
|
|
|
|
})
|
2015-01-31 10:53:08 +03:00
|
|
|
|
2015-01-30 17:58:26 +03:00
|
|
|
pool := client.Pool()
|
|
|
|
Expect(pool.Len()).To(BeNumerically("<=", 10))
|
2018-05-28 17:27:24 +03:00
|
|
|
Expect(pool.IdleLen()).To(BeNumerically("<=", 10))
|
|
|
|
Expect(pool.Len()).To(Equal(pool.IdleLen()))
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
|
2016-10-13 14:36:15 +03:00
|
|
|
It("respects max size on multi", func() {
|
2016-03-16 17:57:24 +03:00
|
|
|
perform(1000, func(id int) {
|
2015-01-15 18:51:22 +03:00
|
|
|
var ping *redis.StatusCmd
|
|
|
|
|
2020-03-11 17:26:42 +03:00
|
|
|
err := client.Watch(ctx, func(tx *redis.Tx) error {
|
|
|
|
cmds, err := tx.Pipelined(ctx, func(pipe redis.Pipeliner) error {
|
|
|
|
ping = pipe.Ping(ctx)
|
2016-05-02 15:54:15 +03:00
|
|
|
return nil
|
|
|
|
})
|
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(cmds).To(HaveLen(1))
|
|
|
|
return err
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
Expect(err).NotTo(HaveOccurred())
|
2016-05-02 15:54:15 +03:00
|
|
|
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(ping.Err()).NotTo(HaveOccurred())
|
|
|
|
Expect(ping.Val()).To(Equal("PONG"))
|
|
|
|
})
|
|
|
|
|
2015-01-31 10:53:08 +03:00
|
|
|
pool := client.Pool()
|
|
|
|
Expect(pool.Len()).To(BeNumerically("<=", 10))
|
2018-05-28 17:27:24 +03:00
|
|
|
Expect(pool.IdleLen()).To(BeNumerically("<=", 10))
|
|
|
|
Expect(pool.Len()).To(Equal(pool.IdleLen()))
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
|
2016-10-02 15:44:01 +03:00
|
|
|
It("respects max size on pipelines", func() {
|
2016-03-16 17:57:24 +03:00
|
|
|
perform(1000, func(id int) {
|
2015-01-15 18:51:22 +03:00
|
|
|
pipe := client.Pipeline()
|
2020-03-11 17:26:42 +03:00
|
|
|
ping := pipe.Ping(ctx)
|
|
|
|
cmds, err := pipe.Exec(ctx)
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(cmds).To(HaveLen(1))
|
|
|
|
Expect(ping.Err()).NotTo(HaveOccurred())
|
|
|
|
Expect(ping.Val()).To(Equal("PONG"))
|
|
|
|
})
|
|
|
|
|
2015-01-31 10:53:08 +03:00
|
|
|
pool := client.Pool()
|
|
|
|
Expect(pool.Len()).To(BeNumerically("<=", 10))
|
2018-05-28 17:27:24 +03:00
|
|
|
Expect(pool.IdleLen()).To(BeNumerically("<=", 10))
|
|
|
|
Expect(pool.Len()).To(Equal(pool.IdleLen()))
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
|
2016-10-02 15:44:01 +03:00
|
|
|
It("removes broken connections", func() {
|
2019-07-04 11:18:06 +03:00
|
|
|
cn, err := client.Pool().Get(context.Background())
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
2017-02-08 12:24:09 +03:00
|
|
|
cn.SetNetConn(&badConn{})
|
2020-08-15 15:36:02 +03:00
|
|
|
client.Pool().Put(ctx, cn)
|
2015-01-15 18:51:22 +03:00
|
|
|
|
2020-03-11 17:26:42 +03:00
|
|
|
val, err := client.Ping(ctx).Result()
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(val).To(Equal("PONG"))
|
|
|
|
|
2022-06-04 15:15:43 +03:00
|
|
|
val, err = client.Ping(ctx).Result()
|
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(val).To(Equal("PONG"))
|
|
|
|
|
2015-01-31 10:53:08 +03:00
|
|
|
pool := client.Pool()
|
|
|
|
Expect(pool.Len()).To(Equal(1))
|
2018-05-28 17:27:24 +03:00
|
|
|
Expect(pool.IdleLen()).To(Equal(1))
|
2016-03-09 14:38:33 +03:00
|
|
|
|
|
|
|
stats := pool.Stats()
|
2019-05-31 17:54:30 +03:00
|
|
|
Expect(stats.Hits).To(Equal(uint32(1)))
|
2017-09-22 12:23:46 +03:00
|
|
|
Expect(stats.Misses).To(Equal(uint32(2)))
|
2016-03-09 14:38:33 +03:00
|
|
|
Expect(stats.Timeouts).To(Equal(uint32(0)))
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
|
|
|
|
2016-10-02 15:44:01 +03:00
|
|
|
It("reuses connections", func() {
|
2021-03-01 11:45:35 +03:00
|
|
|
// explain: https://github.com/go-redis/redis/pull/1675
|
|
|
|
opt := redisOptions()
|
|
|
|
opt.MinIdleConns = 0
|
|
|
|
opt.MaxConnAge = 0
|
|
|
|
opt.IdleTimeout = 2 * time.Second
|
|
|
|
client = redis.NewClient(opt)
|
|
|
|
|
2015-01-15 18:51:22 +03:00
|
|
|
for i := 0; i < 100; i++ {
|
2020-03-11 17:26:42 +03:00
|
|
|
val, err := client.Ping(ctx).Result()
|
2015-01-15 18:51:22 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
Expect(val).To(Equal("PONG"))
|
|
|
|
}
|
|
|
|
|
2015-01-31 10:53:08 +03:00
|
|
|
pool := client.Pool()
|
|
|
|
Expect(pool.Len()).To(Equal(1))
|
2018-05-28 17:27:24 +03:00
|
|
|
Expect(pool.IdleLen()).To(Equal(1))
|
2016-03-09 14:38:33 +03:00
|
|
|
|
|
|
|
stats := pool.Stats()
|
2019-05-31 17:54:30 +03:00
|
|
|
Expect(stats.Hits).To(Equal(uint32(99)))
|
2017-09-22 12:23:46 +03:00
|
|
|
Expect(stats.Misses).To(Equal(uint32(1)))
|
2016-03-09 14:38:33 +03:00
|
|
|
Expect(stats.Timeouts).To(Equal(uint32(0)))
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|
2016-10-02 15:44:01 +03:00
|
|
|
|
|
|
|
It("removes idle connections", func() {
|
2020-03-11 17:26:42 +03:00
|
|
|
err := client.Ping(ctx).Err()
|
2019-05-31 17:54:30 +03:00
|
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
|
2016-10-02 15:44:01 +03:00
|
|
|
stats := client.PoolStats()
|
|
|
|
Expect(stats).To(Equal(&redis.PoolStats{
|
|
|
|
Hits: 0,
|
2017-09-22 12:23:46 +03:00
|
|
|
Misses: 1,
|
2016-10-02 15:44:01 +03:00
|
|
|
Timeouts: 0,
|
|
|
|
TotalConns: 1,
|
2018-05-28 17:27:24 +03:00
|
|
|
IdleConns: 1,
|
2017-09-11 10:12:00 +03:00
|
|
|
StaleConns: 0,
|
2016-10-02 15:44:01 +03:00
|
|
|
}))
|
|
|
|
|
2021-03-01 11:45:35 +03:00
|
|
|
time.Sleep(2 * time.Second)
|
2016-10-02 15:44:01 +03:00
|
|
|
|
|
|
|
stats = client.PoolStats()
|
|
|
|
Expect(stats).To(Equal(&redis.PoolStats{
|
|
|
|
Hits: 0,
|
2017-09-22 12:23:46 +03:00
|
|
|
Misses: 1,
|
2016-10-02 15:44:01 +03:00
|
|
|
Timeouts: 0,
|
|
|
|
TotalConns: 0,
|
2018-08-12 10:08:21 +03:00
|
|
|
IdleConns: 0,
|
2017-09-11 10:12:00 +03:00
|
|
|
StaleConns: 1,
|
2016-10-02 15:44:01 +03:00
|
|
|
}))
|
|
|
|
})
|
2015-01-15 18:51:22 +03:00
|
|
|
})
|