1 files changed, 131 insertions, 0 deletions
diff --git a/relationships/relationshipplugin/tasks.py b/relationships/relationshipplugin/tasks.py
new file mode 100644
index 0000000..b17403c
--- /dev/null
+++ b/relationships/relationshipplugin/tasks.py
@@ -0,0 +1,131 @@
+# ============LICENSE_START=======================================================
+# org.onap.dcae
+# ================================================================================
+# Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
+# ================================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END=========================================================
+#
+# ECOMP is a trademark and service mark of AT&T Intellectual Property.
+
+import json
+from cloudify import ctx
+from cloudify.decorators import operation
+from cloudify.exceptions import NonRecoverableError
+from relationshipplugin import discovery as dis
+import requests
+
+
+SERVICE_COMPONENT_NAME = "service_component_name"
+SELECTED_CONTAINER_DESTINATION = "selected_container_destination"
+CONSUL_HOST = "consul_host"
+
+CONSUL_HOSTNAME = "localhost"
+
+
+# Lifecycle interface calls for component_connect_to
+
+# NOTE: ctx.source and ctx.target are RelationshipSubjectContext
+# Order of operation of relationships is bit confusing. These operations are
+# implemented for `target_interfaces`. By observation, the target node processed,
+# then the source is created, the relationship is run then the source is started.
+# http://getcloudify.org/guide/3.1/dsl-spec-relationships.html#relationship-interfaces
+
+@operation
+def add_relationship(**kwargs):
+    """Adds target to the source relationship list"""
+    try:
+        conn = dis.create_kv_conn(CONSUL_HOSTNAME)
+
+        source_name = ctx.source.instance.runtime_properties[SERVICE_COMPONENT_NAME]
+        # The use case for using the target name override is for the platform
+        # blueprint where the cdap broker needs to connect to a cdap cluster but
+        # the cdap cluster does not not use the component plugins so the name is
+        # not generated.
+        # REVIEW: Re-review this
+        target_name = kwargs["target_name_override"] \
+            if "target_name_override" in kwargs \
+            else ctx.target.instance.runtime_properties[SERVICE_COMPONENT_NAME]
+
+        dis.store_relationship(conn, source_name, target_name)
+        ctx.logger.info("Created relationship: {0} to {1}".format(source_name,
+            target_name))
+    except Exception as e:
+        ctx.logger.error("Unexpected error while adding relationship: {0}"
+                .format(str(e)))
+        raise NonRecoverableError(e)
+
+@operation
+def remove_relationship(**kwargs):
+    """Removes target from the source relationship list"""
+    try:
+        conn = dis.create_kv_conn(CONSUL_HOSTNAME)
+
+        source_name = ctx.source.instance.runtime_properties[SERVICE_COMPONENT_NAME]
+        dis.delete_relationship(conn, source_name)
+        ctx.logger.info("Removed relationship: {0}".format(source_name))
+    except Exception as e:
+        ctx.logger.error("Unexpected error while removing relationship: {0}"
+                .format(str(e)))
+        raise NonRecoverableError(e)
+
+
+# Lifecycle interface calls for component_contained_in
+
+@operation
+def forward_destination_info(**kwargs):
+    try:
+        selected_target = ctx.target.instance.runtime_properties[SERVICE_COMPONENT_NAME]
+        ctx.source.instance.runtime_properties[SELECTED_CONTAINER_DESTINATION] = selected_target
+        ctx.logger.info("Forwarding selected target: {0}".format(ctx.source.instance.id))
+    except Exception as e:
+        ctx.logger.error("Unexpected error while forwarding selected target: {0}"
+                .format(str(e)))
+        raise NonRecoverableError(e)
+
+@operation
+def registered_to(**kwargs):
+    """
+    Intended to be used in platform blueprints, but possible to be reused elsewhere
+    """
+    ctx.logger.info(str(kwargs))
+    address = kwargs["address_to_register"]
+    name = kwargs["name_to_register"]
+    port = kwargs["port_to_register"]
+
+    (consul_host, consul_port) = (CONSUL_HOSTNAME, 8500)
+    #Storing in source because that's who is getting registered
+    ctx.source.instance.runtime_properties[CONSUL_HOST] = "http://{0}:{1}".format(consul_host, consul_port)
+    ctx.source.instance.runtime_properties["name_to_register"] = name #careful! delete does not have access to inputs
+    
+    try:
+        response = requests.put(url = "{0}/v1/agent/service/register".format(ctx.source.instance.runtime_properties[CONSUL_HOST]),
+                     json = {
+                              "name" : name,
+                              "Address" : address,
+                              "Port" : int(port)
+                            },
+                     headers={'Content-Type': 'application/json'})
+        response.raise_for_status() #bomb if not 2xx
+    except Exception as e:
+        ctx.logger.error("Error while registering: {0}".format(str(e)))
+        raise NonRecoverableError(e)
+
+@operation
+def registered_to_delete(**kwargs):
+    """
+    The deletion/opposite of registered_to
+    """
+    requests.put(url = "{0}/v1/agent/service/deregister/{1}".format(ctx.source.instance.runtime_properties[CONSUL_HOST], ctx.source.instance.runtime_properties["name_to_register"]),
+                 headers={'Content-Type': 'Content-Type: application/json'})
+    #this is on delete so do not do any checking