def __init__(self, ecm_hash, template=TEMPLATE_FILE, as_json=False):
if as_json:
ecm_hash = json.loads(ecm_hash)
if not self._check(ecm_hash):
raise Exception('Invalid hash received')
if not path.isfile(template):
warnings.warn("Template file not found, using default config")
template = mktemp()
f = open(template, 'w')
f.write(TEMPLATE_CONFIG)
f.close()
self.template = template
self.ecm_hash = ecm_hash
self.config = HAProxyConfig(template)
self._loads()
class ECMHAConfig:
def __init__(self, ecm_hash, template=TEMPLATE_FILE, as_json=False):
if as_json:
ecm_hash = json.loads(ecm_hash)
if not self._check(ecm_hash):
raise Exception('Invalid hash received')
if not path.isfile(template):
warnings.warn("Template file not found, using default config")
template = mktemp()
f = open(template, 'w')
f.write(TEMPLATE_CONFIG)
f.close()
self.template = template
self.ecm_hash = ecm_hash
self.config = HAProxyConfig(template)
self._loads()
def show(self, as_json=False):
config_to_show = self.config.createHash()
if as_json:
config_to_show = json.dumps(config_to_show)
print config_to_show
def read(self, read_file, as_json=True):
f = open(read_file, 'r')
first_line = f.readline()
f.close()
if first_line.startswith('#'):
json_config = first_line.split('#')[1]
if not as_json:
json_config = json.loads(json_config)
return json_config
return False
def valid(self):
config = self.config.getConfig()
return is_valid(config)
def write(self, write_file):
if write_file == self.template:
raise Exception("Can't use template as final file")
config = self.config.getConfig()
f = open(write_file, 'w')
f.write("#" + json.dumps(self.ecm_hash) + "\n")
f.write(config)
f.close()
def _check(self, ecm_hash):
if ecm_hash.get('health', None) is None:
return False
if ecm_hash.get('listeners', None) is None:
return False
if ecm_hash.get('backends', None) is None:
return False
return True
# private functions
def _loads(self):
health_timeout = self.ecm_hash['health']['timeout']
health_threshold = self.ecm_hash['health']['threshold']
health_interval = self.ecm_hash['health']['threshold']
health_check = self.ecm_hash['health']['check']
listeners = False
# Get check data
(hproto, hport, hpath) = health_check.split('::')
for listener in self.ecm_hash['listeners']:
listeners = True
(fproto, fport, bproto, bport) = listener.split('::')
self.config.addListen(name="front-%s" % fport, port=fport, mode=self._protocol_dic(fproto))
listen = self.config.getListen("front-%s" % fport)
# Add check timeout configuration
listen.addOption(Option('timeout', ('check %i' % (health_timeout*1000),)))
# Add cookie only if there are backend (config crashes otherwise)
if len(self.ecm_hash['backends']) and self._protocol_dic(fproto) == 'http':
listen.addOption(Option('cookie', ('ECM-HA-ID insert indirect nocache maxidle 30m maxlife 8h',)))
if hpath:
check = 'httpchk HEAD %s' % hpath
listen.addOption(Option('option', (check,)))
for backend in self.ecm_hash['backends']:
uuid = backend.keys()[0]
server = '%s %s:%s check inter %i rise %i fall %i' % (uuid, backend[uuid], fport, health_interval*1000,
health_threshold, health_threshold)
if self._protocol_dic(fproto) == 'http':
server += ' cookie %s' % uuid
listen.addOption(Option('server', (server,)))
if not listeners:
# Add default listener
self.config.addListen(name="empty", port=80, mode='http')
@staticmethod
def _protocol_dic(proto):
retval = 'tcp'
if proto.lower() == 'http':
retval = 'http'
return retval
