1
0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00

Fix caching.

Introduce live mode.
This commit is contained in:
Ferdinand Mütsch 2019-05-19 20:34:26 +02:00
parent d5253ba46b
commit 1caf962917

View File

@ -1,7 +1,10 @@
package routes package routes
import ( import (
"crypto/md5"
"fmt"
"net/http" "net/http"
"strconv"
"time" "time"
"github.com/n1try/wakapi/models" "github.com/n1try/wakapi/models"
@ -16,7 +19,7 @@ const (
IntervalLastYear string = "year" IntervalLastYear string = "year"
) )
var summaryCache map[time.Time]*models.Summary var summaryCache map[string]*models.Summary
type SummaryHandler struct { type SummaryHandler struct {
SummarySrvc *services.SummaryService SummarySrvc *services.SummaryService
@ -51,22 +54,42 @@ func (h *SummaryHandler) Get(w http.ResponseWriter, r *http.Request) {
} }
} }
if _, ok := summaryCache[from]; !ok { live := params.Get("live") != "" && params.Get("live") != "false"
to := utils.StartOfDay()
if live {
to = time.Now()
}
var summary *models.Summary
cacheKey := getHash([]time.Time{from, to})
if _, ok := summaryCache[cacheKey]; !ok {
// Cache Miss // Cache Miss
summary, err := h.SummarySrvc.GetSummary(from, utils.StartOfDay(), user) // 'to' is always constant fmt.Println("Cache miss")
summary, err = h.SummarySrvc.GetSummary(from, to, user) // 'to' is always constant
if err != nil { if err != nil {
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
return return
} }
summaryCache[from] = summary if !live {
summaryCache[cacheKey] = summary
}
} else {
summary, _ = summaryCache[cacheKey]
} }
summary, _ := summaryCache[from]
utils.RespondJSON(w, http.StatusOK, summary) utils.RespondJSON(w, http.StatusOK, summary)
} }
func tryInitCache() { func tryInitCache() {
if summaryCache == nil { if summaryCache == nil {
summaryCache = make(map[time.Time]*models.Summary) summaryCache = make(map[string]*models.Summary)
} }
} }
func getHash(times []time.Time) string {
digest := md5.New()
for _, t := range times {
digest.Write([]byte(strconv.Itoa(int(t.Unix()))))
}
return string(digest.Sum(nil))
}