Fixes tail size parameter (#22)

main.go

@@ -23,6 +23,7 @@ var (
 	addr    = ""
 	base    = ""
 	level   = ""
+	tailSize = 300
 	version = "dev"
 	commit  = "none"
 	date    = "unknown"
@@ -37,6 +38,7 @@ func init() {
 	pflag.String("addr", ":8080", "http service address")
 	pflag.String("base", "/", "base address of the application to mount")
 	pflag.String("level", "info", "logging level")
+	pflag.Int("tailSize", 300, "Tail size to use for initial container logs")
 	pflag.Parse()
 
 	viper.AutomaticEnv()
@@ -46,6 +48,7 @@ func init() {
 	addr = viper.GetString("addr")
 	base = viper.GetString("base")
 	level = viper.GetString("level")
+	tailSize = viper.GetInt("tailSize")
 
 	l, _ := log.ParseLevel(level)
 	log.SetLevel(l)
@@ -153,8 +156,7 @@ func (h *handler) streamLogs(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Streaming unsupported!", http.StatusInternalServerError)
 		return
 	}
-
-	messages, err := h.client.ContainerLogs(r.Context(), id)
+	messages, err := h.client.ContainerLogs(r.Context(), id, tailSize)
 
 	w.Header().Set("Content-Type", "text/event-stream")
 	w.Header().Set("Cache-Control", "no-cache")