Fix the cache

Author: AchoArnold

api/pkg/di/container.go

@@ -1321,21 +1321,21 @@ func (container *Container) UserRepository() repositories.UserRepository {
 	return repositories.NewGormUserRepository(
 		container.Logger(),
 		container.Tracer(),
-		container.RistrettoCache(),
+		container.UserRistrettoCache(),
 		container.DB(),
 	)
 }
 
-// RistrettoCache creates an in-memory *ristretto.Cache
-func (container *Container) RistrettoCache() (cache *ristretto.Cache) {
+// UserRistrettoCache creates an in-memory *ristretto.Cache[string, entities.AuthUser]
+func (container *Container) UserRistrettoCache() (cache *ristretto.Cache[string, entities.AuthUser]) {
 	container.logger.Debug(fmt.Sprintf("creating %T", cache))
-	ristrettoCache, err := ristretto.NewCache(&ristretto.Config{
+	ristrettoCache, err := ristretto.NewCache[string, entities.AuthUser](&ristretto.Config[string, entities.AuthUser]{
 		MaxCost:     5000,
 		NumCounters: 5000 * 10,
 		BufferItems: 64,
 	})
 	if err != nil {
-		container.logger.Fatal(stacktrace.Propagate(err, "cannot create ristretto cache"))
+		container.logger.Fatal(stacktrace.Propagate(err, "cannot create user ristretto cache"))
 	}
 	return ristrettoCache
 }

api/pkg/repositories/gorm_user_repository.go

@@ -23,15 +23,15 @@ import (
 type gormUserRepository struct {
 	logger telemetry.Logger
 	tracer telemetry.Tracer
-	cache  *ristretto.Cache
+	cache  *ristretto.Cache[string, entities.AuthUser]
 	db     *gorm.DB
 }
 
 // NewGormUserRepository creates the GORM version of the UserRepository
 func NewGormUserRepository(
 	logger telemetry.Logger,
 	tracer telemetry.Tracer,
-	cache *ristretto.Cache,
+	cache *ristretto.Cache[string, entities.AuthUser],
 	db *gorm.DB,
 ) UserRepository {
 	return &gormUserRepository{
@@ -119,8 +119,8 @@ func (repository *gormUserRepository) LoadAuthUser(ctx context.Context, apiKey s
 	defer span.End()
 
 	if authUser, found := repository.cache.Get(apiKey); found {
-		ctxLogger.Info(fmt.Sprintf("cache hit for user with ID [%s]", authUser.(entities.AuthUser).ID))
-		return authUser.(entities.AuthUser), nil
+		ctxLogger.Info(fmt.Sprintf("cache hit for user with ID [%s]", authUser.ID))
+		return authUser, nil
 	}
 
 	user := new(entities.User)