pikapika/go/main/database/comic_center/center.go

605 lines
15 KiB
Go

package comic_center
import (
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"path"
"pikapika/main/utils"
"sync"
"time"
)
var mutex = sync.Mutex{}
var db *gorm.DB
func InitDBConnect(databaseDir string) {
mutex.Lock()
defer mutex.Unlock()
var err error
db, err = gorm.Open(sqlite.Open(path.Join(databaseDir, "comic_center.db")), utils.GormConfig)
if err != nil {
panic("failed to connect database")
}
db.AutoMigrate(&Category{})
db.AutoMigrate(&ComicView{})
db.AutoMigrate(&RemoteImage{})
db.AutoMigrate(&ComicDownload{})
db.AutoMigrate(&ComicDownloadEp{})
db.AutoMigrate(&ComicDownloadPicture{})
}
func Transaction(t func(tx *gorm.DB) error) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(t)
}
func UpSetCategories(categories *[]Category) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
var in []string
for _, c := range *categories {
if c.ID == "" {
continue
}
in = append(in, c.ID)
err := tx.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.AssignmentColumns([]string{
"updated_at",
"title",
"description",
"is_web",
"active",
"link",
"thumb_original_name",
"thumb_file_server",
"thumb_path",
}),
}).Create(&c).Error
if err != nil {
return err
}
}
err := tx.Unscoped().Model(&Category{}).Where(" id in ?", in).Update("deleted_at", gorm.DeletedAt{
Valid: false,
}).Error
if err != nil {
return err
}
return tx.Unscoped().Model(&Category{}).Where(" id not in ?", in).Update("deleted_at", gorm.DeletedAt{
Time: time.Now(),
Valid: true,
}).Error
})
}
func NoLockActionViewComicUpdateInfoDB(view *ComicView, db *gorm.DB) error {
view.LastViewTime = time.Now()
return db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.AssignmentColumns([]string{
"created_at",
"updated_at",
"title",
"author",
"pages_count",
"eps_count",
"finished",
"categories",
"thumb_original_name",
"thumb_file_server",
"thumb_path",
"likes_count",
"description",
"chinese_team",
"tags",
"allow_download",
"views_count",
"is_favourite",
"is_liked",
"comments_count",
"last_view_time",
}),
}).Create(view).Error
}
func ViewComicUpdateInfo(view *ComicView) error {
mutex.Lock()
defer mutex.Unlock()
return NoLockActionViewComicUpdateInfoDB(view, db)
}
func ViewComic(comicId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicView{}).Where(
"id = ?", comicId,
).Update(
"last_view_time",
time.Now(),
).Error
}
func ViewComicUpdateFavourite(comicId string, favourite bool) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicView{}).Where(
"id = ?", comicId,
).Update(
"is_favourite",
favourite,
).Error
}
func ViewComicUpdateLike(comicId string, like bool) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicView{}).Where(
"id = ?", comicId,
).Update(
"is_like",
like,
).Error
}
func ViewEpAndPicture(comicId string, epOrder int, epTitle string, pictureRank int) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicView{}).Where("id", comicId).Updates(
map[string]interface{}{
"last_view_time": time.Now(),
"last_view_ep_order": epOrder,
"last_view_ep_title": epTitle,
"last_view_picture_rank": pictureRank,
},
).Error
}
func LoadViewLog(comicId string) (*ComicView, error) {
mutex.Lock()
defer mutex.Unlock()
var view ComicView
err := db.First(&view, "id = ?", comicId).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
if err != nil {
return nil, err
}
return &view, nil
}
func FindRemoteImage(fileServer string, path string) *RemoteImage {
mutex.Lock()
defer mutex.Unlock()
var remoteImage RemoteImage
err := db.First(&remoteImage, "file_server = ? AND path = ?", fileServer, path).Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return nil
} else {
panic(err)
}
}
return &remoteImage
}
func SaveRemoteImage(remote *RemoteImage) error {
mutex.Lock()
defer mutex.Unlock()
return db.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "file_server"}, {Name: "path"}},
DoUpdates: clause.AssignmentColumns([]string{
"updated_at",
"file_size",
"format",
"width",
"height",
"local_path",
}),
}).Create(remote).Error
}
func CreateDownload(comic *ComicDownload, epList *[]ComicDownloadEp) error {
mutex.Lock()
defer mutex.Unlock()
comic.SelectedEpCount = int32(len(*epList))
return db.Transaction(func(tx *gorm.DB) error {
err := tx.Create(comic).Error
if err != nil {
return err
}
for _, ep := range *epList {
err := tx.Create(&ep).Error
if err != nil {
return err
}
}
return nil
})
}
func AddDownload(comic *ComicDownload, epList *[]ComicDownloadEp) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
err := tx.Model(comic).Where("id = ?", comic.ID).Updates(map[string]interface{}{
"created_at": comic.CreatedAt,
"updated_at": comic.UpdatedAt,
"title": comic.Title,
"author": comic.Author,
"pages_count": comic.PagesCount,
"eps_count": comic.EpsCount,
"finished": comic.Finished,
"categories": comic.Categories,
"thumb_original_name": comic.ThumbOriginalName,
"thumb_file_server": comic.ThumbFileServer,
"thumb_path": comic.ThumbPath,
"description": comic.Description,
"chinese_team": comic.ChineseTeam,
"tags": comic.Tags,
"download_finished": false, // restart
}).Error
if err != nil {
return err
}
err = tx.Exec(
"UPDATE comic_downloads SET eps_count = selected_ep_count + ? WHERE id = ?",
len(*epList), comic.ID,
).Error
if err != nil {
return err
}
for _, ep := range *epList {
err := tx.Create(&ep).Error
if err != nil {
return err
}
}
return nil
})
}
func UpdateDownloadLogo(comicId string, fileSize int64, format string, width int32, height int32, localPath string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownload{}).Where("id = ?", comicId).Updates(map[string]interface{}{
"thumb_file_size": fileSize,
"thumb_format": format,
"thumb_width": width,
"thumb_height": height,
"thumb_local_path": localPath,
}).Error
}
func FindComicDownloadById(comicId string) (*ComicDownload, error) {
mutex.Lock()
defer mutex.Unlock()
var download ComicDownload
err := db.First(&download, "id = ?", comicId).Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &download, nil
}
func ListDownloadEpByComicId(comicId string) ([]ComicDownloadEp, error) {
mutex.Lock()
defer mutex.Unlock()
var epList []ComicDownloadEp
err := db.Where("comic_id = ?", comicId).Order("ep_order ASC").Find(&epList).Error
return epList, err
}
func ListDownloadPictureByEpId(epId string) ([]ComicDownloadPicture, error) {
mutex.Lock()
defer mutex.Unlock()
var pictureList []ComicDownloadPicture
err := db.Where("ep_id = ?", epId).Order("rank_in_ep ASC").Find(&pictureList).Error
return pictureList, err
}
func AllDownloads() (*[]ComicDownload, error) {
mutex.Lock()
defer mutex.Unlock()
var downloads []ComicDownload
err := db.Table("comic_downloads").
Joins("LEFT JOIN comic_views ON comic_views.id = comic_downloads.id").
Select("comic_downloads.*").
Order("comic_views.last_view_time DESC").
Scan(&downloads).Error
// err := db.Find(&downloads).Error
return &downloads, err
}
func LoadFirstNeedDownload() (*ComicDownload, error) {
mutex.Lock()
defer mutex.Unlock()
var download ComicDownload
err := db.First(&download, "download_failed = 0 AND pause = 0 AND deleting = 0 AND download_finished = 0").Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &download, nil
}
func LoadFirstNeedDownloadEp(comicId string) (*ComicDownloadEp, error) {
mutex.Lock()
defer mutex.Unlock()
var ep ComicDownloadEp
err := db.First(
&ep,
" comic_id = ? AND download_failed = 0 AND download_finished = 0 AND fetched_pictures = 1",
comicId,
).Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &ep, nil
}
// LoadNeedDownloadPictures 根据EP.ID获取需要下载的图片
func LoadNeedDownloadPictures(epId string, limit int) (*[]ComicDownloadPicture, error) {
mutex.Lock()
defer mutex.Unlock()
var pictures []ComicDownloadPicture
err := db.Find(
&pictures,
"ep_id = ? AND download_failed = 0 AND download_finished = 0 LIMIT ?",
epId, limit,
).Error
return &pictures, err
}
func FetchPictures(comicId string, epId string, list *[]ComicDownloadPicture) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
var rankInEp int32
for _, picture := range *list {
rankInEp = rankInEp + 1
picture.RankInEp = rankInEp
err := tx.Create(&picture).Error
if err != nil {
return err
}
}
err := tx.Model(&ComicDownloadEp{}).Where("id = ?", epId).Updates(map[string]interface{}{
"fetched_pictures": true,
"selected_picture_count": len(*list),
}).Error
if err != nil {
return err
}
return tx.Exec(
"UPDATE comic_downloads SET selected_picture_count = selected_picture_count + ? WHERE id = ?",
len(*list), comicId,
).Error
})
}
func DownloadFailed(comicId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownload{}).Where("id = ?", comicId).Update("download_failed", true).Error
}
func DownloadSuccess(comicId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownload{}).Where("id = ?", comicId).Updates(map[string]interface{}{
"download_finished": true,
"download_finished_time": time.Now(),
}).Error
}
func EpFailed(epId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownloadEp{}).Where("id = ?", epId).Update("download_failed", true).Error
}
func EpSuccess(comicId string, epId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
err := tx.Model(&ComicDownloadEp{}).Where("id = ?", epId).Updates(map[string]interface{}{
"download_finished": true,
"download_finished_time": time.Now(),
}).Error
if err != nil {
return err
}
return tx.Exec(
"UPDATE comic_downloads SET download_ep_count = download_ep_count + 1 WHERE id = ?",
comicId,
).Error
})
}
func PictureFailed(pictureId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownloadPicture{}).Where("id = ?", pictureId).Update("download_failed", true).Error
}
func PictureSuccess(
comicId string, epId string, pictureId string,
fileSize int64, format string, width int32, height int32, localPath string,
) error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
err := tx.Model(&ComicDownloadPicture{}).Where("id = ?", pictureId).Updates(map[string]interface{}{
"file_size": fileSize,
"format": format,
"width": width,
"height": height,
"local_path": localPath,
"download_finished": true,
"download_finished_time": time.Now(),
}).Error
if err != nil {
return err
}
err = tx.Exec(
"UPDATE comic_download_eps SET download_picture_count = download_picture_count + 1 WHERE id = ?",
epId,
).Error
if err != nil {
return err
}
return tx.Exec(
"UPDATE comic_downloads SET download_picture_count = download_picture_count + 1 WHERE id = ?",
comicId,
).Error
})
}
func ResetAll() error {
mutex.Lock()
defer mutex.Unlock()
return db.Transaction(func(tx *gorm.DB) error {
err := tx.Model(&ComicDownload{}).Where("1 = 1").
Update("download_failed", false).Error
if err != nil {
return err
}
err = tx.Model(&ComicDownloadEp{}).Where("1 = 1").
Update("download_failed", false).Error
if err != nil {
return err
}
err = tx.Model(&ComicDownloadPicture{}).Where("1 = 1").
Update("download_failed", false).Error
return err
})
}
func ViewLogPage(offset int, limit int) (*[]ComicView, error) {
mutex.Lock()
defer mutex.Unlock()
var list []ComicView
err := db.Offset(offset).Limit(limit).Order("last_view_time DESC").Find(&list).Error
return &list, err
}
func ClearAllViewLog() {
mutex.Lock()
defer mutex.Unlock()
db.Unscoped().Where("1 = 1").Delete(&ComicView{})
}
func DeleteViewLog(id string) {
mutex.Lock()
defer mutex.Unlock()
db.Unscoped().Where("id = ?", id).Delete(&ComicView{})
}
func DeletingComic() (*ComicDownload, error) {
mutex.Lock()
defer mutex.Unlock()
var download ComicDownload
err := db.First(&download, "deleting = 1").Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return &download, err
}
func TrueDelete(comicId string) error {
mutex.Lock()
defer mutex.Unlock()
err := db.Transaction(func(tx *gorm.DB) error {
err := tx.Unscoped().Delete(&ComicDownload{}, "id = ?", comicId).Error
if err != nil {
return err
}
err = tx.Unscoped().Delete(&ComicDownloadEp{}, "comic_id = ?", comicId).Error
if err != nil {
return err
}
err = tx.Unscoped().Delete(&ComicDownloadPicture{}, "comic_id = ?", comicId).Error
if err != nil {
return err
}
return nil
})
if err != nil {
return err
}
return db.Raw("VACUUM").Error
}
func Deleting(comicId string) error {
mutex.Lock()
defer mutex.Unlock()
return db.Model(&ComicDownload{}).Where("id = ?", comicId).Updates(map[string]interface{}{
"deleting": true,
}).Error
}
func RemoveAllRemoteImage() error {
mutex.Lock()
defer mutex.Unlock()
err := db.Unscoped().Delete(&RemoteImage{}, "1 = 1").Error
if err != nil {
return err
}
return db.Raw("VACUUM").Error
}
func EarliestRemoteImage(earliest time.Time, pageSize int) ([]RemoteImage, error) {
mutex.Lock()
defer mutex.Unlock()
var images []RemoteImage
err := db.Where("strftime('%s',updated_at) < strftime('%s',?)", earliest).
Order("updated_at").Limit(pageSize).Find(&images).Error
return images, err
}
func DeleteRemoteImages(images []RemoteImage) error {
mutex.Lock()
defer mutex.Unlock()
if len(images) == 0 {
return nil
}
ids := make([]uint, len(images))
for i := 0; i < len(images); i++ {
ids[i] = images[i].ID
}
return db.Unscoped().Model(&RemoteImage{}).Delete("id in ?", ids).Error
}
func DownloadInfo(comicId string) (*ComicDownload, error) {
mutex.Lock()
defer mutex.Unlock()
var download ComicDownload
err := db.First(&download, "id = ?", comicId).Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return nil, err
}
return &download, nil
}
func VACUUM() error {
mutex.Lock()
defer mutex.Unlock()
return db.Raw("VACUUM").Error
}