chore: fix get all referrals

pagu-project · Dec 7, 2023 · 2117aa8 · 2117aa8
1 parent 256cddb
commit 2117aa8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/discord/discord.go b/discord/discord.go
@@ -49,7 +49,7 @@ func Start(cfg *config.Config, w *wallet.Wallet, ss *SafeStore, rs *ReferralStor
 		log.Printf("error creating Discord session: %v", err)
 		return nil, err
 	}
-	bot := &Bot{cfg: cfg, discordSession: dg, faucetWallet: w, store: ss, cm: cm}
+	bot := &Bot{cfg: cfg, discordSession: dg, faucetWallet: w, store: ss, cm: cm, referralStore: rs}
 
 	// Register the messageCreate func as a callback for MessageCreate events.
 	dg.AddHandler(bot.messageHandler)

diff --git a/discord/referral.go b/discord/referral.go
@@ -86,10 +86,7 @@ func (rs *ReferralStore) GetAllReferrals() []*Referral {
 	rs.syncMap.Range(func(key, value any) bool {
 		referral, _ := value.(*Referral)
 		result = append(result, referral)
-		if value == nil {
-			return true
-		}
-		return false
+		return value == nil
 	})
 
 	return result