add user info metric

Author: haozibi

app/app.go

@@ -35,6 +35,9 @@ type APP struct {
 	userMap   map[string]time.Time
 	recordMap map[string]time.Time
 
+	userInfoMu sync.Mutex
+	userInfo   map[string]time.Time
+
 	wg               tools.WaitGroupWrapper
 	shutdown         chan struct{}
 	shutdownComplete chan struct{}
@@ -49,6 +52,7 @@ func New(cfg Config) *APP {
 		group:            new(singleflight.Group),
 		userMap:          make(map[string]time.Time),
 		recordMap:        make(map[string]time.Time),
+		userInfo:         make(map[string]time.Time),
 		shutdown:         make(chan struct{}),
 		shutdownComplete: make(chan struct{}),
 	}
@@ -89,6 +93,9 @@ func (a *APP) Run() error {
 	a.wg.Wrap(func() {
 		exitFunc(a.runHTTP())
 	})
+	a.wg.Wrap(func() {
+		exitFunc(a.runInterHTTP())
+	})
 
 	if enable {
 		a.Cron(CronSpec)
@@ -111,12 +118,17 @@ func (a *APP) runHTTP() error {
 
 	r := mux.NewRouter()
 	Router(r, a, ioutil.Discard)
+	handle := handlers.RecoveryHandler()(handlers.CompressHandler(r))
+	return a.runhttp(address, handle)
+}
+
+func (a *APP) runhttp(address string, handle http.Handler) error {
 
 	srv := &http.Server{
 		Addr:         address,
 		WriteTimeout: 120 * time.Second,
 		ReadTimeout:  120 * time.Second,
-		Handler:      handlers.RecoveryHandler()(handlers.CompressHandler(r)),
+		Handler:      handle,
 	}
 
 	go func() {

app/config.go

@@ -1,9 +1,10 @@
 package app
 
 type Config struct {
-	Address    string
-	SqlitePath string
-	EnableCron bool
+	Address      string
+	DebugAddress int
+	SqlitePath   string
+	EnableCron   bool
 }
 
 type RedisConfig struct {

app/metrics.go

@@ -0,0 +1,58 @@
+package app
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"sort"
+	"time"
+
+	"github.com/gorilla/mux"
+)
+
+func (a *APP) addUser(name string, isCN bool) {
+	t := time.Now()
+
+	key := name + "_" + fmt.Sprintf("%v", isCN)
+
+	a.userInfoMu.Lock()
+	a.userInfo[key] = t
+	a.userInfoMu.Unlock()
+}
+
+func (a *APP) runInterHTTP() error {
+	var (
+		port = a.config.DebugAddress
+	)
+
+	r := mux.NewRouter()
+	r.HandleFunc("/user_info", a.ShowUserInfo)
+
+	return a.runhttp(fmt.Sprintf("127.0.0.1:%d", port), r)
+}
+
+type MetricUserInfo struct {
+	Name string `json:"name"`
+	// IsCN string `json:"is_cn"`
+	Time string `json:"last_time"`
+}
+
+func (a *APP) ShowUserInfo(w http.ResponseWriter, r *http.Request) {
+
+	a.userInfoMu.Lock()
+	list := make([]MetricUserInfo, 0, len(a.userInfo))
+	for k, v := range a.userInfo {
+		list = append(list, MetricUserInfo{
+			Name: k,
+			Time: v.Format("2006-01-02 15:04:05"),
+		})
+	}
+	a.userInfoMu.Unlock()
+
+	sort.Slice(list, func(i, j int) bool {
+		return list[i].Name < list[j].Name
+	})
+
+	body, _ := json.Marshal(list)
+	w.Write(body)
+}

app/router.go

@@ -119,6 +119,7 @@ func (a *APP) HandlerFunc(badgeType BadgeType, isCN bool) http.Handler {
 			w.WriteHeader(http.StatusNotFound)
 			return
 		}
+		a.addUser(name, isCN)
 		f(badgeType, name, isCN, w, r)
 	})
 }

cmd/run.go

@@ -32,6 +32,7 @@ func NewRunCommand() *cobra.Command {
 	flag := cmd.Flags()
 
 	flag.StringVarP(&opt.Address, "address", "", ":8080", "http listen address")
+	flag.IntVarP(&opt.DebugAddress, "debug-address", "", 18080, "http debug address, only localhost")
 	flag.StringVarP(&opt.SqlitePath, "sqlite-path", "", "./lc.db", "sqlite3 file path")
 	flag.BoolVarP(&opt.EnableCron, "enable-cron", "", false, "if enable cron")