xevion/r2park
1
0
Fork 0
mirror of https://github.com/Xevion/r2park.git synced 2025-12-13 12:12:42 -06:00
Code Issues Packages Projects Releases Wiki Activity

Improve logging, use time.Time for lastReload var

Browse Source
This commit is contained in:
Xevion
2023-12-15 06:57:46 -06:00
parent f0da57c723
commit 52bf7f9484
2 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api.go
View File

@@ -16,7 +16,7 @@ import (
var (
client *http.Client
requestCounter uint
lastReload int64
lastReload time.Time
parsePattern = regexp.MustCompile(`\s*(.+)\n\s+(.+)\n\s+(\d+)\s*`)
cachedLocations []Location
cachedLocationsMap map[uint]Location
@@ -24,6 +24,7 @@ var (
)
func init() {
lastReload = time.Now().Add(-time.Hour * 24 * 365) // Set the last reload time to a year ago
cacheExpiry = time.Now().Add(-time.Second) // Set the cache as expired initially
cookies, err := cookiejar.New(nil)
if err != nil {
@@ -75,13 +76,13 @@ func reload() {
// Attempts to reload the current session based on a given location's name.
// This uses the current request counter and last reload time to determine if a reload is necessary.
func tryReload() {
currentTime := time.Now().Unix()
lastReloadDiff := currentTime - lastReload
currentTime := time.Now()
lastReloadDiff := currentTime.Sub(lastReload)
if requestCounter >= 10 {
log.Info("Reloading session due to request count...")
} else if lastReloadDiff >= 15*60 {
log.Info("Reloading session due to time...")
log.Infof("Reloading session due to time (%s)...", lastReloadDiff)
} else {
return
}

16
main.go
View File

@@ -41,11 +41,11 @@ func Bot() {
// Login handler
session.AddHandler(func(s *discordgo.Session, r *discordgo.Ready) {
log.Printf("Logged in as: %v#%v", s.State.User.Username, s.State.User.Discriminator)
log.Infof("Logged in as: %v#%v", s.State.User.Username, s.State.User.Discriminator)
// Count serers
// Count servers
guilds := s.State.Guilds
log.Printf("Connected to %d server%s", len(guilds), Plural(len(guilds)))
log.Debugf("Connected to %d server%s", len(guilds), Plural(len(guilds)))
})
// Open the session
@@ -70,7 +70,7 @@ func Bot() {
signal.Notify(stop, os.Interrupt)
// Register commands
log.Printf("Adding %d command%s...", len(commandDefinitions), Plural(len(commandDefinitions)))
log.Debugf("Adding %d command%s...", len(commandDefinitions), Plural(len(commandDefinitions)))
registeredCommands := make([]*discordgo.ApplicationCommand, len(commandDefinitions))
for definitionIndex, commandDefinition := range commandDefinitions {
command, err := session.ApplicationCommandCreate(session.State.User.ID, os.Getenv("BOT_TARGET_GUILD"), commandDefinition)
@@ -90,7 +90,7 @@ func Bot() {
<-stop
// Remove commands
log.Infof("Removing %d command%s...\n", len(registeredCommands), Plural(len(registeredCommands)))
log.Debugf("Removing %d command%s...\n", len(registeredCommands), Plural(len(registeredCommands)))
for _, v := range registeredCommands {
log.Debugf("Removing '%v' command (%v)", v.Name, v.ID)
err := session.ApplicationCommandDelete(session.State.User.ID, os.Getenv("BOT_TARGET_GUILD"), v.ID)
@@ -132,13 +132,11 @@ func main() {
switch command {
case "scan":
log.Info("Running scan")
log.Info("Scanning...")
Scan()
case "bot":
log.Info("Running bot")
Bot()
default:
log.Info("Running bot (default)")
log.Info("Starting bot...")
Bot()
}
}