feat: implement wakatime projects endpoint (resolve #196)

services/heartbeat.go

@@ -1,8 +1,11 @@
 package services
 
 import (
+	"fmt"
 	"github.com/muety/wakapi/config"
 	"github.com/muety/wakapi/repositories"
+	"github.com/muety/wakapi/utils"
+	"github.com/patrickmn/go-cache"
 	"time"
 
 	"github.com/muety/wakapi/models"
@@ -10,6 +13,7 @@ import (
 
 type HeartbeatService struct {
 	config              *config.Config
+	cache               *cache.Cache
 	repository          repositories.IHeartbeatRepository
 	languageMappingSrvc ILanguageMappingService
 }
@@ -17,12 +21,14 @@ type HeartbeatService struct {
 func NewHeartbeatService(heartbeatRepo repositories.IHeartbeatRepository, languageMappingService ILanguageMappingService) *HeartbeatService {
 	return &HeartbeatService{
 		config:              config.Get(),
+		cache:               cache.New(24*time.Hour, 24*time.Hour),
 		repository:          heartbeatRepo,
 		languageMappingSrvc: languageMappingService,
 	}
 }
 
 func (srv *HeartbeatService) Insert(heartbeat *models.Heartbeat) error {
+	srv.updateEntityUserCacheByHeartbeat(heartbeat)
 	return srv.repository.InsertBatch([]*models.Heartbeat{heartbeat})
 }
 
@@ -36,6 +42,7 @@ func (srv *HeartbeatService) InsertBatch(heartbeats []*models.Heartbeat) error {
 			filteredHeartbeats = append(filteredHeartbeats, hb)
 			hashes[hb.Hash] = true
 		}
+		srv.updateEntityUserCacheByHeartbeat(hb)
 	}
 
 	return srv.repository.InsertBatch(filteredHeartbeats)
@@ -73,6 +80,20 @@ func (srv *HeartbeatService) GetFirstByUsers() ([]*models.TimeByUser, error) {
 	return srv.repository.GetFirstByUsers()
 }
 
+func (srv *HeartbeatService) GetEntitySetByUser(entityType uint8, user *models.User) ([]string, error) {
+	cacheKey := srv.getEntityUserCacheKey(entityType, user)
+	if results, found := srv.cache.Get(cacheKey); found {
+		return utils.SetToStrings(results.(map[string]bool)), nil
+	}
+
+	results, err := srv.repository.GetEntitySetByUser(entityType, user)
+	if err != nil {
+		return nil, err
+	}
+	srv.cache.Set(cacheKey, utils.StringsToSet(results), cache.DefaultExpiration)
+	return results, nil
+}
+
 func (srv *HeartbeatService) DeleteBefore(t time.Time) error {
 	return srv.repository.DeleteBefore(t)
 }
@@ -89,3 +110,26 @@ func (srv *HeartbeatService) augmented(heartbeats []*models.Heartbeat, userId st
 
 	return heartbeats, nil
 }
+
+func (srv *HeartbeatService) getEntityUserCacheKey(entityType uint8, user *models.User) string {
+	return fmt.Sprintf("entity_set_%d_%s", entityType, user.ID)
+}
+
+func (srv *HeartbeatService) updateEntityUserCache(entityType uint8, entityKey string, user *models.User) {
+	cacheKey := srv.getEntityUserCacheKey(entityType, user)
+	if entities, found := srv.cache.Get(cacheKey); found {
+		if _, ok := entities.(map[string]bool)[entityKey]; !ok {
+			// new project / language / ..., which is not yet present in cache, arrived as part of a heartbeats
+			// -> invalidate cache
+			srv.cache.Delete(cacheKey)
+		}
+	}
+}
+
+func (srv *HeartbeatService) updateEntityUserCacheByHeartbeat(hb *models.Heartbeat) {
+	srv.updateEntityUserCache(models.SummaryProject, hb.Project, hb.User)
+	srv.updateEntityUserCache(models.SummaryLanguage, hb.Language, hb.User)
+	srv.updateEntityUserCache(models.SummaryEditor, hb.Editor, hb.User)
+	srv.updateEntityUserCache(models.SummaryOS, hb.OperatingSystem, hb.User)
+	srv.updateEntityUserCache(models.SummaryMachine, hb.Machine, hb.User)
+}

services/services.go

@@ -35,6 +35,7 @@ type IHeartbeatService interface {
 	GetFirstByUsers() ([]*models.TimeByUser, error)
 	GetLatestByUser(*models.User) (*models.Heartbeat, error)
 	GetLatestByOriginAndUser(string, *models.User) (*models.Heartbeat, error)
+	GetEntitySetByUser(uint8, *models.User) ([]string, error)
 	DeleteBefore(time.Time) error
 }