api/count added

main.go

@@ -15,6 +15,11 @@ go build main.go
 sudo systemctl start server.service
 */
 
+func init() {
+	log.SetFlags(log.LstdFlags)
+	log.SetPrefix("hikan.ru | ")
+}
+
 func main() {
 	var app *models.App
 	var err error
@@ -64,6 +69,11 @@ func setupRoutes(a *models.App) *http.ServeMux {
 		router.Handle("/", m(controllers_pages.MainPageHandler(a)))
 	}
 
+	// api
+	{
+		router.Handle("/api/count/", m(controllers.CountHandler()))
+	}
+
 	return router
 }
 

mvc/controllers/api.go

@@ -0,0 +1,24 @@
+package controllers
+
+import (
+	"log"
+	"main/tools"
+	"net/http"
+)
+
+func CountHandler() http.HandlerFunc {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		var count []byte
+		var err error
+		if count, err = tools.GetJournalctlLogs("server", "hikan.ru", 24); err != nil {
+			log.Printf("%s", err.Error())
+		}
+		SendCount(w, count)
+	})
+}
+
+func SendCount(w http.ResponseWriter, data []byte) {
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.WriteHeader(http.StatusOK)
+	w.Write(data)
+}

mvc/controllers/middlewares.go

@@ -38,6 +38,6 @@ func LoggingMiddleware(next http.Handler) http.Handler {
 
 		next.ServeHTTP(w, r)
 
-		log.Printf("%s %s in %v", r.Method, r.URL.Path, time.Since(start))
+		log.Printf("%s %s %v", r.Method, r.URL.Path, time.Since(start))
 	})
 }

tools/logs.go

@@ -0,0 +1,29 @@
+package tools
+
+import (
+	"bytes"
+	"fmt"
+	"os/exec"
+	"runtime"
+)
+
+func GetJournalctlLogs(serviceName, grepPattern string, hoursAgo int) ([]byte, error) {
+
+	if runtime.GOOS != "linux" {
+		return nil, fmt.Errorf("not a linux")
+	}
+
+	cmd := exec.Command("sh", "-c",
+		fmt.Sprintf("journalctl -u %s --since '%d hours ago' | grep -c '%s'",
+			serviceName, hoursAgo, grepPattern))
+
+	var output bytes.Buffer
+	cmd.Stdout = &output
+
+	err := cmd.Run()
+	if err != nil {
+		return nil, err
+	}
+
+	return output.Bytes(), nil
+}