mirror of
https://github.com/aljazceru/ark.git
synced 2025-12-18 12:44:19 +01:00
3985bd4e14
* Fixes * Fixes to domain layer: * Add Leaf bool field to know to fix the returned list of leaves * Add non-persisted UnsignedForfeitTxs to RoundFinalizationStarted * Store only error msg when round fails instead of full error * Fix wallet interface: * Add Close() to close conn with wallet * Add GetAsset() to fix missing asset err when calling Transfer() * Fix gocron scheduler to correctly run/build the project * Fix badger repo implementation: * Fix datadirs of projection stores * Return error if current round not found * Fix round event deserialization * Fix TxBuilder interface & dummy impl: * Pass asp pubkey as arg of the defined functions * Fix connectorsToInputArgs to return the right number of ins * Fix getTxid() to return the id of an hex encoded tx too * Fix createConnectors() to return a tx if there's only 1 connector * Add leaf bool field to psetWithLevel in case a leaf is not in the last level * Fix node's isLeaf() check * Move to hex encoded pubkeys instead of ark encoded * Fix app layer: * Add Start() and Stop() to the interface & Expect raw pubkeys instead of strings as args * Source & cache pubkey from wallet at startup * Drop usage of scheduler and schedule next task based on occurred round events * Increase verbosity * Use hex instead of ark encoding to store receveirs' pubkeys * Lower faucet amount from 100k to 10k sats in total * Fix finalizeRound() to persist round events even if it failed * Add view() to forfeitTxMap to enrich RoundFinalizationEvent with unsigned forfeit txs * Add app config * Fix interface layer: * Remove repo manager from handler factory * Fix GetEventStream to forward events to stream once they arrive from app layer * Return missing unsigned forfeit txs in RoundFinalizationEvent * Fix extracting user pubkey from address * Add log interceptors * Add config struct * Add factory * Clean interface * Add config and launcher * Tidy deps & Set defaut round interval to 30secs for dev mode
129 lines
2.8 KiB
Go
129 lines
2.8 KiB
Go
package badgerdb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/ark-network/ark/internal/core/domain"
|
|
dbtypes "github.com/ark-network/ark/internal/infrastructure/db/types"
|
|
"github.com/dgraph-io/badger/v4"
|
|
"github.com/timshannon/badgerhold/v4"
|
|
)
|
|
|
|
const roundStoreDir = "rounds"
|
|
|
|
type roundRepository struct {
|
|
store *badgerhold.Store
|
|
}
|
|
|
|
func NewRoundRepository(config ...interface{}) (dbtypes.RoundStore, error) {
|
|
if len(config) != 2 {
|
|
return nil, fmt.Errorf("invalid config")
|
|
}
|
|
baseDir, ok := config[0].(string)
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid base directory")
|
|
}
|
|
var logger badger.Logger
|
|
if config[1] != nil {
|
|
logger, ok = config[1].(badger.Logger)
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid logger")
|
|
}
|
|
}
|
|
|
|
var dir string
|
|
if len(baseDir) > 0 {
|
|
dir = filepath.Join(baseDir, roundStoreDir)
|
|
}
|
|
store, err := createDB(dir, logger)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open round events store: %s", err)
|
|
}
|
|
|
|
return &roundRepository{store}, nil
|
|
}
|
|
|
|
func (r *roundRepository) AddOrUpdateRound(
|
|
ctx context.Context, round domain.Round,
|
|
) error {
|
|
return r.addOrUpdateRound(ctx, round)
|
|
}
|
|
|
|
func (r *roundRepository) GetCurrentRound(
|
|
ctx context.Context,
|
|
) (*domain.Round, error) {
|
|
query := badgerhold.Where("Stage.Ended").Eq(false).And("Stage.Failed").Eq(false)
|
|
rounds, err := r.findRound(ctx, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(rounds) <= 0 {
|
|
return nil, fmt.Errorf("ongoing round not found")
|
|
}
|
|
return &rounds[0], nil
|
|
}
|
|
|
|
func (r *roundRepository) GetRoundWithId(
|
|
ctx context.Context, id string,
|
|
) (*domain.Round, error) {
|
|
query := badgerhold.Where("Id").Eq(id)
|
|
rounds, err := r.findRound(ctx, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(rounds) <= 0 {
|
|
return nil, fmt.Errorf("round with id %s not found", id)
|
|
}
|
|
round := &rounds[0]
|
|
return round, nil
|
|
}
|
|
|
|
func (r *roundRepository) GetRoundWithTxid(
|
|
ctx context.Context, txid string,
|
|
) (*domain.Round, error) {
|
|
query := badgerhold.Where("Txid").Eq(txid)
|
|
rounds, err := r.findRound(ctx, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(rounds) <= 0 {
|
|
return nil, fmt.Errorf("round with txid %s not found", txid)
|
|
}
|
|
round := &rounds[0]
|
|
return round, nil
|
|
}
|
|
|
|
func (r *roundRepository) Close() {
|
|
r.store.Close()
|
|
}
|
|
|
|
func (r *roundRepository) findRound(
|
|
ctx context.Context, query *badgerhold.Query,
|
|
) ([]domain.Round, error) {
|
|
var rounds []domain.Round
|
|
var err error
|
|
|
|
if ctx.Value("tx") != nil {
|
|
tx := ctx.Value("tx").(*badger.Txn)
|
|
err = r.store.TxFind(tx, &rounds, query)
|
|
} else {
|
|
err = r.store.Find(&rounds, query)
|
|
}
|
|
|
|
return rounds, err
|
|
}
|
|
|
|
func (r *roundRepository) addOrUpdateRound(
|
|
ctx context.Context, round domain.Round,
|
|
) (err error) {
|
|
if ctx.Value("tx") != nil {
|
|
tx := ctx.Value("tx").(*badger.Txn)
|
|
err = r.store.TxUpsert(tx, round.Id, round)
|
|
} else {
|
|
err = r.store.Upsert(round.Id, round)
|
|
}
|
|
return
|
|
}
|