depend on io.WriteCloser instead of net.Conn

This commit is contained in:
bartfokker 2018-10-25 22:22:04 +02:00
parent 8d8f1e72eb
commit 78caad2dbd

View File

@ -2,6 +2,7 @@ package logging
import ( import (
"fmt" "fmt"
"io"
"net" "net"
"sync" "sync"
) )
@ -56,13 +57,13 @@ func StopServer() {
} }
} }
func handler(conn net.Conn) { func handler(wc io.WriteCloser) {
server.wg.Add(1) server.wg.Add(1)
defer server.wg.Done() defer server.wg.Done()
defer conn.Close() defer wc.Close()
for msg := range Log.tail() { for msg := range Log.tail() {
msg = fmt.Sprintf("%s\n", msg) msg = fmt.Sprintf("%s\n", msg)
conn.Write([]byte(msg)) wc.Write([]byte(msg))
} }
conn.Write([]byte("bye\n")) wc.Write([]byte("bye\n"))
} }