def __init__(self, host):
super().__init__(host)
self._host = host
self.DNS_NAME = 'seaweedfs-s3-localnode.tls.ai' if not helpers.is_k8s(self._host.SshDirect) else 'seaweedfs-s3.default.svc.cluster.local'
self.filer_host = 'seaweedfs-filer-localnode.tls.ai' if not helpers.is_k8s(self._host.SshDirect) else 'seaweedfs-filer.default.svc.cluster.local'
self.filer_port = 8888
self.PORT = 8333
def __init__(self, host):
self._host = host
self.DNS = 'kafka.tls.ai' if not helpers.is_k8s(self._host.SshDirect) \
else 'kafka-cluster-kafka-brokers.default.svc.cluster.local'
self.PORT = '9092'
self._rpyc = None
self._init_lock = threading.Lock()
def __init__(self, host):
self._host = host
self.DNS_NAME = 'rabbitmq.tls.ai' if not helpers.is_k8s(
self._host.SshDirect) else 'rabbitmq-ha.default.svc.cluster.local'
self.PORT = 5672 # amqp
self.ADMIN_PORT = 15672 # admin
self.password = "******" if not helpers.is_k8s(
self._host.SshDirect
) else self._host.SshDirect.execute(
"kubectl get secret --namespace default rabbitmq-secret -o jsonpath='{.data.password}' | base64 --decode"
)
self.user = '******' if not helpers.is_k8s(
self._host.SshDirect) else 'root'
self.virtual_host = '/'
self._amqp_tunnel = None
self._admin_tunnel = None
self._amqp_connection = None
def __init__(self, host):
self._host = host
self.DNS_NAME = 'consul.tls.ai' if not helpers.is_k8s(
self._host.SshDirect
) else 'consul-server.default.svc.cluster.local'
self.PORT = 8500
self.DEFAULT_KEY = "DEFAULT"
self.OVERRIDE_KEY = "OVERRIDE"
self.APPLICATION_KEY = "APPLICATION"
def _create_connection(self, **kwargs):
password = "******" if not helpers.is_k8s(
self._host.SshDirect
) else self._host.SshDirect.execute(
"kubectl get secret --namespace default memsql-secret -o jsonpath='{.data.password}' | base64 --decode"
)
memsql_kwargs = copy.copy(kwargs)
if memsql_kwargs.get('password', None) is None:
memsql_kwargs['password'] = self.password
memsql_kwargs.setdefault('password', password)
memsql_kwargs.setdefault('user', 'root')
memsql_kwargs.setdefault('client_flag', CLIENT.MULTI_STATEMENTS)
memsql_kwargs.setdefault('cursorclass', pymysql.cursors.DictCursor)
return Connection(pymysql.connect(**memsql_kwargs))
def __init__(self, host):
super().__init__(host)
self.is_k8s = helpers.is_k8s(self._host.SshDirect)
self.DNS_NAME = host.ip if not self.is_k8s else prometheus_connection_config[
'host']['k8s']
self.PORT = prometheus_connection_config['port']['compose'] if not self.is_k8s \
else prometheus_connection_config['port']['k8s']
self.start_tunnel(self.DNS_NAME, self.PORT)
self.url = f'{prometheus_connection_config["url"]["compose"]}:{self.local_bind_port}' if not self.is_k8s \
else f'{prometheus_connection_config["url"]["k8s"]}'
if self.is_k8s:
with open('/etc/hosts', 'r+') as f:
content = f.read()
if f"{host.ip} {self.DNS_NAME}" not in content:
logging.info(
f"write new line in hosts file: {host.ip} {self.DNS_NAME}"
)
f.write(f'\n{host.ip} {self.DNS_NAME}\n')
self.headers = None if not self.is_k8s \
else {'Authorization': f'Basic {prometheus_connection_config["auth"]}'}
self._prom = Prometheus(url=self.url, headers=self.headers)
def __init__(self, host):
self._host = host
self.DNS_NAME = 'postgres.tls.ai' if not is_k8s(
self._host.SshDirect) else 'postgres.default.svc.cluster.local'
self.PORT = 5432
self._connection = None
def password(self):
if not helpers.is_k8s(self._host.SshDirect):
return 'password'
return self._host.SshDirect.execute(
"kubectl get secret --namespace default memsql-secret -o jsonpath='{.data.password}' | base64 --decode"
)
def __init__(self, host):
self._host = host
self.DNS_NAME = 'memsql.tls.ai' if not helpers.is_k8s(
self._host.SshDirect) else 'memsql.default.svc.cluster.local'
self.PORT = 3306
self._connection = None
def restart_container_by_service(host, name):
if helpers.is_k8s(host.SshDirect):
host.K8s.restart_pod_by_service_name(name)
else:
host.Docker.restart_container_by_service_name(name)
def start_container_by_service(host, name):
if helpers.is_k8s(host.SshDirect):
raise NotImplementedError("Start container not implemented yet")
else:
host.Docker.run_container_by_service(name)
def reset_state(self):
if not helpers.is_k8s(self._host.SshDirect):
self.delete_storage_compose()
else:
raise Exception(
"Infra doesnt know how to reset_state of kafka on K8s")
def __init__(self, host):
self._host = host
self.DNS_NAME = 'mongodb.tls.ai' if not helpers.is_k8s(
self._host.SshDirect) else 'mongodb.default.svc.cluster.local'
self.PORT = 27017
def __init__(self, host):
self._host = host
self.DNS_NAME = 'redis.tls.ai' if not helpers.is_k8s(
self._host.SshDirect) else 'redis-server.default.svc.cluster.local'
self.PORT = 6379
