tweaks
This commit is contained in:
@@ -56,31 +56,32 @@ func getDiscordSender(config Config) DiscordSender {
|
||||
return DiscordSender{Config: config}
|
||||
}
|
||||
|
||||
func Runner() {
|
||||
func getMessages(fetchers []Fetcher) []string {
|
||||
messages := []string{}
|
||||
for _, fetcher := range fetchers {
|
||||
fetcher_messages := fetcher.Fetch()
|
||||
messages = append(messages, fetcher_messages...)
|
||||
}
|
||||
return messages
|
||||
}
|
||||
|
||||
func loadEnv() {
|
||||
if err := godotenv.Load(); err != nil {
|
||||
log.Println("Warning: .env file not found, using environment variables")
|
||||
}
|
||||
}
|
||||
|
||||
func Runner() {
|
||||
loadEnv()
|
||||
config := getConfig()
|
||||
cache := getCache()
|
||||
discord_sender := getDiscordSender(config)
|
||||
fetchers := getFetchers(config, &cache)
|
||||
messages := []string{}
|
||||
for {
|
||||
log.Println("Run updater...")
|
||||
|
||||
messages = []string{} // Clear messages for the current iteration
|
||||
for _, fetcher := range fetchers {
|
||||
fetcher_messages := fetcher.Fetch()
|
||||
messages = append(messages, fetcher_messages...)
|
||||
}
|
||||
|
||||
for _, message := range messages {
|
||||
log.Println("Send message: ", message)
|
||||
discord_sender.Send(message)
|
||||
}
|
||||
|
||||
messages := getMessages(fetchers)
|
||||
discord_sender.SendBatch(messages)
|
||||
cache.Save()
|
||||
|
||||
time.Sleep(config.Interval)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -17,3 +17,9 @@ func (d DiscordSender) Send(msg string) {
|
||||
}
|
||||
http.Post(d.Config.DiscordWebhook, "application/json", bytes.NewBuffer(b))
|
||||
}
|
||||
|
||||
func (d DiscordSender) SendBatch(msgs []string) {
|
||||
for _, msg := range msgs {
|
||||
d.Send(msg)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user