Add adapter_api.py
This commit is contained in:
182
adapter_api.py
Normal file
182
adapter_api.py
Normal file
@@ -0,0 +1,182 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Vector Zulu — KoreID Adapter API
|
||||
Port 8766 on Charlie.
|
||||
Provides status, queue monitoring, and manual event injection.
|
||||
"""
|
||||
|
||||
import os, json, time, logging
|
||||
from datetime import datetime, timezone
|
||||
from flask import Flask, jsonify, request, abort
|
||||
import redis
|
||||
import psycopg2
|
||||
from koreid import KoreIDGenerator, wrap_ethereum_event
|
||||
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s [ADAPTER-API] %(message)s')
|
||||
app = Flask(__name__)
|
||||
|
||||
QUEUE_KEY = 'korechain:outbound_queue'
|
||||
DEAD_LETTER = 'korechain:dead_letter'
|
||||
|
||||
|
||||
def get_redis():
|
||||
return redis.Redis(
|
||||
host=os.environ.get('REDIS_HOST', 'localhost'),
|
||||
port=6379, decode_responses=True
|
||||
)
|
||||
|
||||
|
||||
def get_pg():
|
||||
return psycopg2.connect(
|
||||
host=os.environ.get('POSTGRES_HOST', 'localhost'),
|
||||
dbname=os.environ.get('POSTGRES_DB', 'vz_oracle'),
|
||||
user=os.environ.get('POSTGRES_USER', 'vz_oracle'),
|
||||
password=os.environ.get('POSTGRES_PASSWORD', ''),
|
||||
connect_timeout=10
|
||||
)
|
||||
|
||||
|
||||
@app.route('/health')
|
||||
def health():
|
||||
try:
|
||||
r = get_redis()
|
||||
r.ping()
|
||||
queue_len = r.llen(QUEUE_KEY)
|
||||
retry_len = r.zcard('korechain:retry_queue')
|
||||
dead_len = r.llen(DEAD_LETTER)
|
||||
last_block = r.get('eth_listener:last_block')
|
||||
return jsonify({
|
||||
'status': 'ok',
|
||||
'queue_depth': queue_len,
|
||||
'retry_queue': retry_len,
|
||||
'dead_letter': dead_len,
|
||||
'last_eth_block': last_block,
|
||||
'timestamp': datetime.now(timezone.utc).isoformat(),
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({'status': 'error', 'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/queue')
|
||||
def queue_status():
|
||||
try:
|
||||
r = get_redis()
|
||||
queue_items = r.lrange(QUEUE_KEY, 0, 9) # last 10
|
||||
return jsonify({
|
||||
'queue_depth': r.llen(QUEUE_KEY),
|
||||
'recent': [json.loads(i) for i in queue_items],
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/events')
|
||||
def recent_events():
|
||||
try:
|
||||
pg = get_pg()
|
||||
with pg.cursor() as cur:
|
||||
cur.execute("""
|
||||
SELECT koreid, guid, event_type, token_symbol,
|
||||
tx_hash, amount, status, posted_at
|
||||
FROM kore_outbound_events
|
||||
ORDER BY posted_at DESC
|
||||
LIMIT 50
|
||||
""")
|
||||
rows = cur.fetchall()
|
||||
pg.close()
|
||||
return jsonify({'events': [{
|
||||
'koreid': r[0],
|
||||
'guid': r[1],
|
||||
'event_type': r[2],
|
||||
'token': r[3],
|
||||
'tx_hash': r[4],
|
||||
'amount': float(r[5]) if r[5] else 0,
|
||||
'status': r[6],
|
||||
'posted_at': r[7].isoformat() if r[7] else None,
|
||||
} for r in rows]})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/generate', methods=['POST'])
|
||||
def generate_koreid():
|
||||
"""Manually generate a KoreID for testing."""
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
abort(400)
|
||||
try:
|
||||
gen = KoreIDGenerator()
|
||||
result = gen.generate(
|
||||
event_type=data.get('event_type', 'EXTERNAL'),
|
||||
currency=data.get('currency', 'USD'),
|
||||
current_network=data.get('network', 'ETHEREUM'),
|
||||
)
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/inject', methods=['POST'])
|
||||
def inject_event():
|
||||
"""
|
||||
Manually inject an Ethereum event into the queue.
|
||||
Useful for testing or replaying missed events.
|
||||
"""
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
abort(400)
|
||||
try:
|
||||
r = get_redis()
|
||||
gen = KoreIDGenerator()
|
||||
payload = wrap_ethereum_event(
|
||||
event_type=data['event_type'],
|
||||
amount=float(data['amount']),
|
||||
currency=data['currency'],
|
||||
tx_hash=data['tx_hash'],
|
||||
block_number=int(data.get('block_number', 0)),
|
||||
from_address=data.get('from_address', '0x0'),
|
||||
to_address=data.get('to_address', '0x0'),
|
||||
token_symbol=data['token_symbol'],
|
||||
koreid_generator=gen,
|
||||
linked_koreid=data.get('linked_koreid'),
|
||||
)
|
||||
r.lpush(QUEUE_KEY, json.dumps(payload))
|
||||
return jsonify({'queued': True, 'koreid': payload['koreid']})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/dead-letter')
|
||||
def dead_letter():
|
||||
try:
|
||||
r = get_redis()
|
||||
items = r.lrange(DEAD_LETTER, 0, 19)
|
||||
return jsonify({
|
||||
'count': r.llen(DEAD_LETTER),
|
||||
'items': [json.loads(i) for i in items],
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
@app.route('/dead-letter/retry', methods=['POST'])
|
||||
def retry_dead_letter():
|
||||
"""Move all dead letter items back to main queue for retry."""
|
||||
try:
|
||||
r = get_redis()
|
||||
count = 0
|
||||
while True:
|
||||
item = r.rpop(DEAD_LETTER)
|
||||
if not item:
|
||||
break
|
||||
payload = json.loads(item)
|
||||
payload['_retries'] = 0 # reset retry count
|
||||
r.lpush(QUEUE_KEY, json.dumps(payload))
|
||||
count += 1
|
||||
return jsonify({'requeued': count})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=8766, debug=False)
|
||||
Reference in New Issue
Block a user