Add task to faasmtools to flush planner (#117)

* planner: add task to faasmtools to flush planner

* fix: add default values

Author: csegarragonz

faasmtools/endpoints.py

@@ -6,8 +6,10 @@
 DEFAULT_KNATIVE_HEADERS = {"Host": "faasm-worker.faasm.example.com"}
 
 DEFAULT_INVOKE_HOST = "worker"
+DEFAULT_PLANNER_HOST = "planner"
 DEFAULT_UPLOAD_HOST = "upload"
 DEFAULT_INVOKE_PORT = 8080
+DEFAULT_PLANNER_PORT = 8081
 DEFAULT_UPLOAD_PORT = 8002
 
 
@@ -45,6 +47,16 @@ def get_faasm_invoke_host_port():
     return host, port
 
 
+def get_faasm_planner_host_port():
+    if not faasm_config_exists():
+        return DEFAULT_PLANNER_HOST, DEFAULT_PLANNER_PORT
+
+    host = get_faasm_ini_value("Faasm", "planner_host")
+    port = get_faasm_ini_value("Faasm", "planner_port")
+
+    return host, port
+
+
 def get_knative_headers():
     if not faasm_config_exists():
         return DEFAULT_KNATIVE_HEADERS

tasks/func.py

@@ -9,6 +9,7 @@
 from faasmtools.env import PROJ_ROOT
 from faasmtools.endpoints import (
     get_faasm_invoke_host_port,
+    get_faasm_planner_host_port,
     get_faasm_upload_host_port,
     get_knative_headers,
 )
@@ -151,7 +152,7 @@ def flush(ctx):
     Flush the Faasm cluster
     """
     headers = get_knative_headers()
-    host, port = get_faasm_invoke_host_port()
+    host, port = get_faasm_planner_host_port()
 
     url = "http://{}:{}".format(host, port)
     data = {"type": FAABRIC_MSG_TYPE_FLUSH}