From 75cf4fbb4687e12dfd249f7acd9e4511d8f6e052 Mon Sep 17 00:00:00 2001
From: Kane York <rikingcoding@gmail.com>
Date: Wed, 16 Dec 2015 11:47:51 -0800
Subject: [PATCH] rename package to 'logstasher'

---
 socketserver/server/handlecore.go                      | 10 ++++++----
 .../server/{logstash => logstasher}/elasticsearch.go   |  3 ++-
 2 files changed, 8 insertions(+), 5 deletions(-)
 rename socketserver/server/{logstash => logstasher}/elasticsearch.go (98%)

diff --git a/socketserver/server/handlecore.go b/socketserver/server/handlecore.go
index 4d96f704..d502d071 100644
--- a/socketserver/server/handlecore.go
+++ b/socketserver/server/handlecore.go
@@ -20,7 +20,7 @@ import (
 	"time"
 	"unicode/utf8"
 
-	"./logstash"
+	"./logstasher"
 )
 
 // SuccessCommand is a Reply Command to indicate success in reply to a C2S Command.
@@ -96,6 +96,8 @@ func SetupServerAndHandle(config *ConfigFile, serveMux *http.ServeMux) {
 		resp.Body.Close()
 	}
 
+	logstasher.Setup(Configuration.ESServer, Configuration.ESIndexPrefix, Configuration.ESHostName)
+
 	go authorizationJanitor()
 	go bunchCacheJanitor()
 	go pubsubJanitor()
@@ -253,7 +255,7 @@ func RunSocketConnection(conn *websocket.Conn) {
 	// Close the connection when we're done.
 	defer closer()
 
-	var report logstash.ConnectionReport
+	var report logstasher.ConnectionReport
 	report.ConnectTime = time.Now()
 	report.RemoteAddr = conn.RemoteAddr()
 
@@ -306,7 +308,7 @@ func RunSocketConnection(conn *websocket.Conn) {
 		atomic.AddUint64(&Statistics.CurrentClientCount, NegativeOne)
 	}
 
-	logstash.Submit(report)
+	logstasher.Submit(&report)
 }
 
 func runSocketReader(conn *websocket.Conn, errorChan chan<- error, clientChan chan<- ClientMessage, stoppedChan <-chan struct{}) {
@@ -407,7 +409,7 @@ func getDeadline() time.Time {
 	return time.Now().Add(1 * time.Minute)
 }
 
-func closeConnection(conn *websocket.Conn, closeMsg websocket.CloseError, report *esConnectionReport) {
+func closeConnection(conn *websocket.Conn, closeMsg websocket.CloseError, report *logstasher.ConnectionReport) {
 	closeTxt := closeMsg.Text
 	if strings.Contains(closeTxt, "read: connection reset by peer") {
 		closeTxt = "read: connection reset by peer"
diff --git a/socketserver/server/logstash/elasticsearch.go b/socketserver/server/logstasher/elasticsearch.go
similarity index 98%
rename from socketserver/server/logstash/elasticsearch.go
rename to socketserver/server/logstasher/elasticsearch.go
index b8c88299..9eece14c 100644
--- a/socketserver/server/logstash/elasticsearch.go
+++ b/socketserver/server/logstasher/elasticsearch.go
@@ -1,4 +1,4 @@
-package server
+package logstasher
 
 import (
 	"bytes"
@@ -146,6 +146,7 @@ func checkServerPresent() {
 		resp.Body.Close()
 		serverPresent = true
 		submitChan = make(chan Report, 8)
+		fmt.Println("elasticsearch reports enabled")
 		go submissionWorker()
 	} else {
 		serverPresent = false