diff options
Diffstat (limited to 'internal/cache')
| -rw-r--r-- | internal/cache/account.go | 6 | ||||
| -rw-r--r-- | internal/cache/status.go | 6 |
2 files changed, 10 insertions, 2 deletions
diff --git a/internal/cache/account.go b/internal/cache/account.go index bb402d60f..98f1684a4 100644 --- a/internal/cache/account.go +++ b/internal/cache/account.go @@ -4,6 +4,7 @@ import ( "sync" "github.com/ReneKroon/ttlcache" + "github.com/sirupsen/logrus" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" ) @@ -26,7 +27,10 @@ func NewAccountCache() *AccountCache { // Set callback to purge lookup maps on expiration c.cache.SetExpirationCallback(func(key string, value interface{}) { - account := value.(*gtsmodel.Account) + account, ok := value.(*gtsmodel.Account) + if !ok { + logrus.Panicf("AccountCache could not assert entry with key %s to *gtsmodel.Account", key) + } c.mutex.Lock() delete(c.urls, account.URL) diff --git a/internal/cache/status.go b/internal/cache/status.go index 5ea528055..1e92c9c63 100644 --- a/internal/cache/status.go +++ b/internal/cache/status.go @@ -4,6 +4,7 @@ import ( "sync" "github.com/ReneKroon/ttlcache" + "github.com/sirupsen/logrus" "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" ) @@ -26,7 +27,10 @@ func NewStatusCache() *StatusCache { // Set callback to purge lookup maps on expiration c.cache.SetExpirationCallback(func(key string, value interface{}) { - status := value.(*gtsmodel.Status) + status, ok := value.(*gtsmodel.Status) + if !ok { + logrus.Panicf("StatusCache could not assert entry with key %s to *gtsmodel.Status", key) + } c.mutex.Lock() delete(c.urls, status.URL) |