49 lines
1.8 KiB
Go
49 lines
1.8 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/jackc/pgx/v5"
|
||
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
||
|
|
e "mic-wallet/server/repository/entities"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
AddBlockSql = `INSERT INTO blocks (id, hash, prev_hash, started_at, finished_at) VALUES ($1, $2, $3, $4, $5)`
|
||
|
|
SetBlockFinishedSql = `UPDATE blocks SET hash = $1 finished_at = $2 WHERE id = $3`
|
||
|
|
GetBlockSql = `SELECT * FROM blocks WHERE id = $1`
|
||
|
|
GetBlockByHashSql = `SELECT * FROM blocks WHERE hash = $1`
|
||
|
|
GetLastFinishedBlockSql = `SELECT * FROM blocks ORDER BY finished_at DESC LIMIT 1`
|
||
|
|
)
|
||
|
|
|
||
|
|
type BlocksRepository struct {
|
||
|
|
IBlocksRepository
|
||
|
|
DB *pgxpool.Pool
|
||
|
|
}
|
||
|
|
|
||
|
|
func ScanBlockRow(row pgx.Row) (block *e.Block, err error) {
|
||
|
|
block = &e.Block{}
|
||
|
|
return block, row.Scan(&block.Id, &block.Hash, &block.PrevHash, &block.StartedAt, &block.FinishedAt)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo BlocksRepository) NewBLock(ctx context.Context, opts e.NewBlockOpts) (blockID int64, err error) {
|
||
|
|
return blockID, repo.DB.QueryRow(ctx, AddBlockSql,
|
||
|
|
opts.Id, opts.Hash, opts.PrevHash, opts.StartedAt, opts.FinishedAt).Scan(&blockID)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo BlocksRepository) FinishBlock(ctx context.Context, opts e.BlockFinishedOpts) (err error) {
|
||
|
|
_, err = repo.DB.Exec(ctx, SetBlockFinishedSql, opts.Hash, opts.FinishedAt, opts.Id)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo BlocksRepository) GetBlock(ctx context.Context, id int64) (block *e.Block, err error) {
|
||
|
|
return ScanBlockRow(repo.DB.QueryRow(ctx, GetBlockSql, id))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo BlocksRepository) GetBlockByHash(ctx context.Context, hash string) (block *e.Block, err error) {
|
||
|
|
return ScanBlockRow(repo.DB.QueryRow(ctx, GetBlockByHashSql, hash))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (repo BlocksRepository) GetLastFinishedBlock(ctx context.Context) (block *e.Block, err error) {
|
||
|
|
return ScanBlockRow(repo.DB.QueryRow(ctx, GetLastFinishedBlockSql))
|
||
|
|
}
|