blob: 06c5ec95f3ab7c25c35ce4b6878245862a52725a [file] [log] [blame]
Jon Hall3b489db2015-10-05 14:38:37 -07001#!/usr/bin/env python
Jon Halle9b1fa32015-12-08 15:32:21 -08002"""
Jon Hall3b489db2015-10-05 14:38:37 -07003 Generate the partitions json file from the $OC* environment variables
4
5 Usage: onos-gen-partitions [output file]
6 If output file is not provided, the json is written to stdout.
Jon Halle9b1fa32015-12-08 15:32:21 -08007"""
Jon Hall3b489db2015-10-05 14:38:37 -07008
9from os import environ
10from collections import deque, OrderedDict
11import re
12import json
13import sys
Jon Halle9b1fa32015-12-08 15:32:21 -080014import hashlib
Jon Hall3b489db2015-10-05 14:38:37 -070015
16convert = lambda text: int(text) if text.isdigit() else text.lower()
17alphanum_key = lambda key: [convert(c) for c in re.split('([0-9]+)', key)]
18
19def get_OC_vars():
20 vars = []
21 for var in environ:
22 if re.match(r"OC[0-9]+", var):
23 vars.append(var)
24 return sorted(vars, key=alphanum_key)
25
26def get_nodes(vars, port=9876):
Jon Hall37677302015-10-23 12:17:49 -070027 node = lambda k: { 'id': k, 'ip': k, 'port': port }
Jon Hall3b489db2015-10-05 14:38:37 -070028 return [ node(environ[v]) for v in vars ]
29
Jon Hall9288fdd2016-01-19 11:51:27 -080030def generate_base_partition(nodes):
31 return {
32 'id': 0,
33 'members': nodes
34 }
35
36def generate_extended_partitions(nodes, k):
Jon Hall3b489db2015-10-05 14:38:37 -070037 l = deque(nodes)
Jon Hall37677302015-10-23 12:17:49 -070038 perms = []
Jon Hall3b489db2015-10-05 14:38:37 -070039 for i in range(1, len(nodes)+1):
Jon Hall37677302015-10-23 12:17:49 -070040 part = {
Jon Hall9288fdd2016-01-19 11:51:27 -080041 'id': i,
Jon Hall37677302015-10-23 12:17:49 -070042 'members': list(l)[:k]
43 }
44 perms.append(part)
Jon Hall3b489db2015-10-05 14:38:37 -070045 l.rotate(-1)
Jon Hall37677302015-10-23 12:17:49 -070046 return perms
Jon Hall3b489db2015-10-05 14:38:37 -070047
Jon Hall9288fdd2016-01-19 11:51:27 -080048def generate_extended_partitions_HA(nodes, k):
Jon Hall3b489db2015-10-05 14:38:37 -070049 l = deque(nodes)
Jon Hall37677302015-10-23 12:17:49 -070050 perms = []
Jon Hall3b489db2015-10-05 14:38:37 -070051 for i in range(1, (len(nodes) + 1) / 2 + 1):
Jon Hall37677302015-10-23 12:17:49 -070052 part = {
Jon Hall9288fdd2016-01-19 11:51:27 -080053 'id': i,
Jon Hall37677302015-10-23 12:17:49 -070054 'members': list(l)[:k]
55 }
56 perms.append(part)
Jon Hall3b489db2015-10-05 14:38:37 -070057 l.rotate(-2)
Jon Hall37677302015-10-23 12:17:49 -070058 return perms
Jon Hall3b489db2015-10-05 14:38:37 -070059
60if __name__ == '__main__':
61 vars = get_OC_vars()
62 nodes = get_nodes(vars)
Jon Hall9288fdd2016-01-19 11:51:27 -080063 base_partition = generate_base_partition([v.get('id') for v in nodes])
64 extended_partitions = generate_extended_partitions_HA([v.get('id') for v in nodes], 3)
65 partitions = []
66 partitions.append(base_partition)
67 partitions.extend(extended_partitions)
Jon Halle9b1fa32015-12-08 15:32:21 -080068 name = 0
69 for node in nodes:
Jon Hall9288fdd2016-01-19 11:51:27 -080070 name = name ^ hash(node['ip'])
Jon Hall3b489db2015-10-05 14:38:37 -070071 data = {
Jon Halle9b1fa32015-12-08 15:32:21 -080072 'name': name,
Jon Hall3b489db2015-10-05 14:38:37 -070073 'nodes': nodes,
74 'partitions': partitions
75 }
76 output = json.dumps(data, indent=4)
77
78 if len(sys.argv) == 2:
79 filename = sys.argv[1]
80 with open(filename, 'w') as f:
81 f.write(output)
82 else:
83 print output