src: Add DMA localagent

[barometer.git] / src / dma / vendor / github.com / go-redis / redis / internal / error.go

diff --git a/src/dma/vendor/github.com/go-redis/redis/internal/error.go b/src/dma/vendor/github.com/go-redis/redis/internal/error.go
new file mode 100644 (file)
index 0000000..e0ff863
--- /dev/null
+++ b/src/dma/vendor/github.com/go-redis/redis/internal/error.go
@@ -0,0 +1,83 @@
+package internal
+
+import (
+       "io"
+       "net"
+       "strings"
+
+       "github.com/go-redis/redis/internal/proto"
+)
+
+func IsRetryableError(err error, retryNetError bool) bool {
+       if IsNetworkError(err) {
+               return retryNetError
+       }
+       s := err.Error()
+       if s == "ERR max number of clients reached" {
+               return true
+       }
+       if strings.HasPrefix(s, "LOADING ") {
+               return true
+       }
+       if strings.HasPrefix(s, "READONLY ") {
+               return true
+       }
+       if strings.HasPrefix(s, "CLUSTERDOWN ") {
+               return true
+       }
+       return false
+}
+
+func IsRedisError(err error) bool {
+       _, ok := err.(proto.RedisError)
+       return ok
+}
+
+func IsNetworkError(err error) bool {
+       if err == io.EOF {
+               return true
+       }
+       _, ok := err.(net.Error)
+       return ok
+}
+
+func IsBadConn(err error, allowTimeout bool) bool {
+       if err == nil {
+               return false
+       }
+       if IsRedisError(err) {
+               return strings.HasPrefix(err.Error(), "READONLY ")
+       }
+       if allowTimeout {
+               if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
+                       return false
+               }
+       }
+       return true
+}
+
+func IsMovedError(err error) (moved bool, ask bool, addr string) {
+       if !IsRedisError(err) {
+               return
+       }
+
+       s := err.Error()
+       if strings.HasPrefix(s, "MOVED ") {
+               moved = true
+       } else if strings.HasPrefix(s, "ASK ") {
+               ask = true
+       } else {
+               return
+       }
+
+       ind := strings.LastIndex(s, " ")
+       if ind == -1 {
+               return false, false, ""
+       }
+       addr = s[ind+1:]
+       return
+}
+
+func IsLoadingError(err error) bool {
+       return strings.HasPrefix(err.Error(), "LOADING ")
+}