103 lines
3.3 KiB
Go
103 lines
3.3 KiB
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"seanime/internal/database/models"
|
|
"seanime/internal/util/result"
|
|
)
|
|
|
|
var mangaMappingCache = result.NewResultMap[string, *models.MangaMapping]()
|
|
|
|
func formatMangaMappingCacheKey(provider string, mediaId int) string {
|
|
return fmt.Sprintf("%s$%d", provider, mediaId)
|
|
}
|
|
|
|
func (db *Database) GetMangaMapping(provider string, mediaId int) (*models.MangaMapping, bool) {
|
|
|
|
if res, ok := mangaMappingCache.Get(formatMangaMappingCacheKey(provider, mediaId)); ok {
|
|
return res, true
|
|
}
|
|
|
|
var res models.MangaMapping
|
|
err := db.gormdb.Where("provider = ? AND media_id = ?", provider, mediaId).First(&res).Error
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
mangaMappingCache.Set(formatMangaMappingCacheKey(provider, mediaId), &res)
|
|
|
|
return &res, true
|
|
}
|
|
|
|
func (db *Database) InsertMangaMapping(provider string, mediaId int, mangaId string) error {
|
|
mapping := models.MangaMapping{
|
|
Provider: provider,
|
|
MediaID: mediaId,
|
|
MangaID: mangaId,
|
|
}
|
|
|
|
mangaMappingCache.Set(formatMangaMappingCacheKey(provider, mediaId), &mapping)
|
|
|
|
return db.gormdb.Save(&mapping).Error
|
|
}
|
|
|
|
func (db *Database) DeleteMangaMapping(provider string, mediaId int) error {
|
|
err := db.gormdb.Where("provider = ? AND media_id = ?", provider, mediaId).Delete(&models.MangaMapping{}).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mangaMappingCache.Delete(formatMangaMappingCacheKey(provider, mediaId))
|
|
return nil
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var mangaChapterContainerCache = result.NewResultMap[string, *models.MangaChapterContainer]()
|
|
|
|
func formatMangaChapterContainerCacheKey(provider string, mediaId int, chapterId string) string {
|
|
return fmt.Sprintf("%s$%d$%s", provider, mediaId, chapterId)
|
|
}
|
|
|
|
func (db *Database) GetMangaChapterContainer(provider string, mediaId int, chapterId string) (*models.MangaChapterContainer, bool) {
|
|
|
|
if res, ok := mangaChapterContainerCache.Get(formatMangaChapterContainerCacheKey(provider, mediaId, chapterId)); ok {
|
|
return res, true
|
|
}
|
|
|
|
var res models.MangaChapterContainer
|
|
err := db.gormdb.Where("provider = ? AND media_id = ? AND chapter_id = ?", provider, mediaId, chapterId).First(&res).Error
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
mangaChapterContainerCache.Set(formatMangaChapterContainerCacheKey(provider, mediaId, chapterId), &res)
|
|
|
|
return &res, true
|
|
}
|
|
|
|
func (db *Database) InsertMangaChapterContainer(provider string, mediaId int, chapterId string, chapterContainer []byte) error {
|
|
container := models.MangaChapterContainer{
|
|
Provider: provider,
|
|
MediaID: mediaId,
|
|
ChapterID: chapterId,
|
|
Data: chapterContainer,
|
|
}
|
|
|
|
mangaChapterContainerCache.Set(formatMangaChapterContainerCacheKey(provider, mediaId, chapterId), &container)
|
|
|
|
return db.gormdb.Save(&container).Error
|
|
}
|
|
|
|
func (db *Database) DeleteMangaChapterContainer(provider string, mediaId int, chapterId string) error {
|
|
err := db.gormdb.Where("provider = ? AND media_id = ? AND chapter_id = ?", provider, mediaId, chapterId).Delete(&models.MangaChapterContainer{}).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mangaChapterContainerCache.Delete(formatMangaChapterContainerCacheKey(provider, mediaId, chapterId))
|
|
return nil
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|