def remove_tc(server_id: int, port_num: int):
kwargs = {
"server_id": server_id,
"port_num": port_num,
}
print(f"Sending tc_runner task, kwargs: {kwargs}")
celery_app.send_task("app.tasks.tc.tc_runner", kwargs=kwargs)
def trigger_tc(port: Port):
kwargs = {
"server_id": port.server.id,
"port_num": port.num,
"egress_limit": port.config.get("egress_limit"),
"ingress_limit": port.config.get("ingress_limit"),
}
print(f"Sending tc_runner task, kwargs: {kwargs}")
celery_app.send_task("app.tasks.tc.tc_runner", kwargs=kwargs)
def trigger_install_gost(server_id):
kwargs = {
"port_id": 0,
"server_id": server_id,
"port_num": 0,
"gost_config": {},
"update_gost": True,
"update_status": False,
}
print(f"Sending gost install gost_runner task, kwargs: {kwargs}")
celery_app.send_task("app.tasks.gost.gost_runner", kwargs=kwargs)
def send_gost(
rule: PortForwardRule,
port: Port,
old: PortForwardRuleOut = None,
new: PortForwardRuleOut = None,
update_gost: bool = False,
):
gost_config = generate_gost_config(rule)
kwargs = {
"port_id": port.id,
"server_id": port.server.id,
"port_num": port.num,
"gost_config": gost_config,
"remote_ip": get_gost_remote_ip(gost_config),
"update_gost": update_gost,
"update_status": bool(new and new.method == MethodEnum.GOST),
}
print(f"Sending gost_runner task, kwargs: {kwargs}")
celery_app.send_task("app.tasks.gost.gost_runner", kwargs=kwargs)
def send_iptables(
rule: PortForwardRule,
port: Port,
old: PortForwardRuleOut = None,
new: PortForwardRuleOut = None,
update_gost: bool = False,
):
kwargs = {
"port_id": port.id,
"server_id": port.server.id,
"local_port": port.num,
}
if new and new.method == MethodEnum.IPTABLES:
kwargs["update_status"] = True
kwargs["remote_ip"] = new.config.get("remote_ip")
kwargs["remote_port"] = new.config.get("remote_port")
kwargs["forward_type"] = new.config.get("type", "ALL").upper()
print(f"Sending iptables_runner task, kwargs: {kwargs}")
celery_app.send_task("app.tasks.iptables.iptables_runner", kwargs=kwargs)
def trigger_iptables_reset(port: Port):
kwargs = {"server_id": port.server.id, "port_num": port.num}
print(f"Sending iptables.iptables_reset_runner task")
celery_app.send_task("app.tasks.iptables.iptables_reset_runner",
kwargs=kwargs)
def trigger_ansible_hosts():
print(f"Sending ansible_hosts_runner task")
celery_app.send_task("app.tasks.ansible.ansible_hosts_runner")
async def run_task():
celery_app.send_task("app.tasks.traffic.traffic_runner")
return {"message": "ok"}