148 lines
4.3 KiB
Python
Executable File
148 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from counteriter import CounterIterator
|
|
from toporender import main as renderer
|
|
|
|
virtual_servers_per_physical_server = 0
|
|
physical_servers_per_rack_tower = 1
|
|
rack_tower_per_corridor = 2
|
|
corridors = 3
|
|
spines = 4
|
|
|
|
linkspd_virtual_physical = 0.1 # 100 mbps
|
|
linkspd_physicalhost_switch = 0.1 # 100 mbps
|
|
linkspd_physical_tower = 0.1 # 100 mbps
|
|
linkspd_tower_corridor = 1 # 1 gbps
|
|
linkspd_corridor_spine = 1 # 1 gbps
|
|
linkspd_spine_spine = 10 # 10 gbps
|
|
|
|
# overriding some stuff
|
|
linkspd_virtual_physical = 1 # 1 mbps
|
|
linkspd_physicalhost_switch = 1 # 1 mbps
|
|
linkspd_physical_tower = 1 # 1 mbps
|
|
linkspd_tower_corridor = 1 # 1 mbps
|
|
linkspd_corridor_spine = 1 # 1 mbps
|
|
linkspd_spine_spine = 1 # 1 mbps
|
|
|
|
hosts_iter = CounterIterator()
|
|
switch_iter = CounterIterator()
|
|
|
|
|
|
def reset_iters():
|
|
global hosts_iter
|
|
global switch_iter
|
|
hosts_iter = CounterIterator()
|
|
switch_iter = CounterIterator()
|
|
|
|
|
|
def create_virtual_server():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
this_hosts.append("h%d" % next(hosts_iter))
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def create_physical_server():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
serv = "h%d" % next(hosts_iter)
|
|
sw = "s%d" % next(switch_iter)
|
|
for _ in range(virtual_servers_per_physical_server):
|
|
chd_hosts, chd_switches, chd_links = create_virtual_server()
|
|
this_hosts += chd_hosts
|
|
this_switches += chd_switches
|
|
this_links += chd_links
|
|
this_links.append((chd_hosts[-1], sw, linkspd_virtual_physical))
|
|
this_hosts.append(serv)
|
|
this_switches.append(sw)
|
|
this_links.append((serv, sw, linkspd_physicalhost_switch))
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def create_racktower():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
sw = "s%d" % next(switch_iter)
|
|
for _ in range(physical_servers_per_rack_tower):
|
|
chd_hosts, chd_switches, chd_links = create_physical_server()
|
|
this_hosts += chd_hosts
|
|
this_switches += chd_switches
|
|
this_links += chd_links
|
|
lsw = chd_switches[-1]
|
|
this_links.append((lsw, sw, linkspd_physical_tower))
|
|
this_switches.append(sw)
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def create_corridor():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
sw1 = "s%d" % next(switch_iter)
|
|
sw2 = "s%d" % next(switch_iter)
|
|
for _ in range(rack_tower_per_corridor):
|
|
chd_hosts, chd_switches, chd_links = create_racktower()
|
|
this_hosts += chd_hosts
|
|
this_switches += chd_switches
|
|
this_links += chd_links
|
|
lsw = chd_switches[-1]
|
|
this_links.append((lsw, sw1, linkspd_tower_corridor))
|
|
this_links.append((lsw, sw2, linkspd_tower_corridor))
|
|
this_switches.append(sw1)
|
|
this_switches.append(sw2)
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def create_spine():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
swsps = ["s%d" % next(switch_iter) for _ in range(spines)]
|
|
for i in range(spines-1):
|
|
chd_hosts, chd_switches, chd_links = create_corridor()
|
|
this_hosts += chd_hosts
|
|
this_switches += chd_switches
|
|
this_links += chd_links
|
|
lsw1, lsw2 = chd_switches[-2:]
|
|
this_links.append((lsw1, swsps[i], linkspd_corridor_spine))
|
|
this_links.append((lsw2, swsps[i+1], linkspd_corridor_spine))
|
|
this_switches += swsps
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def create_datacenter():
|
|
this_hosts = []
|
|
this_switches = []
|
|
this_links = []
|
|
last_spined_corridor = []
|
|
for _ in range(corridors):
|
|
chd_hosts, chd_switches, chd_links = create_spine()
|
|
this_hosts += chd_hosts
|
|
this_switches += chd_switches
|
|
this_links += chd_links
|
|
spined_corridor = chd_switches[-spines:]
|
|
if len(last_spined_corridor) > 0:
|
|
for i in range(spines):
|
|
this_links.append(
|
|
(last_spined_corridor[i], spined_corridor[i], linkspd_spine_spine))
|
|
last_spined_corridor = spined_corridor
|
|
return this_hosts, this_switches, this_links
|
|
|
|
|
|
def main(fn: str = 'bigtopo'):
|
|
reset_iters()
|
|
datacenter = create_datacenter()
|
|
reset_iters()
|
|
Path(f'{fn}.json').write_text(json.dumps(datacenter))
|
|
renderer(fn)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|