Просмотр исходного кода

Print URLs on start to stdout

tags/0.9
9seconds 8 лет назад
Родитель
Сommit
42e5be36bf
1 измененных файлов: 13 добавлений и 0 удалений
  1. 13
    0
      main.go

+ 13
- 0
main.go Просмотреть файл

4
 
4
 
5
 import (
5
 import (
6
 	"encoding/hex"
6
 	"encoding/hex"
7
+	"encoding/json"
7
 	"io"
8
 	"io"
8
 	"io/ioutil"
9
 	"io/ioutil"
9
 	"net/http"
10
 	"net/http"
119
 
120
 
120
 	stat := proxy.NewStats(*serverName, *portToShow, *secret)
121
 	stat := proxy.NewStats(*serverName, *portToShow, *secret)
121
 	go stat.Serve(*statsIP, *statsPort)
122
 	go stat.Serve(*statsIP, *statsPort)
123
+	printURLs(stat.URLs)
122
 
124
 
123
 	srv := proxy.NewServer(*bindIP, int(*bindPort), secretBytes, logger,
125
 	srv := proxy.NewServer(*bindIP, int(*bindPort), secretBytes, logger,
124
 		*readTimeout, *writeTimeout, *preferIPv6, stat)
126
 		*readTimeout, *writeTimeout, *preferIPv6, stat)
127
 	}
129
 	}
128
 }
130
 }
129
 
131
 
132
+func printURLs(data interface{}) {
133
+	encoder := json.NewEncoder(os.Stdout)
134
+	encoder.SetEscapeHTML(false)
135
+	encoder.SetIndent("", "  ")
136
+
137
+	err := encoder.Encode(data)
138
+	if err != nil {
139
+		panic(err)
140
+	}
141
+}
142
+
130
 func usage(msg string) {
143
 func usage(msg string) {
131
 	io.WriteString(os.Stderr, msg+"\n")
144
 	io.WriteString(os.Stderr, msg+"\n")
132
 	os.Exit(1)
145
 	os.Exit(1)

Загрузка…
Отмена
Сохранить