81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"runtime/debug"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
func task_main() {
|
||
|
|
defer func() {
|
||
|
|
if err := recover(); err != nil {
|
||
|
|
stack := strings.Split(string(debug.Stack()), "\n")
|
||
|
|
stack = stack[7:]
|
||
|
|
log.Println("error running tasks:", err.(error).Error())
|
||
|
|
fmt.Println(strings.Join(stack, "\n"))
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
log.Println("running per-minute scheduled tasks")
|
||
|
|
|
||
|
|
update_db_meta()
|
||
|
|
update_stats()
|
||
|
|
}
|
||
|
|
|
||
|
|
var table_stat_keys = []string{"system", "member", "group", "switch", "message"}
|
||
|
|
|
||
|
|
func plural(key string) string {
|
||
|
|
if key[len(key)-1] == 'h' {
|
||
|
|
return key + "es"
|
||
|
|
}
|
||
|
|
return key + "s"
|
||
|
|
}
|
||
|
|
|
||
|
|
func update_db_meta() {
|
||
|
|
log.Println("updating database stats")
|
||
|
|
for _, key := range table_stat_keys {
|
||
|
|
if key == "message" {
|
||
|
|
// updating message count from data db takes way too long, so we do it on a separate timer (every 10 minutes)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
q := fmt.Sprintf("update info set %s_count = (select count(*) from %s)", key, plural(key))
|
||
|
|
log.Println("data db query:", q)
|
||
|
|
run_simple_pg_query(data_db, q)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func update_db_message_meta() {
|
||
|
|
// since we're doing this concurrently, it needs a separate db connection
|
||
|
|
tmp_db := pg_connect(get_env_var("DATA_DB_URI"))
|
||
|
|
defer tmp_db.Close(context.Background())
|
||
|
|
|
||
|
|
key := "message"
|
||
|
|
q := fmt.Sprintf("update info set %s_count = (select count(*) from %s)", key, plural(key))
|
||
|
|
log.Println("data db query:", q)
|
||
|
|
run_simple_pg_query(tmp_db, q)
|
||
|
|
}
|
||
|
|
|
||
|
|
func update_stats() {
|
||
|
|
redisStats := run_redis_query()
|
||
|
|
|
||
|
|
guild_count := 0
|
||
|
|
channel_count := 0
|
||
|
|
|
||
|
|
for _, v := range redisStats {
|
||
|
|
log.Println(v.GuildCount, v.ChannelCount)
|
||
|
|
guild_count += v.GuildCount
|
||
|
|
channel_count += v.ChannelCount
|
||
|
|
}
|
||
|
|
|
||
|
|
do_stats_insert("guilds", guild_count)
|
||
|
|
do_stats_insert("channels", channel_count)
|
||
|
|
|
||
|
|
data_stats := run_data_stats_query()
|
||
|
|
for _, key := range table_stat_keys {
|
||
|
|
val := data_stats[key+"_count"].(int32)
|
||
|
|
do_stats_insert(plural(key), int(val))
|
||
|
|
}
|
||
|
|
}
|