Changeset 491 in code for trunk/cmd/soju/main.go

Timestamp:

Mar 31, 2021, 5:02:40 PM (4 years ago)

Author:

contact

Message:

cmd/soju: allow specifying -listen multiple times

Closes: ​https://todo.sr.ht/~emersion/soju/67

File:

• trunk/cmd/soju/main.go (modified) (3 diffs)

trunk/cmd/soju/main.go

@@ -5 +5 @@
         "crypto/tls"
         "flag"
+        "fmt"
         "log"
         "net"
@@ -25 +26 @@
 const downstreamKeepAlive = 1 * time.Hour
 
+type stringSliceFlag []string
+
+func (v *stringSliceFlag) String() string {
+        return fmt.Sprint([]string(*v))
+}
+
+func (v *stringSliceFlag) Set(s string) error {
+        *v = append(*v, s)
+        return nil
+}
+
 func main() {
-        var listen, configPath string
+        var listen []string
+        var configPath string
         var debug bool
-        flag.StringVar(&listen, "listen", "", "listening address")
+        flag.Var((*stringSliceFlag)(&listen), "listen", "listening address")
         flag.StringVar(&configPath, "config", "", "path to configuration file")
         flag.BoolVar(&debug, "debug", false, "enable debug logging")
@@ -44 +57 @@
         }
 
-        if listen != "" {
-                cfg.Listen = append(cfg.Listen, listen)
-        }
+        cfg.Listen = append(cfg.Listen, listen...)
         if len(cfg.Listen) == 0 {
                 cfg.Listen = []string{":6697"}