eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-16 18:50:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: avatar cleanup service

Browse source
This commit is contained in:
alyssa 2024-10-26 03:30:58 +09:00
parent 81251282b6
commit 73c444c31d
13 changed files with 256 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

30
services/scheduled_tasks/tasks.go
View file

@ -18,6 +18,11 @@ func plural(key string) string {
return key + "s"
}
func update_prom() {
count := get_image_cleanup_queue_length()
cleanupQueueLength.Set(float64(count))
}
func update_db_meta() {
for _, key := range table_stat_keys {
q := fmt.Sprintf("update info set %s_count = (select count(*) from %s)", key, plural(key))
@ -71,3 +76,28 @@ func update_discord_stats() {
panic(err)
}
}
// MUST add new image columns here
var deletedImageCleanupQuery = `
insert into image_cleanup_jobs
select id from images where
not exists (select from image_cleanup_jobs j where j.id = images.id)
and not exists (select from systems where avatar_url = images.url)
and not exists (select from systems where banner_image = images.url)
and not exists (select from system_guild where avatar_url = images.url)
and not exists (select from members where avatar_url = images.url)
and not exists (select from members where banner_image = images.url)
and not exists (select from members where webhook_avatar_url = images.url)
and not exists (select from member_guild where avatar_url = images.url)
and not exists (select from groups where icon = images.url)
and not exists (select from groups where banner_image = images.url);
`
func queue_deleted_image_cleanup() {
_, err := data_db.Exec(context.Background(), deletedImageCleanupQuery)
if err != nil {
panic(err)
}
}