ソースを参照

Serve the blacklist zone over HTTP

Signed-off-by: Antoine Tenart <antoine.tenart@ack.tf>
Antoine Tenart 5 年 前
コミット
f73e023ca8
1 ファイル変更5 行追加4 行削除
  1. 5 4
      main.go

+ 5 - 4
main.go

@@ -37,11 +37,11 @@ var (
 )
 
 func main() {
-	fmt.Print(soa)
-	fmt.Print(generateList())
+	http.HandleFunc("/", generateList)
+	http.ListenAndServe(":8000", nil)
 }
 
-func generateList() string {
+func generateList(w http.ResponseWriter, r *http.Request) {
 	c := make(chan string)
 	for _, list := range lists {
 		go handle(list, c)
@@ -52,7 +52,8 @@ func generateList() string {
 		zones = append(zones, <-c)
 	}
 
-	return strings.Join(zones, "\n")
+	fmt.Fprintf(w, soa)
+	fmt.Fprintf(w, strings.Join(zones, "\n"))
 }
 
 func handle(url string, zone chan string) {