mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-19 02:53:25 +02:00
* update github.com/PuerkitoBio/goquery * update github.com/alecthomas/chroma * update github.com/blevesearch/bleve/v2 * update github.com/caddyserver/certmagic * update github.com/go-enry/go-enry/v2 * update github.com/go-git/go-billy/v5 * update github.com/go-git/go-git/v5 * update github.com/go-redis/redis/v8 * update github.com/go-testfixtures/testfixtures/v3 * update github.com/jaytaylor/html2text * update github.com/json-iterator/go * update github.com/klauspost/compress * update github.com/markbates/goth * update github.com/mattn/go-isatty * update github.com/mholt/archiver/v3 * update github.com/microcosm-cc/bluemonday * update github.com/minio/minio-go/v7 * update github.com/prometheus/client_golang * update github.com/unrolled/render * update github.com/xanzy/go-gitlab * update github.com/yuin/goldmark * update github.com/yuin/goldmark-highlighting Co-authored-by: techknowlogick <techknowlogick@gitea.io>
100 lines
1.8 KiB
Go
Vendored
100 lines
1.8 KiB
Go
Vendored
package redis
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
func (c *ClusterClient) DBSize(ctx context.Context) *IntCmd {
|
|
cmd := NewIntCmd(ctx, "dbsize")
|
|
var size int64
|
|
err := c.ForEachMaster(ctx, func(ctx context.Context, master *Client) error {
|
|
n, err := master.DBSize(ctx).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
atomic.AddInt64(&size, n)
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
cmd.SetErr(err)
|
|
return cmd
|
|
}
|
|
cmd.val = size
|
|
return cmd
|
|
}
|
|
|
|
func (c *ClusterClient) ScriptLoad(ctx context.Context, script string) *StringCmd {
|
|
cmd := NewStringCmd(ctx, "script", "load", script)
|
|
mu := &sync.Mutex{}
|
|
err := c.ForEachShard(ctx, func(ctx context.Context, shard *Client) error {
|
|
val, err := shard.ScriptLoad(ctx, script).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mu.Lock()
|
|
if cmd.Val() == "" {
|
|
cmd.val = val
|
|
}
|
|
mu.Unlock()
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
cmd.SetErr(err)
|
|
}
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (c *ClusterClient) ScriptFlush(ctx context.Context) *StatusCmd {
|
|
cmd := NewStatusCmd(ctx, "script", "flush")
|
|
_ = c.ForEachShard(ctx, func(ctx context.Context, shard *Client) error {
|
|
shard.ScriptFlush(ctx)
|
|
|
|
return nil
|
|
})
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (c *ClusterClient) ScriptExists(ctx context.Context, hashes ...string) *BoolSliceCmd {
|
|
args := make([]interface{}, 2+len(hashes))
|
|
args[0] = "script"
|
|
args[1] = "exists"
|
|
for i, hash := range hashes {
|
|
args[2+i] = hash
|
|
}
|
|
cmd := NewBoolSliceCmd(ctx, args...)
|
|
|
|
result := make([]bool, len(hashes))
|
|
for i := range result {
|
|
result[i] = true
|
|
}
|
|
|
|
mu := &sync.Mutex{}
|
|
err := c.ForEachShard(ctx, func(ctx context.Context, shard *Client) error {
|
|
val, err := shard.ScriptExists(ctx, hashes...).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mu.Lock()
|
|
for i, v := range val {
|
|
result[i] = result[i] && v
|
|
}
|
|
mu.Unlock()
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
cmd.SetErr(err)
|
|
}
|
|
|
|
cmd.val = result
|
|
|
|
return cmd
|
|
}
|