refactor: move scheduled_tasks service to services/ folder

services/scheduled_tasks/tasks.go

@@ -0,0 +1,80 @@
+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))
+	}
+}