feat(config): preload role reactfeat/leave-gorm

1 files changed, 19 insertions, 4 deletions

diff --git a/config/xp_role.go b/config/xp_role.go
index f0d57c8..ab684c5 100644
--- a/config/xp_role.go
+++ b/config/xp_role.go
@@ -2,6 +2,7 @@ package config
 
 import (
 	"context"
+	"database/sql"
 
 	"git.anhgelus.world/anhgelus/les-copaings-bot/common"
 )
@@ -12,8 +13,12 @@ type XpRole struct {
 	GuildID uint64
 }
 
-func (xp XpRole) Save(ctx context.Context) error {
-	_, err := common.GetDB(ctx).ExecContext(
+func (xp *XpRole) Load(_ context.Context, _ *sql.DB, rows *sql.Rows) error {
+	return rows.Scan(&xp.XP, &xp.RoleID, &xp.GuildID)
+}
+
+func (xp *XpRole) Save(ctx context.Context, db *sql.DB) error {
+	_, err := db.ExecContext(
 		ctx,
 		`INSERT INTO xp_roles (xp, role, guild_id) VALUES (?, ?, ?)`,
 		xp.XP, xp.RoleID, xp.GuildID,
@@ -21,11 +26,21 @@ func (xp XpRole) Save(ctx context.Context) error {
 	return err
 }
 
-func (xp XpRole) Delete(ctx context.Context) error {
-	_, err := common.GetDB(ctx).ExecContext(
+func (xp *XpRole) Delete(ctx context.Context, db *sql.DB) error {
+	_, err := db.ExecContext(
 		ctx,
 		`DELETE FROM xp_roles WHERE xp = ? AND role = ? AND guild_id = ?`,
 		xp.XP, xp.RoleID, xp.GuildID,
 	)
 	return err
 }
+
+func getXpRoles(ctx context.Context, db *sql.DB, gID uint64) ([]*XpRole, error) {
+	return common.GetValues[*XpRole](
+		ctx,
+		db,
+		`xp_roles`,
+		`xp, role, guild_id`,
+		`guild_id = ?`, gID,
+	)
+}