X-Git-Url: http://git.scottworley.com/reliable-chat/blobdiff_plain/c282d87802c8a36e1ef5e4b367e7e753c2ba05f5..b199796a3d2c82a84243e3a5ad3099315d8f1b55:/server/server.go?ds=inline

diff --git a/server/server.go b/server/server.go
index 7572447..8ce9678 100644
--- a/server/server.go
+++ b/server/server.go
@@ -2,12 +2,23 @@ package main
 
 import "container/list"
 import "encoding/json"
+import "expvar"
+import "flag"
 import "log"
 import "net/http"
+import "strconv"
 import "time"
 
+var port = flag.Int("port", 21059, "Port to listen on")
+
+var speak_count = expvar.NewInt("speak_count")
+var fetch_count = expvar.NewInt("fetch_count")
+var fetch_wait_count = expvar.NewInt("fetch_wait_count")
+var fetch_wake_count = expvar.NewInt("fetch_wake_count")
+
 type Message struct {
 	Time time.Time
+	ID   string
 	Text string
 }
 
@@ -35,12 +46,14 @@ main:
 			if !ok {
 				break main
 			}
-			messages.PushBack(new_message)
+			speak_count.Add(1)
 			for waiter := waiting.Front(); waiter != nil; waiter = waiter.Next() {
 				waiter.Value.(*StoreRequest).Messages <- []Message{*new_message}
 				close(waiter.Value.(*StoreRequest).Messages)
+				fetch_wake_count.Add(1)
 			}
 			waiting.Init()
+			messages.PushBack(new_message)
 			if message_count < max_messages {
 				message_count++
 			} else {
@@ -50,8 +63,10 @@ main:
 			if !ok {
 				break main
 			}
+			fetch_count.Add(1)
 			if messages.Back() == nil || !request.StartTime.Before(messages.Back().Value.(*Message).Time) {
 				waiting.PushBack(request)
+				fetch_wait_count.Add(1)
 			} else {
 				start := messages.Back()
 				response_size := 1
@@ -107,10 +122,13 @@ func start_server(store Store) {
 	})
 
 	http.HandleFunc("/speak", func(w http.ResponseWriter, r *http.Request) {
-		store.Add <- &Message{time.Now(), r.FormValue("text")}
+		store.Add <- &Message{
+			time.Now(),
+			r.FormValue("id"),
+			r.FormValue("text")}
 	})
 
-	log.Fatal(http.ListenAndServe(":8080", nil))
+	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(*port), nil))
 }
 
 func main() {