import sys
import psycopg

if len(sys.argv) != 2:
    print("Usage: python generate_multixids.py <count>")
    sys.exit(1)

count = int(sys.argv[1])

print(f"Generating {count} multixact IDs...")

# Determine per-worker iterations
num_conns = 10

conns = []
for i in range(num_conns):
    conn = psycopg.connect(host="localhost", dbname="postgres", user="postgres")
    conns.append(conn)

for i in range(count):
    if i % 1000 == 0:
        print(f"processing {i} / {count}...")

    conn = conns[i % num_conns];
    conn.commit()
    cur = conn.cursor()
    cur.execute("SELECT * FROM t WHERE id=1 FOR SHARE")

print("Done generating multixact IDs.")