Add -log option to proxy-go.

This commit is contained in:
David Fifield 2017-06-29 15:11:14 -07:00
parent 62f50b29b2
commit c3f87281be

View file

@ -12,6 +12,7 @@ import (
"net" "net"
"net/http" "net/http"
"net/url" "net/url"
"os"
"strings" "strings"
"sync" "sync"
"time" "time"
@ -291,13 +292,24 @@ func runSession(sid string) {
} }
func main() { func main() {
var logFilename string
opt = new(snowflakeOptions) opt = new(snowflakeOptions)
flag.IntVar(&opt.capacity, "capacity", 10, "maximum concurrent clients") flag.IntVar(&opt.capacity, "capacity", 10, "maximum concurrent clients")
flag.StringVar(&opt.broker, "broker", "https://snowflake-reg.appspot.com/", "broker URL") flag.StringVar(&opt.broker, "broker", "https://snowflake-reg.appspot.com/", "broker URL")
flag.StringVar(&opt.relay, "relay", "wss://snowflake.bamsoftware.com/", "websocket relay URL") flag.StringVar(&opt.relay, "relay", "wss://snowflake.bamsoftware.com/", "websocket relay URL")
flag.StringVar(&opt.stun, "stun", "stun:stun.l.google.com:19302", "stun URL") flag.StringVar(&opt.stun, "stun", "stun:stun.l.google.com:19302", "stun URL")
flag.StringVar(&logFilename, "log", "", "log filename")
flag.Parse() flag.Parse()
if logFilename != "" {
f, err := os.OpenFile(logFilename, os.O_WRONLY | os.O_APPEND, 0600)
if err != nil {
log.Fatal(err)
}
defer f.Close()
log.SetOutput(io.MultiWriter(os.Stderr, f))
}
var err error var err error
opt.brokerURL, err = url.Parse(opt.broker) opt.brokerURL, err = url.Parse(opt.broker)
if err != nil { if err != nil {