add --skip-master option

ttkzw · Oct 18, 2018 · ad52d89 · ad52d89
1 parent f4cd485
commit ad52d89
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/check-redis/lib/check-redis.go b/check-redis/lib/check-redis.go
@@ -136,8 +136,13 @@ func checkReachable(args []string) *checkers.Checker {
 	)
 }
 
+type replicationOpts struct {
+	redisSetting
+	SkipMaster bool `long:"skip-master" description:"return ok if redis role is master"`
+}
+
 func checkReplication(args []string) *checkers.Checker {
-	opts := redisSetting{}
+	opts := replicationOpts{}
 	psr := flags.NewParser(&opts, flags.Default)
 	psr.Usage = "replication [OPTIONS]"
 	_, err := psr.ParseArgs(args)
@@ -146,14 +151,14 @@ func checkReplication(args []string) *checkers.Checker {
 		os.Exit(1)
 	}
 
-	c, info, err := connectRedisGetInfo(opts)
+	c, info, err := connectRedisGetInfo(opts.redisSetting)
 	if err != nil {
 		return checkers.Unknown(err.Error())
 	}
 	defer c.Close()
 
 	if role, ok := (*info)["role"]; ok {
-		if role != "slave" {
+		if role != "slave" && opts.SkipMaster {
 			return checkers.Ok("role is not slave")
 		}
 	} else {