Github/Mensa-API
1
0
Fork 0
mirror of https://github.com/LeRoid-hub/Mensa-API.git synced 2025-01-31 03:34:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

cache for mensa

Browse Source
This commit is contained in:
Jan 2024-09-15 21:39:17 +02:00
parent a5bb066df0
commit c036ce104b
5 changed files with 26 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cache/cache.go vendored
View File

@ -1,6 +1,8 @@
package cache
import (
"errors"
"github.com/LeRoid-hub/Mensa-API/models"
)
@ -11,15 +13,15 @@ func HasCacheData(key string) bool {
return ok
}
func GetCacheData(key string) string {
func GetCacheData(key string) (models.Mensa, error) {
Item, ok := Cache[key]
if !ok {
return ""
return models.Mensa{}, errors.New("no data in cache")
}
return Item.GetData()
}
func SetCacheData(key string, data string, lifetime ...int64) {
func SetCacheData(key string, data models.Mensa, lifetime ...int64) {
Item, ok := Cache[key]
if !ok {
Item = models.CacheItem{}

19
models/cacheItem.go
View File

@ -1,14 +1,17 @@
package models
import "time"
import (
"errors"
"time"
)
type CacheItem struct {
data string
data Mensa
lastUpdated time.Time
lifetime int64
}
func (c *CacheItem) SetData(data string, lifetime ...int64) {
func (c *CacheItem) SetData(data Mensa, lifetime ...int64) {
if len(lifetime) > 0 {
c.lifetime = lifetime[0]
} else {
@ -19,14 +22,14 @@ func (c *CacheItem) SetData(data string, lifetime ...int64) {
c.lastUpdated = time.Now()
}
func (c *CacheItem) GetData() string {
func (c *CacheItem) GetData() (Mensa, error) {
if time.Now().Unix()-c.lastUpdated.Unix() > c.lifetime {
return ""
return Mensa{}, errors.New("cache expired")
}
if c.data == "" {
return ""
if c.lastUpdated.IsZero() {
return Mensa{}, errors.New("no data in cache")
}
return c.data
return c.data, nil
}

6
server/city.go
View File

@ -1,7 +1,6 @@
package server
import (
"github.com/LeRoid-hub/Mensa-API/cache"
"github.com/LeRoid-hub/Mensa-API/fetch"
"github.com/LeRoid-hub/Mensa-API/scrape"
"github.com/gin-gonic/gin"
@ -16,11 +15,6 @@ func city(c *gin.Context) {
return
}
if cache.HasCacheData(city) {
c.JSON(200, cache.GetCacheData(city))
return
}
resp, err := fetch.Fetch(city)
if err != nil {
c.JSON(500, gin.H{

11
server/mensa.go
View File

@ -24,7 +24,14 @@ func mensa(c *gin.Context) {
}
if cache.HasCacheData(city + "/" + mensa) {
c.JSON(200, cache.GetCacheData(mensa))
cacheData, err := cache.GetCacheData(city + "/" + mensa)
if err != nil {
c.JSON(500, gin.H{
"error": err.Error(),
})
return
}
c.JSON(200, cacheData)
return
}
@ -46,7 +53,7 @@ func mensa(c *gin.Context) {
scraped := scrape.ScrapeMensa(resp.Body)
//cache.SetCacheData(city+"/"+mensa, scraped)
cache.SetCacheData(city+"/"+mensa, scraped)
c.JSON(200, scraped)
}

7
server/state.go
View File

@ -1,7 +1,6 @@
package server
import (
"github.com/LeRoid-hub/Mensa-API/cache"
"github.com/LeRoid-hub/Mensa-API/fetch"
"github.com/LeRoid-hub/Mensa-API/scrape"
"github.com/gin-gonic/gin"
@ -16,11 +15,6 @@ func state(c *gin.Context) {
return
}
if cache.HasCacheData(state) {
c.JSON(200, cache.GetCacheData(state))
return
}
resp, err := fetch.Fetch(state + ".html")
if err != nil {
c.JSON(500, gin.H{
@ -38,5 +32,6 @@ func state(c *gin.Context) {
}
scraped := scrape.ScrapeState(resp.Body)
c.JSON(200, scraped)
}