X-Git-Url: http://git.scottworley.com/reliable-chat/blobdiff_plain/0537aeda2c8cebca1d20fbf6477e1e13cc70fc76..cf44a97649f42952e85dca82a91f18fc4e14c1e9:/server/server.go?ds=inline diff --git a/server/server.go b/server/server.go index 695ec10..8dce69b 100644 --- a/server/server.go +++ b/server/server.go @@ -28,11 +28,14 @@ import "time" var port = flag.Int("port", 21059, "Port to listen on") var localaddress = flag.String("localaddress", "", "Local address to bind to") +var max_messages = flag.Int("maxmessages", 1000, "Maximum number of messages to retain") +var start_time = expvar.NewInt("start_time") 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") +var drop_due_to_limit_count = expvar.NewInt("drop_due_to_limit_count") type Message struct { Time time.Time @@ -55,7 +58,6 @@ type Store struct { func manage_store(store Store) { messages := list.New() message_count := 0 - max_messages := 1000 waiting := list.New() main: for { @@ -72,10 +74,11 @@ main: } waiting.Init() messages.PushBack(new_message) - if message_count < max_messages { + if message_count < *max_messages { message_count++ } else { messages.Remove(messages.Front()) + drop_due_to_limit_count.Add(1) } case request, ok := <-store.Get: if !ok { @@ -152,7 +155,7 @@ func start_server(store Store) { }) http.HandleFunc("/robots.txt", func(w http.ResponseWriter, r *http.Request) { - w.Write([]byte(robots_txt)); + w.Write([]byte(robots_txt)) }) log.Fatal(http.ListenAndServe(*localaddress+":"+strconv.Itoa(*port), nil)) @@ -160,6 +163,7 @@ func start_server(store Store) { func main() { flag.Parse() + start_time.Set(time.Now().UnixNano()) store := start_store() start_server(store) }