boyska 5 роки тому
батько
коміт
bf145240c2
3 змінених файлів з 10 додано та 9 видалено
  1. 7 6
      cmd/circologd/http.go
  2. 2 2
      cmd/circologd/main.go
  3. 1 1
      hub.go

+ 7 - 6
http.go → cmd/circologd/http.go

@@ -4,18 +4,19 @@ import (
 	"net/http"
 	"time"
 
+	"git.lattuga.net/boyska/circolog"
 	"github.com/gorilla/websocket"
 	"gopkg.in/mcuadros/go-syslog.v2/format"
 )
 
-func setupHttp(hub Hub) {
+func setupHttp(hub circolog.Hub) {
 	http.HandleFunc("/", getHTTPHandler(hub))
 	http.HandleFunc("/ws", getWSHandler(hub))
 }
 
-func getHTTPHandler(hub Hub) http.HandlerFunc {
+func getHTTPHandler(hub circolog.Hub) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
-		client := Client{
+		client := circolog.Client{
 			Messages: make(chan format.LogParts),
 			Nofollow: true}
 		hub.Register <- client
@@ -32,7 +33,7 @@ func getHTTPHandler(hub Hub) http.HandlerFunc {
 	}
 }
 
-func getWSHandler(hub Hub) http.HandlerFunc {
+func getWSHandler(hub circolog.Hub) http.HandlerFunc {
 	var upgrader = websocket.Upgrader{
 		ReadBufferSize:  1024,
 		WriteBufferSize: 1024,
@@ -42,12 +43,12 @@ func getWSHandler(hub Hub) http.HandlerFunc {
 		if err != nil {
 			return
 		}
-		client := Client{Messages: make(chan format.LogParts)}
+		client := circolog.Client{Messages: make(chan format.LogParts)}
 		hub.Register <- client
 
 		// Allow collection of memory referenced by the caller by doing all work in
 		// new goroutines.
-		go func(conn *websocket.Conn, c Client) {
+		go func(conn *websocket.Conn, c circolog.Client) {
 			defer func() {
 				hub.Unregister <- c
 				conn.Close()

+ 2 - 2
main.go → cmd/circologd/main.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"os"
 
+	"git.lattuga.net/boyska/circolog"
 	syslog "gopkg.in/mcuadros/go-syslog.v2"
 )
 
@@ -18,8 +19,7 @@ func main() {
 	queryAddr := flag.String("query-addr", "127.0.0.1:9080", "Address:port where to bind the query service")
 	flag.Parse()
 
-	var hub Hub
-	hub = NewHub(*bufsize)
+	hub := circolog.NewHub(*bufsize)
 	handler := syslog.NewChannelHandler(hub.LogMessages)
 
 	server := syslog.NewServer()

+ 1 - 1
hub.go

@@ -1,4 +1,4 @@
-package main
+package circolog
 
 import (
 	"container/ring"