MINOR: peer: add shard option to peer and server lines

Author: mjuraga

params/server-options.go

@@ -206,6 +206,7 @@ func getServerOptions() []ServerOption {
 		&ServerOptionValue{Name: "resolve-net"},
 		&ServerOptionValue{Name: "resolvers"},
 		&ServerOptionValue{Name: "proxy-v2-options"},
+		&ServerOptionValue{Name: "shard"},
 		&ServerOptionValue{Name: "slowstart"},
 		&ServerOptionValue{Name: "sni"},
 		&ServerOptionValue{Name: "source"},

parsers/peer.go

@@ -35,12 +35,16 @@ func (l *Peer) parse(line string, parts []string, comment string) (*types.Peer,
 		adr, p, found := common.CutRight(parts[2], ":")
 		if found && len(adr) > 0 {
 			if port, err := strconv.ParseInt(p, 10, 64); err == nil {
-				return &types.Peer{
+				peer := &types.Peer{
 					Name:    parts[1],
 					IP:      adr,
 					Port:    port,
 					Comment: comment,
-				}, nil
+				}
+				if len(parts) > 4 && parts[3] == "shard" {
+					peer.Shard = parts[4]
+				}
+				return peer, nil
 			}
 		}
 	}
@@ -53,8 +57,12 @@ func (l *Peer) Result() ([]common.ReturnResultLine, error) {
 	}
 	result := make([]common.ReturnResultLine, len(l.data))
 	for index, peer := range l.data {
+		str := fmt.Sprintf("peer %s %s:%d", peer.Name, peer.IP, peer.Port)
+		if peer.Shard != "" {
+			str = fmt.Sprintf("%s shard %s", str, peer.Shard)
+		}
 		result[index] = common.ReturnResultLine{
-			Data:    fmt.Sprintf("peer %s %s:%d", peer.Name, peer.IP, peer.Port),
+			Data:    str,
 			Comment: peer.Comment,
 		}
 	}

tests/configs/haproxy_generated.cfg.go

@@ -861,6 +861,7 @@ type ExternalCheckCommand struct {
 //name:peer
 //is:multiple
 //test:ok:peer name 127.0.0.1:8080
+//test:ok:peer name 127.0.0.1:8080 shard 1
 //test:fail:peer name 127.0.0.1
 //test:fail:peer name :8080
 //test:fail:peer
@@ -869,6 +870,7 @@ type Peer struct {
 	Name    string
 	IP      string
 	Port    int64
+	Shard   string
 	Comment string
 }