forked from LBRYCommunity/lbry-sdk
963 lines
40 KiB
Python
963 lines
40 KiB
Python
import sqlite3
|
|
import struct
|
|
from typing import Union, Tuple, Set, List
|
|
from binascii import unhexlify
|
|
from itertools import chain
|
|
|
|
|
|
from torba.server.db import DB
|
|
from torba.server.util import class_logger
|
|
from torba.client.basedatabase import query, constraints_to_sql
|
|
|
|
from lbrynet.schema.url import URL, normalize_name
|
|
from lbrynet.schema.mime_types import guess_stream_type
|
|
from lbrynet.wallet.ledger import MainNetLedger, RegTestLedger
|
|
from lbrynet.wallet.transaction import Transaction, Output
|
|
from lbrynet.wallet.server.canonical import register_canonical_functions
|
|
from lbrynet.wallet.server.trending import (
|
|
CREATE_TREND_TABLE, calculate_trending, register_trending_functions
|
|
)
|
|
|
|
|
|
ATTRIBUTE_ARRAY_MAX_LENGTH = 100
|
|
CLAIM_TYPES = {
|
|
'stream': 1,
|
|
'channel': 2,
|
|
}
|
|
STREAM_TYPES = {
|
|
'video': 1,
|
|
'audio': 2,
|
|
'image': 3,
|
|
'document': 4,
|
|
'binary': 5,
|
|
'model': 6
|
|
}
|
|
|
|
|
|
def _apply_constraints_for_array_attributes(constraints, attr):
|
|
any_items = constraints.pop(f'any_{attr}s', [])[:ATTRIBUTE_ARRAY_MAX_LENGTH]
|
|
if any_items:
|
|
constraints.update({
|
|
f'$any_{attr}{i}': item for i, item in enumerate(any_items)
|
|
})
|
|
values = ', '.join(
|
|
f':$any_{attr}{i}' for i in range(len(any_items))
|
|
)
|
|
constraints[f'claim.claim_hash__in#_any_{attr}'] = f"""
|
|
SELECT DISTINCT claim_hash FROM {attr} WHERE {attr} IN ({values})
|
|
"""
|
|
|
|
all_items = constraints.pop(f'all_{attr}s', [])[:ATTRIBUTE_ARRAY_MAX_LENGTH]
|
|
if all_items:
|
|
constraints[f'$all_{attr}_count'] = len(all_items)
|
|
constraints.update({
|
|
f'$all_{attr}{i}': item for i, item in enumerate(all_items)
|
|
})
|
|
values = ', '.join(
|
|
f':$all_{attr}{i}' for i in range(len(all_items))
|
|
)
|
|
constraints[f'claim.claim_hash__in#_all_{attr}'] = f"""
|
|
SELECT claim_hash FROM {attr} WHERE {attr} IN ({values})
|
|
GROUP BY claim_hash HAVING COUNT({attr}) = :$all_{attr}_count
|
|
"""
|
|
|
|
not_items = constraints.pop(f'not_{attr}s', [])[:ATTRIBUTE_ARRAY_MAX_LENGTH]
|
|
if not_items:
|
|
constraints.update({
|
|
f'$not_{attr}{i}': item for i, item in enumerate(not_items)
|
|
})
|
|
values = ', '.join(
|
|
f':$not_{attr}{i}' for i in range(len(not_items))
|
|
)
|
|
constraints[f'claim.claim_hash__not_in#_not_{attr}'] = f"""
|
|
SELECT DISTINCT claim_hash FROM {attr} WHERE {attr} IN ({values})
|
|
"""
|
|
|
|
|
|
class SQLDB:
|
|
|
|
PRAGMAS = """
|
|
pragma journal_mode=WAL;
|
|
"""
|
|
|
|
CREATE_CLAIM_TABLE = """
|
|
create table if not exists claim (
|
|
claim_hash bytes primary key,
|
|
claim_id text not null,
|
|
claim_name text not null,
|
|
normalized text not null,
|
|
txo_hash bytes not null,
|
|
tx_position integer not null,
|
|
amount integer not null,
|
|
timestamp integer not null, -- last updated timestamp
|
|
creation_timestamp integer not null,
|
|
height integer not null, -- last updated height
|
|
creation_height integer not null,
|
|
activation_height integer,
|
|
expiration_height integer not null,
|
|
release_time integer not null,
|
|
|
|
short_url text not null, -- normalized#shortest-unique-claim_id
|
|
canonical_url text, -- channel's-short_url/normalized#shortest-unique-claim_id-within-channel
|
|
|
|
claim_type integer,
|
|
|
|
-- streams
|
|
stream_type text,
|
|
media_type text,
|
|
|
|
-- claims which are channels
|
|
public_key_bytes bytes,
|
|
public_key_hash bytes,
|
|
claims_in_channel integer,
|
|
|
|
-- claims which are inside channels
|
|
channel_hash bytes,
|
|
channel_join integer, -- height at which claim got valid signature / joined channel
|
|
signature bytes,
|
|
signature_digest bytes,
|
|
signature_valid bool,
|
|
|
|
effective_amount integer not null default 0,
|
|
support_amount integer not null default 0,
|
|
trending_group integer not null default 0,
|
|
trending_mixed integer not null default 0,
|
|
trending_local integer not null default 0,
|
|
trending_global integer not null default 0
|
|
);
|
|
|
|
create index if not exists claim_resolve_idx on claim (normalized, claim_id);
|
|
create index if not exists claim_claims_in_channel_idx on claim (signature_valid, channel_hash, normalized);
|
|
|
|
create index if not exists claim_id_idx on claim (claim_id);
|
|
create index if not exists claim_normalized_idx on claim (normalized);
|
|
create index if not exists claim_txo_hash_idx on claim (txo_hash);
|
|
create index if not exists claim_channel_hash_idx on claim (channel_hash);
|
|
create index if not exists claim_release_time_idx on claim (release_time);
|
|
create index if not exists claim_timestamp_idx on claim (timestamp);
|
|
create index if not exists claim_height_idx on claim (height);
|
|
create index if not exists claim_activation_height_idx on claim (activation_height);
|
|
create index if not exists claim_expiration_height_idx on claim (expiration_height);
|
|
create index if not exists claim_public_key_hash_idx on claim (public_key_hash);
|
|
|
|
create index if not exists claim_claim_type_idx on claim (claim_type);
|
|
create index if not exists claim_stream_type_idx on claim (stream_type);
|
|
create index if not exists claim_media_type_idx on claim (media_type);
|
|
|
|
create index if not exists claim_signature_valid_idx on claim (signature_valid);
|
|
|
|
create index if not exists claim_effective_amount_idx on claim (effective_amount);
|
|
create index if not exists claim_trending_group_idx on claim (trending_group);
|
|
create index if not exists claim_trending_mixed_idx on claim (trending_mixed);
|
|
create index if not exists claim_trending_local_idx on claim (trending_local);
|
|
create index if not exists claim_trending_global_idx on claim (trending_global);
|
|
"""
|
|
|
|
CREATE_SUPPORT_TABLE = """
|
|
create table if not exists support (
|
|
txo_hash bytes primary key,
|
|
tx_position integer not null,
|
|
height integer not null,
|
|
claim_hash bytes not null,
|
|
amount integer not null
|
|
);
|
|
create index if not exists support_txo_hash_idx on support (txo_hash);
|
|
create index if not exists support_claim_hash_idx on support (claim_hash, height);
|
|
"""
|
|
|
|
CREATE_TAG_TABLE = """
|
|
create table if not exists tag (
|
|
tag text not null,
|
|
claim_hash bytes not null,
|
|
height integer not null
|
|
);
|
|
create index if not exists tag_tag_idx on tag (tag);
|
|
create index if not exists tag_claim_hash_idx on tag (claim_hash);
|
|
create index if not exists tag_height_idx on tag (height);
|
|
"""
|
|
|
|
CREATE_CLAIMTRIE_TABLE = """
|
|
create table if not exists claimtrie (
|
|
normalized text primary key,
|
|
claim_hash bytes not null,
|
|
last_take_over_height integer not null
|
|
);
|
|
create index if not exists claimtrie_claim_hash_idx on claimtrie (claim_hash);
|
|
"""
|
|
|
|
CREATE_TABLES_QUERY = (
|
|
PRAGMAS +
|
|
CREATE_CLAIM_TABLE +
|
|
CREATE_TREND_TABLE +
|
|
CREATE_SUPPORT_TABLE +
|
|
CREATE_CLAIMTRIE_TABLE +
|
|
CREATE_TAG_TABLE
|
|
)
|
|
|
|
def __init__(self, main, path):
|
|
self.main = main
|
|
self._db_path = path
|
|
self.db = None
|
|
self.logger = class_logger(__name__, self.__class__.__name__)
|
|
self.ledger = MainNetLedger if self.main.coin.NET == 'mainnet' else RegTestLedger
|
|
|
|
def open(self):
|
|
self.db = sqlite3.connect(self._db_path, isolation_level=None, check_same_thread=False)
|
|
self.db.row_factory = sqlite3.Row
|
|
self.db.executescript(self.CREATE_TABLES_QUERY)
|
|
register_canonical_functions(self.db)
|
|
register_trending_functions(self.db)
|
|
|
|
def close(self):
|
|
self.db.close()
|
|
|
|
@staticmethod
|
|
def _insert_sql(table: str, data: dict) -> Tuple[str, list]:
|
|
columns, values = [], []
|
|
for column, value in data.items():
|
|
columns.append(column)
|
|
values.append(value)
|
|
sql = (
|
|
f"INSERT INTO {table} ({', '.join(columns)}) "
|
|
f"VALUES ({', '.join(['?'] * len(values))})"
|
|
)
|
|
return sql, values
|
|
|
|
@staticmethod
|
|
def _update_sql(table: str, data: dict, where: str,
|
|
constraints: Union[list, tuple]) -> Tuple[str, list]:
|
|
columns, values = [], []
|
|
for column, value in data.items():
|
|
columns.append("{} = ?".format(column))
|
|
values.append(value)
|
|
values.extend(constraints)
|
|
return f"UPDATE {table} SET {', '.join(columns)} WHERE {where}", values
|
|
|
|
@staticmethod
|
|
def _delete_sql(table: str, constraints: dict) -> Tuple[str, dict]:
|
|
where, values = constraints_to_sql(constraints)
|
|
return f"DELETE FROM {table} WHERE {where}", values
|
|
|
|
def execute(self, *args):
|
|
return self.db.execute(*args)
|
|
|
|
def begin(self):
|
|
self.execute('begin;')
|
|
|
|
def commit(self):
|
|
self.execute('commit;')
|
|
|
|
def _upsertable_claims(self, txos: List[Output], header, clear_first=False):
|
|
claim_hashes, claims, tags = [], [], []
|
|
for txo in txos:
|
|
tx = txo.tx_ref.tx
|
|
|
|
try:
|
|
assert txo.claim_name
|
|
assert txo.normalized_name
|
|
except:
|
|
#self.logger.exception(f"Could not decode claim name for {tx.id}:{txo.position}.")
|
|
continue
|
|
|
|
claim_hash = sqlite3.Binary(txo.claim_hash)
|
|
claim_hashes.append(claim_hash)
|
|
claim_record = {
|
|
'claim_hash': claim_hash,
|
|
'claim_id': txo.claim_id,
|
|
'claim_name': txo.claim_name,
|
|
'normalized': txo.normalized_name,
|
|
'txo_hash': sqlite3.Binary(txo.ref.hash),
|
|
'tx_position': tx.position,
|
|
'amount': txo.amount,
|
|
'timestamp': header['timestamp'],
|
|
'height': tx.height,
|
|
'claim_type': None,
|
|
'stream_type': None,
|
|
'media_type': None,
|
|
'release_time': None,
|
|
}
|
|
claims.append(claim_record)
|
|
|
|
try:
|
|
claim = txo.claim
|
|
except:
|
|
#self.logger.exception(f"Could not parse claim protobuf for {tx.id}:{txo.position}.")
|
|
continue
|
|
|
|
if claim.is_stream:
|
|
claim_record['claim_type'] = CLAIM_TYPES['stream']
|
|
claim_record['media_type'] = claim.stream.source.media_type
|
|
claim_record['stream_type'] = STREAM_TYPES[guess_stream_type(claim_record['media_type'])]
|
|
if claim.stream.release_time:
|
|
claim_record['release_time'] = claim.stream.release_time
|
|
elif claim.is_channel:
|
|
claim_record['claim_type'] = CLAIM_TYPES['channel']
|
|
|
|
for tag in claim.message.tags:
|
|
tags.append((tag, claim_hash, tx.height))
|
|
|
|
if clear_first:
|
|
self._clear_claim_metadata(claim_hashes)
|
|
|
|
if tags:
|
|
self.db.executemany(
|
|
"INSERT INTO tag (tag, claim_hash, height) VALUES (?, ?, ?)", tags
|
|
)
|
|
|
|
return claims
|
|
|
|
def insert_claims(self, txos: List[Output], header):
|
|
claims = self._upsertable_claims(txos, header)
|
|
if claims:
|
|
self.db.executemany("""
|
|
INSERT OR IGNORE INTO claim (
|
|
claim_hash, claim_id, claim_name, normalized, txo_hash, tx_position, amount,
|
|
claim_type, media_type, stream_type, timestamp, creation_timestamp, height,
|
|
creation_height, release_time, activation_height, expiration_height, short_url)
|
|
VALUES (
|
|
:claim_hash, :claim_id, :claim_name, :normalized, :txo_hash, :tx_position, :amount,
|
|
:claim_type, :media_type, :stream_type, :timestamp, :timestamp, :height, :height,
|
|
CASE WHEN :release_time IS NOT NULL THEN :release_time ELSE :timestamp END,
|
|
CASE WHEN :normalized NOT IN (SELECT normalized FROM claimtrie) THEN :height END,
|
|
CASE WHEN :height >= 137181 THEN :height+2102400 ELSE :height+262974 END,
|
|
:claim_name||COALESCE(
|
|
(SELECT shortest_id(claim_id, :claim_id) FROM claim WHERE normalized = :normalized),
|
|
'#'||substr(:claim_id, 1, 1)
|
|
)
|
|
)""", claims)
|
|
|
|
def update_claims(self, txos: List[Output], header):
|
|
claims = self._upsertable_claims(txos, header, clear_first=True)
|
|
if claims:
|
|
self.db.executemany("""
|
|
UPDATE claim SET
|
|
txo_hash=:txo_hash, tx_position=:tx_position, amount=:amount, height=:height,
|
|
claim_type=:claim_type, media_type=:media_type, stream_type=:stream_type,
|
|
timestamp=:timestamp,
|
|
release_time=CASE WHEN :release_time IS NOT NULL THEN :release_time ELSE release_time END
|
|
WHERE claim_hash=:claim_hash;
|
|
""", claims)
|
|
|
|
def delete_claims(self, claim_hashes: Set[bytes]):
|
|
""" Deletes claim supports and from claimtrie in case of an abandon. """
|
|
if claim_hashes:
|
|
binary_claim_hashes = [sqlite3.Binary(claim_hash) for claim_hash in claim_hashes]
|
|
for table in ('claim', 'support', 'claimtrie'):
|
|
self.execute(*self._delete_sql(table, {'claim_hash__in': binary_claim_hashes}))
|
|
self._clear_claim_metadata(binary_claim_hashes)
|
|
|
|
def _clear_claim_metadata(self, binary_claim_hashes: List[sqlite3.Binary]):
|
|
if binary_claim_hashes:
|
|
for table in ('tag',): # 'language', 'location', etc
|
|
self.execute(*self._delete_sql(table, {'claim_hash__in': binary_claim_hashes}))
|
|
|
|
def split_inputs_into_claims_supports_and_other(self, txis):
|
|
txo_hashes = {txi.txo_ref.hash for txi in txis}
|
|
claims = self.execute(*query(
|
|
"SELECT txo_hash, claim_hash, normalized FROM claim",
|
|
txo_hash__in=[sqlite3.Binary(txo_hash) for txo_hash in txo_hashes]
|
|
)).fetchall()
|
|
txo_hashes -= {r['txo_hash'] for r in claims}
|
|
supports = {}
|
|
if txo_hashes:
|
|
supports = self.execute(*query(
|
|
"SELECT txo_hash, claim_hash FROM support",
|
|
txo_hash__in=[sqlite3.Binary(txo_hash) for txo_hash in txo_hashes]
|
|
)).fetchall()
|
|
txo_hashes -= {r['txo_hash'] for r in supports}
|
|
return claims, supports, txo_hashes
|
|
|
|
def insert_supports(self, txos: List[Output]):
|
|
supports = []
|
|
for txo in txos:
|
|
tx = txo.tx_ref.tx
|
|
supports.append((
|
|
sqlite3.Binary(txo.ref.hash), tx.position, tx.height,
|
|
sqlite3.Binary(txo.claim_hash), txo.amount
|
|
))
|
|
if supports:
|
|
self.db.executemany(
|
|
"INSERT INTO support ("
|
|
" txo_hash, tx_position, height, claim_hash, amount"
|
|
") "
|
|
"VALUES (?, ?, ?, ?, ?)", supports
|
|
)
|
|
|
|
def delete_supports(self, txo_hashes: Set[bytes]):
|
|
if txo_hashes:
|
|
self.execute(*self._delete_sql(
|
|
'support', {'txo_hash__in': [sqlite3.Binary(txo_hash) for txo_hash in txo_hashes]}
|
|
))
|
|
|
|
def validate_channel_signatures(self, height, new_claims, updated_claims, spent_claims, timer):
|
|
if not new_claims and not updated_claims and not spent_claims:
|
|
return
|
|
|
|
sub_timer = timer.add_timer('segregate channels and signables')
|
|
sub_timer.start()
|
|
channels, new_channel_keys, signables = {}, {}, {}
|
|
for txo in chain(new_claims, updated_claims):
|
|
try:
|
|
claim = txo.claim
|
|
except:
|
|
continue
|
|
if claim.is_channel:
|
|
channels[txo.claim_hash] = txo
|
|
new_channel_keys[txo.claim_hash] = claim.channel.public_key_bytes
|
|
else:
|
|
signables[txo.claim_hash] = txo
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('make list of channels we need to lookup')
|
|
sub_timer.start()
|
|
missing_channel_keys = set()
|
|
for txo in signables.values():
|
|
claim = txo.claim
|
|
if claim.is_signed and claim.signing_channel_hash not in new_channel_keys:
|
|
missing_channel_keys.add(claim.signing_channel_hash)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('lookup missing channels')
|
|
sub_timer.start()
|
|
all_channel_keys = {}
|
|
if new_channel_keys or missing_channel_keys:
|
|
all_channel_keys = dict(self.execute(*query(
|
|
"SELECT claim_hash, public_key_bytes FROM claim",
|
|
claim_hash__in=[
|
|
sqlite3.Binary(channel_hash) for channel_hash in
|
|
set(new_channel_keys) | missing_channel_keys
|
|
]
|
|
)))
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('prepare for updating claims')
|
|
sub_timer.start()
|
|
changed_channel_keys = {}
|
|
for claim_hash, new_key in new_channel_keys.items():
|
|
if claim_hash not in all_channel_keys or all_channel_keys[claim_hash] != new_key:
|
|
all_channel_keys[claim_hash] = new_key
|
|
changed_channel_keys[claim_hash] = new_key
|
|
|
|
claim_updates = []
|
|
|
|
for claim_hash, txo in signables.items():
|
|
claim = txo.claim
|
|
update = {
|
|
'claim_hash': sqlite3.Binary(claim_hash),
|
|
'channel_hash': None,
|
|
'signature': None,
|
|
'signature_digest': None,
|
|
'signature_valid': None
|
|
}
|
|
if claim.is_signed:
|
|
update.update({
|
|
'channel_hash': sqlite3.Binary(claim.signing_channel_hash),
|
|
'signature': sqlite3.Binary(txo.get_encoded_signature()),
|
|
'signature_digest': sqlite3.Binary(txo.get_signature_digest(self.ledger)),
|
|
'signature_valid': 0
|
|
})
|
|
claim_updates.append(update)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('find claims affected by a change in channel key')
|
|
sub_timer.start()
|
|
if changed_channel_keys:
|
|
sql = f"""
|
|
SELECT * FROM claim WHERE
|
|
channel_hash IN ({','.join('?' for _ in changed_channel_keys)}) AND
|
|
signature IS NOT NULL
|
|
"""
|
|
for affected_claim in self.execute(sql, [sqlite3.Binary(h) for h in changed_channel_keys]):
|
|
if affected_claim['claim_hash'] not in signables:
|
|
claim_updates.append({
|
|
'claim_hash': sqlite3.Binary(affected_claim['claim_hash']),
|
|
'channel_hash': sqlite3.Binary(affected_claim['channel_hash']),
|
|
'signature': sqlite3.Binary(affected_claim['signature']),
|
|
'signature_digest': sqlite3.Binary(affected_claim['signature_digest']),
|
|
'signature_valid': 0
|
|
})
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('verify signatures')
|
|
sub_timer.start()
|
|
for update in claim_updates:
|
|
channel_pub_key = all_channel_keys.get(update['channel_hash'])
|
|
if channel_pub_key and update['signature']:
|
|
update['signature_valid'] = Output.is_signature_valid(
|
|
bytes(update['signature']), bytes(update['signature_digest']), channel_pub_key
|
|
)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('update claims')
|
|
sub_timer.start()
|
|
if claim_updates:
|
|
self.db.executemany(f"""
|
|
UPDATE claim SET
|
|
channel_hash=:channel_hash, signature=:signature, signature_digest=:signature_digest,
|
|
signature_valid=:signature_valid,
|
|
channel_join=CASE
|
|
WHEN signature_valid=1 AND :signature_valid=1 THEN channel_join
|
|
WHEN :signature_valid=1 THEN {height}
|
|
END,
|
|
canonical_url=CASE
|
|
WHEN signature_valid=1 AND :signature_valid=1 THEN canonical_url
|
|
WHEN :signature_valid=1 THEN
|
|
(SELECT short_url FROM claim WHERE claim_hash=:channel_hash)||'/'||
|
|
claim_name||COALESCE(
|
|
(SELECT shortest_id(other_claim.claim_id, claim.claim_id) FROM claim AS other_claim
|
|
WHERE other_claim.signature_valid = 1 AND
|
|
other_claim.channel_hash = :channel_hash AND
|
|
other_claim.normalized = claim.normalized),
|
|
'#'||substr(claim_id, 1, 1)
|
|
)
|
|
END
|
|
WHERE claim_hash=:claim_hash;
|
|
""", claim_updates)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('update claims affected by spent channels')
|
|
sub_timer.start()
|
|
if spent_claims:
|
|
self.execute(
|
|
f"""
|
|
UPDATE claim SET
|
|
signature_valid=CASE WHEN signature IS NOT NULL THEN 0 END,
|
|
channel_join=NULL, canonical_url=NULL
|
|
WHERE channel_hash IN ({','.join('?' for _ in spent_claims)})
|
|
""", [sqlite3.Binary(cid) for cid in spent_claims]
|
|
)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('update channels')
|
|
sub_timer.start()
|
|
if channels:
|
|
self.db.executemany(
|
|
"""
|
|
UPDATE claim SET
|
|
public_key_bytes=:public_key_bytes,
|
|
public_key_hash=:public_key_hash
|
|
WHERE claim_hash=:claim_hash""", [{
|
|
'claim_hash': sqlite3.Binary(claim_hash),
|
|
'public_key_bytes': sqlite3.Binary(txo.claim.channel.public_key_bytes),
|
|
'public_key_hash': sqlite3.Binary(
|
|
self.ledger.address_to_hash160(
|
|
self.ledger.public_key_to_address(txo.claim.channel.public_key_bytes)))
|
|
} for claim_hash, txo in channels.items()]
|
|
)
|
|
sub_timer.stop()
|
|
|
|
sub_timer = timer.add_timer('update claims_in_channel counts')
|
|
sub_timer.start()
|
|
if all_channel_keys:
|
|
self.db.executemany(f"""
|
|
UPDATE claim SET
|
|
claims_in_channel=(
|
|
SELECT COUNT(*) FROM claim AS claim_in_channel
|
|
WHERE claim_in_channel.signature_valid=1 AND
|
|
claim_in_channel.channel_hash=claim.claim_hash
|
|
)
|
|
WHERE claim_hash = ?
|
|
""", [(sqlite3.Binary(channel_hash),) for channel_hash in all_channel_keys.keys()])
|
|
sub_timer.stop()
|
|
|
|
def _update_support_amount(self, claim_hashes):
|
|
if claim_hashes:
|
|
self.execute(f"""
|
|
UPDATE claim SET
|
|
support_amount = COALESCE(
|
|
(SELECT SUM(amount) FROM support WHERE support.claim_hash=claim.claim_hash), 0
|
|
)
|
|
WHERE claim_hash IN ({','.join('?' for _ in claim_hashes)})
|
|
""", claim_hashes)
|
|
|
|
def _update_effective_amount(self, height, claim_hashes=None):
|
|
self.execute(
|
|
f"UPDATE claim SET effective_amount = amount + support_amount "
|
|
f"WHERE activation_height = {height}"
|
|
)
|
|
if claim_hashes:
|
|
self.execute(
|
|
f"UPDATE claim SET effective_amount = amount + support_amount "
|
|
f"WHERE activation_height < {height} "
|
|
f" AND claim_hash IN ({','.join('?' for _ in claim_hashes)})",
|
|
claim_hashes
|
|
)
|
|
|
|
def _calculate_activation_height(self, height):
|
|
last_take_over_height = f"""COALESCE(
|
|
(SELECT last_take_over_height FROM claimtrie
|
|
WHERE claimtrie.normalized=claim.normalized),
|
|
{height}
|
|
)
|
|
"""
|
|
self.execute(f"""
|
|
UPDATE claim SET activation_height =
|
|
{height} + min(4032, cast(({height} - {last_take_over_height}) / 32 AS INT))
|
|
WHERE activation_height IS NULL
|
|
""")
|
|
|
|
def _perform_overtake(self, height, changed_claim_hashes, deleted_names):
|
|
deleted_names_sql = claim_hashes_sql = ""
|
|
if changed_claim_hashes:
|
|
claim_hashes_sql = f"OR claim_hash IN ({','.join('?' for _ in changed_claim_hashes)})"
|
|
if deleted_names:
|
|
deleted_names_sql = f"OR normalized IN ({','.join('?' for _ in deleted_names)})"
|
|
overtakes = self.execute(f"""
|
|
SELECT winner.normalized, winner.claim_hash,
|
|
claimtrie.claim_hash AS current_winner,
|
|
MAX(winner.effective_amount)
|
|
FROM (
|
|
SELECT normalized, claim_hash, effective_amount FROM claim
|
|
WHERE normalized IN (
|
|
SELECT normalized FROM claim WHERE activation_height={height} {claim_hashes_sql}
|
|
) {deleted_names_sql}
|
|
ORDER BY effective_amount DESC, height ASC, tx_position ASC
|
|
) AS winner LEFT JOIN claimtrie USING (normalized)
|
|
GROUP BY winner.normalized
|
|
HAVING current_winner IS NULL OR current_winner <> winner.claim_hash
|
|
""", changed_claim_hashes+deleted_names)
|
|
for overtake in overtakes:
|
|
if overtake['current_winner']:
|
|
self.execute(
|
|
f"UPDATE claimtrie SET claim_hash = ?, last_take_over_height = {height} "
|
|
f"WHERE normalized = ?",
|
|
(sqlite3.Binary(overtake['claim_hash']), overtake['normalized'])
|
|
)
|
|
else:
|
|
self.execute(
|
|
f"INSERT INTO claimtrie (claim_hash, normalized, last_take_over_height) "
|
|
f"VALUES (?, ?, {height})",
|
|
(sqlite3.Binary(overtake['claim_hash']), overtake['normalized'])
|
|
)
|
|
self.execute(
|
|
f"UPDATE claim SET activation_height = {height} WHERE normalized = ? "
|
|
f"AND (activation_height IS NULL OR activation_height > {height})",
|
|
(overtake['normalized'],)
|
|
)
|
|
|
|
def _copy(self, height):
|
|
if height > 50:
|
|
self.execute(f"DROP TABLE claimtrie{height-50}")
|
|
self.execute(f"CREATE TABLE claimtrie{height} AS SELECT * FROM claimtrie")
|
|
|
|
def update_claimtrie(self, height, changed_claim_hashes, deleted_names, timer):
|
|
r = timer.run
|
|
binary_claim_hashes = [
|
|
sqlite3.Binary(claim_hash) for claim_hash in changed_claim_hashes
|
|
]
|
|
|
|
r(self._calculate_activation_height, height)
|
|
r(self._update_support_amount, binary_claim_hashes)
|
|
|
|
r(self._update_effective_amount, height, binary_claim_hashes)
|
|
r(self._perform_overtake, height, binary_claim_hashes, list(deleted_names))
|
|
|
|
r(self._update_effective_amount, height)
|
|
r(self._perform_overtake, height, [], [])
|
|
|
|
def get_expiring(self, height):
|
|
return self.execute(
|
|
f"SELECT claim_hash, normalized FROM claim WHERE expiration_height = {height}"
|
|
)
|
|
|
|
def advance_txs(self, height, all_txs, header, daemon_height, timer):
|
|
insert_claims = []
|
|
update_claims = []
|
|
delete_claim_hashes = set()
|
|
insert_supports = []
|
|
delete_support_txo_hashes = set()
|
|
recalculate_claim_hashes = set() # added/deleted supports, added/updated claim
|
|
deleted_claim_names = set()
|
|
delete_others = set()
|
|
body_timer = timer.add_timer('body')
|
|
for position, (etx, txid) in enumerate(all_txs):
|
|
tx = timer.run(
|
|
Transaction, etx.serialize(), height=height, position=position
|
|
)
|
|
# Inputs
|
|
spent_claims, spent_supports, spent_others = timer.run(
|
|
self.split_inputs_into_claims_supports_and_other, tx.inputs
|
|
)
|
|
body_timer.start()
|
|
delete_claim_hashes.update({r['claim_hash'] for r in spent_claims})
|
|
deleted_claim_names.update({r['normalized'] for r in spent_claims})
|
|
delete_support_txo_hashes.update({r['txo_hash'] for r in spent_supports})
|
|
recalculate_claim_hashes.update({r['claim_hash'] for r in spent_supports})
|
|
delete_others.update(spent_others)
|
|
# Outputs
|
|
for output in tx.outputs:
|
|
if output.is_support:
|
|
insert_supports.append(output)
|
|
recalculate_claim_hashes.add(output.claim_hash)
|
|
elif output.script.is_claim_name:
|
|
insert_claims.append(output)
|
|
recalculate_claim_hashes.add(output.claim_hash)
|
|
elif output.script.is_update_claim:
|
|
claim_hash = output.claim_hash
|
|
update_claims.append(output)
|
|
recalculate_claim_hashes.add(claim_hash)
|
|
delete_claim_hashes.discard(claim_hash)
|
|
delete_others.discard(output.ref.hash) # claim insertion and update occurring in the same block
|
|
body_timer.stop()
|
|
|
|
skip_claim_timer = timer.add_timer('skip insertion of abandoned claims')
|
|
skip_claim_timer.start()
|
|
for new_claim in list(insert_claims):
|
|
if new_claim.ref.hash in delete_others:
|
|
insert_claims.remove(new_claim)
|
|
skip_claim_timer.stop()
|
|
|
|
expire_timer = timer.add_timer('recording expired claims')
|
|
expire_timer.start()
|
|
for expired in self.get_expiring(height):
|
|
delete_claim_hashes.add(expired['claim_hash'])
|
|
deleted_claim_names.add(expired['normalized'])
|
|
expire_timer.stop()
|
|
|
|
r = timer.run
|
|
r(self.delete_claims, delete_claim_hashes)
|
|
r(self.delete_supports, delete_support_txo_hashes)
|
|
r(self.insert_claims, insert_claims, header)
|
|
r(self.update_claims, update_claims, header)
|
|
r(self.validate_channel_signatures, height, insert_claims,
|
|
update_claims, delete_claim_hashes, forward_timer=True)
|
|
r(self.insert_supports, insert_supports)
|
|
r(self.update_claimtrie, height, recalculate_claim_hashes, deleted_claim_names, forward_timer=True)
|
|
r(calculate_trending, self.db, height, self.main.first_sync, daemon_height)
|
|
|
|
def get_claims(self, cols, join=True, **constraints):
|
|
if 'order_by' in constraints:
|
|
sql_order_by = []
|
|
for order_by in constraints['order_by']:
|
|
is_asc = order_by.startswith('^')
|
|
column = order_by[1:] if is_asc else order_by
|
|
if column not in self.ORDER_FIELDS:
|
|
raise NameError(f'{column} is not a valid order_by field')
|
|
if column == 'name':
|
|
column = 'normalized'
|
|
sql_order_by.append(
|
|
f"claim.{column} ASC" if is_asc else f"claim.{column} DESC"
|
|
)
|
|
constraints['order_by'] = sql_order_by
|
|
|
|
ops = {'<=': '__lte', '>=': '__gte', '<': '__lt', '>': '__gt'}
|
|
for constraint in self.INTEGER_PARAMS:
|
|
if constraint in constraints:
|
|
value = constraints.pop(constraint)
|
|
postfix = ''
|
|
if isinstance(value, str):
|
|
if len(value) >= 2 and value[:2] in ops:
|
|
postfix, value = ops[value[:2]], int(value[2:])
|
|
elif len(value) >= 1 and value[0] in ops:
|
|
postfix, value = ops[value[0]], int(value[1:])
|
|
constraints[f'claim.{constraint}{postfix}'] = value
|
|
|
|
if constraints.pop('is_controlling', False):
|
|
if {'sequence', 'amount_order'}.isdisjoint(constraints):
|
|
join = True
|
|
constraints['claimtrie.claim_hash__is_not_null'] = ''
|
|
if 'sequence' in constraints:
|
|
constraints['order_by'] = 'claim.activation_height ASC'
|
|
constraints['offset'] = int(constraints.pop('sequence')) - 1
|
|
constraints['limit'] = 1
|
|
if 'amount_order' in constraints:
|
|
constraints['order_by'] = 'claim.effective_amount DESC'
|
|
constraints['offset'] = int(constraints.pop('amount_order')) - 1
|
|
constraints['limit'] = 1
|
|
|
|
if 'claim_id' in constraints:
|
|
claim_id = constraints.pop('claim_id')
|
|
if len(claim_id) == 40:
|
|
constraints['claim.claim_id'] = claim_id
|
|
else:
|
|
constraints['claim.claim_id__like'] = f'{claim_id[:40]}%'
|
|
|
|
if 'name' in constraints:
|
|
constraints['claim.normalized'] = normalize_name(constraints.pop('name'))
|
|
|
|
if 'public_key_id' in constraints:
|
|
constraints['claim.public_key_hash'] = sqlite3.Binary(
|
|
self.ledger.address_to_hash160(constraints.pop('public_key_id')))
|
|
|
|
if 'channel' in constraints:
|
|
channel_url = constraints.pop('channel')
|
|
match = self._resolve_one(channel_url)
|
|
if isinstance(match, sqlite3.Row):
|
|
constraints['channel_hash'] = match['claim_hash']
|
|
else:
|
|
raise LookupError(f'Could not resolve channel "{channel_url}".')
|
|
if 'channel_hash' in constraints:
|
|
constraints['claim.channel_hash'] = sqlite3.Binary(constraints.pop('channel_hash'))
|
|
if 'channel_ids' in constraints:
|
|
channel_ids = constraints.pop('channel_ids')
|
|
if channel_ids:
|
|
constraints['claim.channel_hash__in'] = [
|
|
sqlite3.Binary(unhexlify(cid)[::-1]) for cid in channel_ids
|
|
]
|
|
|
|
if 'txid' in constraints:
|
|
tx_hash = unhexlify(constraints.pop('txid'))[::-1]
|
|
nout = constraints.pop('nout', 0)
|
|
constraints['claim.txo_hash'] = sqlite3.Binary(
|
|
tx_hash + struct.pack('<I', nout)
|
|
)
|
|
|
|
if 'claim_type' in constraints:
|
|
constraints['claim.claim_type'] = CLAIM_TYPES[constraints.pop('claim_type')]
|
|
if 'stream_types' in constraints:
|
|
stream_types = constraints.pop('stream_types')
|
|
if stream_types:
|
|
constraints['claim.stream_type__in'] = [
|
|
STREAM_TYPES[stream_type] for stream_type in stream_types
|
|
]
|
|
if 'media_types' in constraints:
|
|
media_types = constraints.pop('media_types')
|
|
if media_types:
|
|
constraints['claim.media_type__in'] = media_types
|
|
|
|
_apply_constraints_for_array_attributes(constraints, 'tag')
|
|
_apply_constraints_for_array_attributes(constraints, 'language')
|
|
_apply_constraints_for_array_attributes(constraints, 'location')
|
|
|
|
select = f"SELECT {cols} FROM claim"
|
|
|
|
sql, values = query(
|
|
select if not join else select+"""
|
|
LEFT JOIN claimtrie USING (claim_hash)
|
|
LEFT JOIN claim as channel ON (claim.channel_hash=channel.claim_hash)
|
|
""", **constraints
|
|
)
|
|
try:
|
|
return self.db.execute(sql, values).fetchall()
|
|
except:
|
|
self.logger.exception(f'Failed to execute claim search query: {sql}')
|
|
raise
|
|
|
|
def get_claims_count(self, **constraints):
|
|
constraints.pop('offset', None)
|
|
constraints.pop('limit', None)
|
|
constraints.pop('order_by', None)
|
|
count = self.get_claims('count(*)', join=False, **constraints)
|
|
return count[0][0]
|
|
|
|
def _search(self, **constraints):
|
|
return self.get_claims(
|
|
"""
|
|
claimtrie.claim_hash as is_controlling,
|
|
claim.claim_hash, claim.txo_hash,
|
|
claim.claims_in_channel,
|
|
claim.height, claim.creation_height,
|
|
claim.activation_height, claim.expiration_height,
|
|
claim.effective_amount, claim.support_amount,
|
|
claim.trending_group, claim.trending_mixed,
|
|
claim.trending_local, claim.trending_global,
|
|
claim.short_url, claim.canonical_url,
|
|
claim.channel_hash, channel.txo_hash AS channel_txo_hash,
|
|
channel.height AS channel_height, claim.signature_valid
|
|
""", **constraints
|
|
)
|
|
|
|
INTEGER_PARAMS = {
|
|
'height', 'creation_height', 'activation_height', 'expiration_height',
|
|
'timestamp', 'creation_timestamp', 'release_time',
|
|
'tx_position', 'channel_join', 'signature_valid',
|
|
'amount', 'effective_amount', 'support_amount',
|
|
'trending_group', 'trending_mixed',
|
|
'trending_local', 'trending_global',
|
|
}
|
|
|
|
SEARCH_PARAMS = {
|
|
'name', 'claim_id', 'txid', 'nout', 'channel', 'channel_ids', 'public_key_id',
|
|
'claim_type', 'stream_types', 'media_types',
|
|
'any_tags', 'all_tags', 'not_tags',
|
|
'any_locations', 'all_locations', 'not_locations',
|
|
'any_languages', 'all_languages', 'not_languages',
|
|
'is_controlling', 'limit', 'offset', 'order_by'
|
|
} | INTEGER_PARAMS
|
|
|
|
ORDER_FIELDS = {
|
|
'name',
|
|
} | INTEGER_PARAMS
|
|
|
|
def search(self, constraints) -> Tuple[List, List, int, int]:
|
|
assert set(constraints).issubset(self.SEARCH_PARAMS), \
|
|
f"Search query contains invalid arguments: {set(constraints).difference(self.SEARCH_PARAMS)}"
|
|
total = self.get_claims_count(**constraints)
|
|
constraints['offset'] = abs(constraints.get('offset', 0))
|
|
constraints['limit'] = min(abs(constraints.get('limit', 10)), 50)
|
|
if 'order_by' not in constraints:
|
|
constraints['order_by'] = ["height", "^name"]
|
|
txo_rows = self._search(**constraints)
|
|
channel_hashes = set(txo['channel_hash'] for txo in txo_rows if txo['channel_hash'])
|
|
extra_txo_rows = []
|
|
if channel_hashes:
|
|
extra_txo_rows = self._search(**{'claim.claim_hash__in': [sqlite3.Binary(h) for h in channel_hashes]})
|
|
return txo_rows, extra_txo_rows, constraints['offset'], total
|
|
|
|
def _resolve_one(self, raw_url):
|
|
try:
|
|
url = URL.parse(raw_url)
|
|
except ValueError as e:
|
|
return e
|
|
|
|
channel = None
|
|
|
|
if url.has_channel:
|
|
query = url.channel.to_dict()
|
|
if set(query) == {'name'}:
|
|
query['is_controlling'] = True
|
|
else:
|
|
query['order_by'] = ['^height']
|
|
matches = self._search(**query, limit=1)
|
|
if matches:
|
|
channel = matches[0]
|
|
else:
|
|
return LookupError(f'Could not find channel in "{raw_url}".')
|
|
|
|
if url.has_stream:
|
|
query = url.stream.to_dict()
|
|
if channel is not None:
|
|
if set(query) == {'name'}:
|
|
# temporarily emulate is_controlling for claims in channel
|
|
query['order_by'] = ['effective_amount']
|
|
else:
|
|
query['order_by'] = ['^channel_join']
|
|
query['channel_hash'] = channel['claim_hash']
|
|
query['signature_valid'] = 1
|
|
elif set(query) == {'name'}:
|
|
query['is_controlling'] = 1
|
|
matches = self._search(**query, limit=1)
|
|
if matches:
|
|
return matches[0]
|
|
else:
|
|
return LookupError(f'Could not find stream in "{raw_url}".')
|
|
|
|
return channel
|
|
|
|
def resolve(self, urls) -> Tuple[List, List]:
|
|
result = []
|
|
channel_hashes = set()
|
|
for raw_url in urls:
|
|
match = self._resolve_one(raw_url)
|
|
result.append(match)
|
|
if isinstance(match, sqlite3.Row) and match['channel_hash']:
|
|
channel_hashes.add(match['channel_hash'])
|
|
extra_txo_rows = []
|
|
if channel_hashes:
|
|
extra_txo_rows = self._search(**{'claim.claim_hash__in': [sqlite3.Binary(h) for h in channel_hashes]})
|
|
return result, extra_txo_rows
|
|
|
|
|
|
class LBRYDB(DB):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.sql = SQLDB(self, 'claims.db')
|
|
|
|
def close(self):
|
|
super().close()
|
|
self.sql.close()
|
|
|
|
async def _open_dbs(self, *args, **kwargs):
|
|
await super()._open_dbs(*args, **kwargs)
|
|
self.sql.open()
|