45 lines
883 B
Python
45 lines
883 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
'''
|
||
|
|
Kafka benchmark dataset generator.
|
||
|
|
'''
|
||
|
|
|
||
|
|
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
from argparse import ArgumentParser
|
||
|
|
|
||
|
|
|
||
|
|
def get_edge():
|
||
|
|
from_node = random.randint(0, args.nodes)
|
||
|
|
to_node = random.randint(0, args.nodes)
|
||
|
|
|
||
|
|
while from_node == to_node:
|
||
|
|
to_node = random.randint(0, args.nodes)
|
||
|
|
return (from_node, to_node)
|
||
|
|
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
argp = ArgumentParser(description=__doc__)
|
||
|
|
argp.add_argument("--nodes", type=int, default=100, help="Number of nodes.")
|
||
|
|
argp.add_argument("--edges", type=int, default=30, help="Number of edges.")
|
||
|
|
return argp.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
args = parse_args()
|
||
|
|
edges = set()
|
||
|
|
|
||
|
|
for i in range(args.nodes):
|
||
|
|
print("%d\n" % i)
|
||
|
|
|
||
|
|
for i in range(args.edges):
|
||
|
|
edge = get_edge()
|
||
|
|
while edge in edges:
|
||
|
|
edge = get_edge()
|
||
|
|
|
||
|
|
edges.add(edge)
|
||
|
|
print("%d %d\n" % edge)
|
||
|
|
|
||
|
|
sys.exit(0)
|