feat: more runes queries

modules/runes/database/postgresql/queries/data.sql

@@ -1,5 +1,8 @@
--- name: GetBalancesAtBlock :many
-SELECT DISTINCT ON (rune_id) * FROM runes_balances WHERE pkscript = $1 AND block_height <= $2 ORDER BY block_height DESC;
+-- name: GetBalancesByPkScript :many
+SELECT DISTINCT ON (rune_id) * FROM runes_balances WHERE pkscript = $1 AND block_height <= $2 ORDER BY rune_id, block_height DESC;
+
+-- name: GetBalancesByRuneId :many
+SELECT DISTINCT ON (pkscript) * FROM runes_balances WHERE rune_id = $1 AND block_height <= $2 ORDER BY pkscript, block_height DESC;
 
 -- name: GetOutPointBalances :many
 SELECT * FROM runes_outpoint_balances WHERE rune_id = $1 AND tx_hash = $2 AND tx_idx = $3;

modules/runes/internal/repository/postgres/gen/data.sql.go

@@ -9,17 +9,51 @@ import (
 	"context"
 )
 
-const getBalancesAtBlock = `-- name: GetBalancesAtBlock :many
-SELECT DISTINCT ON (rune_id) pkscript, block_height, rune_id, value FROM runes_balances WHERE pkscript = $1 AND block_height <= $2 ORDER BY block_height DESC
+const getBalancesByPkScript = `-- name: GetBalancesByPkScript :many
+SELECT DISTINCT ON (rune_id) pkscript, block_height, rune_id, value FROM runes_balances WHERE pkscript = $1 AND block_height <= $2 ORDER BY rune_id, block_height DESC
 `
 
-type GetBalancesAtBlockParams struct {
+type GetBalancesByPkScriptParams struct {
 	Pkscript    string
 	BlockHeight int32
 }
 
-func (q *Queries) GetBalancesAtBlock(ctx context.Context, arg GetBalancesAtBlockParams) ([]RunesBalance, error) {
-	rows, err := q.db.Query(ctx, getBalancesAtBlock, arg.Pkscript, arg.BlockHeight)
+func (q *Queries) GetBalancesByPkScript(ctx context.Context, arg GetBalancesByPkScriptParams) ([]RunesBalance, error) {
+	rows, err := q.db.Query(ctx, getBalancesByPkScript, arg.Pkscript, arg.BlockHeight)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []RunesBalance
+	for rows.Next() {
+		var i RunesBalance
+		if err := rows.Scan(
+			&i.Pkscript,
+			&i.BlockHeight,
+			&i.RuneID,
+			&i.Value,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const getBalancesByRuneId = `-- name: GetBalancesByRuneId :many
+SELECT DISTINCT ON (pkscript) pkscript, block_height, rune_id, value FROM runes_balances WHERE rune_id = $1 AND block_height <= $2 ORDER BY pkscript, block_height DESC
+`
+
+type GetBalancesByRuneIdParams struct {
+	RuneID      string
+	BlockHeight int32
+}
+
+func (q *Queries) GetBalancesByRuneId(ctx context.Context, arg GetBalancesByRuneIdParams) ([]RunesBalance, error) {
+	rows, err := q.db.Query(ctx, getBalancesByRuneId, arg.RuneID, arg.BlockHeight)
 	if err != nil {
 		return nil, err
 	}

modules/runes/internal/repository/postgres/gen/info.sql.go

@@ -0,0 +1,51 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.26.0
+// source: info.sql
+
+package gen
+
+import (
+	"context"
+)
+
+const getCurrentDBVersion = `-- name: GetCurrentDBVersion :one
+SELECT "version" FROM runes_indexer_db_version ORDER BY id DESC LIMIT 1
+`
+
+func (q *Queries) GetCurrentDBVersion(ctx context.Context) (int32, error) {
+	row := q.db.QueryRow(ctx, getCurrentDBVersion)
+	var version int32
+	err := row.Scan(&version)
+	return version, err
+}
+
+const getCurrentIndexerStats = `-- name: GetCurrentIndexerStats :one
+SELECT "client_version", "network" FROM runes_indexer_stats ORDER BY id DESC LIMIT 1
+`
+
+type GetCurrentIndexerStatsRow struct {
+	ClientVersion string
+	Network       string
+}
+
+func (q *Queries) GetCurrentIndexerStats(ctx context.Context) (GetCurrentIndexerStatsRow, error) {
+	row := q.db.QueryRow(ctx, getCurrentIndexerStats)
+	var i GetCurrentIndexerStatsRow
+	err := row.Scan(&i.ClientVersion, &i.Network)
+	return i, err
+}
+
+const updateIndexerStats = `-- name: UpdateIndexerStats :exec
+INSERT INTO runes_indexer_stats (client_version, network) VALUES ($1, $2)
+`
+
+type UpdateIndexerStatsParams struct {
+	ClientVersion string
+	Network       string
+}
+
+func (q *Queries) UpdateIndexerStats(ctx context.Context, arg UpdateIndexerStatsParams) error {
+	_, err := q.db.Exec(ctx, updateIndexerStats, arg.ClientVersion, arg.Network)
+	return err
+}