254 lines
9.3 KiB
Python
254 lines
9.3 KiB
Python
|
import logging
|
||
|
|
from typing import Tuple, Union
|
||
|
|
|
||
|
|
from sqlalchemy import case, func, desc
|
||
|
|
from sqlalchemy.future import select
|
||
|
|
|
||
|
|
from lbry.db.queries.txio import (
|
||
|
|
minimum_txo_columns, row_to_txo,
|
||
|
|
where_unspent_txos, where_claims_with_changed_supports,
|
||
|
|
count_unspent_txos, where_channels_with_changed_content,
|
||
|
|
where_abandoned_claims
|
||
|
|
)
|
||
|
|
from lbry.db.query_context import ProgressContext, event_emitter
|
||
|
|
from lbry.db.tables import TX, TXO, Claim, Support
|
||
|
|
from lbry.db.utils import least
|
||
|
|
from lbry.db.constants import TXO_TYPES
|
||
|
|
from lbry.blockchain.transaction import Output
|
||
|
|
|
||
|
|
from .context import get_or_initialize_lbrycrd
|
||
|
|
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def channel_content_count_calc(signable):
|
||
|
|
return (
|
||
|
|
select(func.count(signable.c.claim_hash))
|
||
|
|
.where((signable.c.channel_hash == Claim.c.claim_hash) & signable.c.is_signature_valid)
|
||
|
|
.scalar_subquery()
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
support = TXO.alias('support')
|
||
|
|
|
||
|
|
|
||
|
|
def staked_support_aggregation(aggregate):
|
||
|
|
return (
|
||
|
|
select(aggregate).where(
|
||
|
|
(support.c.txo_type == TXO_TYPES['support']) &
|
||
|
|
(support.c.spent_height == 0)
|
||
|
|
).scalar_subquery()
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def staked_support_amount_calc(other):
|
||
|
|
return (
|
||
|
|
staked_support_aggregation(func.coalesce(func.sum(support.c.amount), 0))
|
||
|
|
.where(support.c.claim_hash == other.c.claim_hash)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def staked_support_count_calc(other):
|
||
|
|
return (
|
||
|
|
staked_support_aggregation(func.coalesce(func.count('*'), 0))
|
||
|
|
.where(support.c.claim_hash == other.c.claim_hash)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def make_label(action, blocks):
|
||
|
|
if blocks[0] == blocks[-1]:
|
||
|
|
return f"{action} {blocks[0]}"
|
||
|
|
else:
|
||
|
|
return f"{action} {blocks[0]}-{blocks[-1]}"
|
||
|
|
|
||
|
|
|
||
|
|
def select_claims_for_saving(
|
||
|
|
txo_types: Union[int, Tuple[int, ...]],
|
||
|
|
blocks: Tuple[int, int],
|
||
|
|
missing_in_claims_table=False,
|
||
|
|
missing_or_stale_in_claims_table=False,
|
||
|
|
):
|
||
|
|
select_claims = select(
|
||
|
|
*minimum_txo_columns, TXO.c.claim_hash,
|
||
|
|
staked_support_amount_calc(TXO).label('staked_support_amount'),
|
||
|
|
staked_support_count_calc(TXO).label('staked_support_count')
|
||
|
|
).where(
|
||
|
|
where_unspent_txos(
|
||
|
|
txo_types, blocks,
|
||
|
|
missing_in_claims_table=missing_in_claims_table,
|
||
|
|
missing_or_stale_in_claims_table=missing_or_stale_in_claims_table,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
if txo_types != TXO_TYPES['channel']:
|
||
|
|
channel_txo = TXO.alias('channel_txo')
|
||
|
|
channel_claim = Claim.alias('channel_claim')
|
||
|
|
return (
|
||
|
|
select_claims.add_columns(
|
||
|
|
TXO.c.signature, TXO.c.signature_digest,
|
||
|
|
case([(
|
||
|
|
TXO.c.channel_hash.isnot(None),
|
||
|
|
select(channel_txo.c.public_key).select_from(channel_txo).where(
|
||
|
|
(channel_txo.c.txo_type == TXO_TYPES['channel']) &
|
||
|
|
(channel_txo.c.claim_hash == TXO.c.channel_hash) &
|
||
|
|
(channel_txo.c.height <= TXO.c.height)
|
||
|
|
).order_by(desc(channel_txo.c.height)).limit(1).scalar_subquery()
|
||
|
|
)]).label('channel_public_key'),
|
||
|
|
channel_claim.c.short_url.label('channel_url')
|
||
|
|
).select_from(
|
||
|
|
TXO.join(TX).join(
|
||
|
|
channel_claim, channel_claim.c.claim_hash == TXO.c.channel_hash, isouter=True
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
return select_claims.select_from(TXO.join(TX))
|
||
|
|
|
||
|
|
|
||
|
|
def row_to_claim_for_saving(row) -> Tuple[Output, dict]:
|
||
|
|
txo = row_to_txo(row)
|
||
|
|
extra = {
|
||
|
|
'staked_support_amount': int(row.staked_support_amount),
|
||
|
|
'staked_support_count': int(row.staked_support_count),
|
||
|
|
}
|
||
|
|
if hasattr(row, 'signature'):
|
||
|
|
extra.update({
|
||
|
|
'signature': row.signature,
|
||
|
|
'signature_digest': row.signature_digest,
|
||
|
|
'channel_public_key': row.channel_public_key,
|
||
|
|
'channel_url': row.channel_url
|
||
|
|
})
|
||
|
|
return txo, extra
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.insert", "claims")
|
||
|
|
def claims_insert(
|
||
|
|
txo_types: Union[int, Tuple[int, ...]],
|
||
|
|
blocks: Tuple[int, int],
|
||
|
|
missing_in_claims_table: bool,
|
||
|
|
p: ProgressContext
|
||
|
|
):
|
||
|
|
chain = get_or_initialize_lbrycrd(p.ctx)
|
||
|
|
|
||
|
|
p.start(
|
||
|
|
count_unspent_txos(
|
||
|
|
txo_types, blocks,
|
||
|
|
missing_in_claims_table=missing_in_claims_table,
|
||
|
|
), progress_id=blocks[0], label=make_label("add claims at", blocks)
|
||
|
|
)
|
||
|
|
|
||
|
|
with p.ctx.engine.connect().execution_options(stream_results=True) as c:
|
||
|
|
loader = p.ctx.get_bulk_loader()
|
||
|
|
cursor = c.execute(select_claims_for_saving(
|
||
|
|
txo_types, blocks, missing_in_claims_table=missing_in_claims_table
|
||
|
|
).order_by(TXO.c.claim_hash))
|
||
|
|
for rows in cursor.partitions(900):
|
||
|
|
claim_metadata = iter(chain.db.sync_get_claim_metadata(
|
||
|
|
claim_hashes=[row['claim_hash'] for row in rows]
|
||
|
|
))
|
||
|
|
for row in rows:
|
||
|
|
metadata = next(claim_metadata, None)
|
||
|
|
if metadata is None or metadata['claim_hash'] != row.claim_hash:
|
||
|
|
log.error(
|
||
|
|
r"During sync'ing a claim in our db couldn't find a "
|
||
|
|
r"match in lbrycrd's db. This could be because lbrycrd "
|
||
|
|
r"moved a block forward and updated its own claim table "
|
||
|
|
r"while we were still on a previous block, or it could be "
|
||
|
|
r"a more fundamental issue... ¯\_(ツ)_/¯"
|
||
|
|
)
|
||
|
|
if metadata is None:
|
||
|
|
break
|
||
|
|
if metadata['claim_hash'] != row.claim_hash:
|
||
|
|
continue
|
||
|
|
txo, extra = row_to_claim_for_saving(row)
|
||
|
|
extra.update({
|
||
|
|
'short_url': metadata['short_url'],
|
||
|
|
'creation_height': metadata['creation_height'],
|
||
|
|
'activation_height': metadata['activation_height'],
|
||
|
|
'expiration_height': metadata['expiration_height'],
|
||
|
|
'takeover_height': metadata['takeover_height'],
|
||
|
|
})
|
||
|
|
loader.add_claim(txo, **extra)
|
||
|
|
if len(loader.claims) >= 25_000:
|
||
|
|
p.add(loader.flush(Claim))
|
||
|
|
p.add(loader.flush(Claim))
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.update", "claims")
|
||
|
|
def claims_update(txo_types: Union[int, Tuple[int, ...]], blocks: Tuple[int, int], p: ProgressContext):
|
||
|
|
p.start(
|
||
|
|
count_unspent_txos(txo_types, blocks, missing_or_stale_in_claims_table=True),
|
||
|
|
progress_id=blocks[0], label=make_label("update claims at", blocks)
|
||
|
|
)
|
||
|
|
with p.ctx.engine.connect().execution_options(stream_results=True) as c:
|
||
|
|
loader = p.ctx.get_bulk_loader()
|
||
|
|
cursor = c.execute(select_claims_for_saving(
|
||
|
|
txo_types, blocks, missing_or_stale_in_claims_table=True
|
||
|
|
))
|
||
|
|
for row in cursor:
|
||
|
|
txo, extra = row_to_claim_for_saving(row)
|
||
|
|
loader.update_claim(txo, **extra)
|
||
|
|
if len(loader.update_claims) >= 500:
|
||
|
|
p.add(loader.flush(Claim))
|
||
|
|
p.add(loader.flush(Claim))
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.delete", "claims")
|
||
|
|
def claims_delete(claims, p: ProgressContext):
|
||
|
|
p.start(claims, label="delete claims")
|
||
|
|
deleted = p.ctx.execute(Claim.delete().where(where_abandoned_claims()))
|
||
|
|
p.step(deleted.rowcount)
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.takeovers", "claims")
|
||
|
|
def update_takeovers(blocks: Tuple[int, int], takeovers, p: ProgressContext):
|
||
|
|
p.start(takeovers, label="winning")
|
||
|
|
chain = get_or_initialize_lbrycrd(p.ctx)
|
||
|
|
for takeover in chain.db.sync_get_takeovers(start_height=blocks[0], end_height=blocks[-1]):
|
||
|
|
update_claims = (
|
||
|
|
Claim.update()
|
||
|
|
.where(Claim.c.normalized == takeover['normalized'])
|
||
|
|
.values(
|
||
|
|
is_controlling=case(
|
||
|
|
[(Claim.c.claim_hash == takeover['claim_hash'], True)],
|
||
|
|
else_=False
|
||
|
|
),
|
||
|
|
takeover_height=case(
|
||
|
|
[(Claim.c.claim_hash == takeover['claim_hash'], takeover['height'])],
|
||
|
|
else_=None
|
||
|
|
),
|
||
|
|
activation_height=least(Claim.c.activation_height, takeover['height']),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
result = p.ctx.execute(update_claims)
|
||
|
|
p.add(result.rowcount)
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.stakes", "claims")
|
||
|
|
def update_stakes(blocks: Tuple[int, int], claims: int, p: ProgressContext):
|
||
|
|
p.start(claims)
|
||
|
|
sql = (
|
||
|
|
Claim.update()
|
||
|
|
.where(where_claims_with_changed_supports(blocks))
|
||
|
|
.values(
|
||
|
|
staked_support_amount=staked_support_amount_calc(Claim),
|
||
|
|
staked_support_count=staked_support_count_calc(Claim),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
result = p.ctx.execute(sql)
|
||
|
|
p.step(result.rowcount)
|
||
|
|
|
||
|
|
|
||
|
|
@event_emitter("blockchain.sync.claims.channels", "channels")
|
||
|
|
def update_channel_stats(blocks: Tuple[int, int], initial_sync: int, channels: int, p: ProgressContext):
|
||
|
|
p.start(channels, label="channel stats")
|
||
|
|
update_sql = Claim.update().values(
|
||
|
|
signed_claim_count=channel_content_count_calc(Claim.alias('content')),
|
||
|
|
signed_support_count=channel_content_count_calc(Support),
|
||
|
|
)
|
||
|
|
if initial_sync:
|
||
|
|
update_sql = update_sql.where(Claim.c.claim_type == TXO_TYPES['channel'])
|
||
|
|
else:
|
||
|
|
update_sql = update_sql.where(where_channels_with_changed_content(blocks))
|
||
|
|
result = p.ctx.execute(update_sql)
|
||
|
|
p.step(result.rowcount)
|