From efc611652efc8802f951552d4183e3cb15875ce3 Mon Sep 17 00:00:00 2001
From: Zane Bitter <zbitter@redhat.com>
Date: Tue, 3 Jul 2012 18:56:22 +0200
Subject: [PATCH] Make LoadBalancer a subclass of a nested stack

This avoids duplicating the code for dealing with the nested stack
instance.

Change-Id: Iaeb96f6184cc39677be81bb5619c1fc3e3c9a2a2
Signed-off-by: Zane Bitter <zbitter@redhat.com>
---
 heat/engine/loadbalancer.py | 66 +++----------------------------------
 heat/engine/stack.py        | 14 +++++---
 2 files changed, 14 insertions(+), 66 deletions(-)

diff --git a/heat/engine/loadbalancer.py b/heat/engine/loadbalancer.py
index 4304dd68c4..d2e72d0b1b 100644
--- a/heat/engine/loadbalancer.py
+++ b/heat/engine/loadbalancer.py
@@ -18,7 +18,7 @@ import json
 import logging
 
 from heat.common import exception
-from heat.engine.resources import Resource
+from heat.engine import stack
 from heat.db import api as db_api
 from heat.engine import parser
 from novaclient.exceptions import NotFound
@@ -94,7 +94,7 @@ lb_template = '''
 # at the moment this is because we will probably need to implement a
 # LoadBalancer based on keepalived as well (for for ssl support).
 #
-class LoadBalancer(Resource):
+class LoadBalancer(stack.Stack):
 
     listeners_schema = {
         'InstancePort': {'Type': 'Integer',
@@ -141,32 +141,12 @@ class LoadBalancer(Resource):
                     'Implemented': False}
     }
 
-    def __init__(self, name, json_snippet, stack):
-        Resource.__init__(self, name, json_snippet, stack)
-        self._nested = None
-
     def _params(self):
         # total hack - probably need an admin key here.
         params = {'KeyName': {'Ref': 'KeyName'}}
         p = self.stack.resolve_static_data(params)
         return p
 
-    def nested(self):
-        if self._nested is None:
-            if self.instance_id is None:
-                return None
-
-            st = db_api.stack_get(self.stack.context, self.instance_id)
-            if not st:
-                raise exception.NotFound('Nested stack not found in DB')
-
-            n = parser.Stack(self.stack.context, st.name,
-                             st.raw_template.parsed_template.template,
-                             self.instance_id, self._params())
-            self._nested = n
-
-        return self._nested
-
     def _instance_to_ipaddress(self, inst):
         '''
         Return the server's IP address, fetching it from Nova
@@ -235,41 +215,7 @@ class LoadBalancer(Resource):
         cfg = self._haproxy_config(templ)
         files['/etc/haproxy/haproxy.cfg']['content'] = cfg
 
-        self._nested = parser.Stack(self.stack.context,
-                                    self.name,
-                                    templ,
-                                    parms=self._params(),
-                                    metadata_server=self.stack.metadata_server)
-
-        rt = {'template': templ, 'stack_name': self.name}
-        new_rt = db_api.raw_template_create(None, rt)
-
-        parent_stack = db_api.stack_get(self.stack.context, self.stack.id)
-
-        s = {'name': self.name,
-             'owner_id': self.stack.id,
-             'raw_template_id': new_rt.id,
-             'user_creds_id': parent_stack.user_creds_id,
-             'username': self.stack.context.username}
-        new_s = db_api.stack_create(None, s)
-        self._nested.id = new_s.id
-
-        pt = {'template': self._nested.t, 'raw_template_id': new_rt.id}
-        new_pt = db_api.parsed_template_create(None, pt)
-
-        self._nested.parsed_template_id = new_pt.id
-
-        self._nested.create()
-        self.instance_id_set(self._nested.id)
-
-    def handle_delete(self):
-        try:
-            stack = self.nested()
-        except exception.NotFound:
-            logger.info("Stack not found to delete")
-        else:
-            if stack is not None:
-                stack.delete()
+        self.create_with_template(templ)
 
     def FnGetAtt(self, key):
         '''
@@ -285,11 +231,7 @@ class LoadBalancer(Resource):
             raise exception.InvalidTemplateAttribute(resource=self.name,
                                                      key=key)
 
-        stack = self.nested()
-        if stack is None:
-            # This seems like a hack, to get past validation
-            return ''
         if key == 'DNSName':
-            return stack.output('PublicIp')
+            return stack.Stack.FnGetAtt(self, 'Outputs.PublicIp')
         else:
             return ''
diff --git a/heat/engine/stack.py b/heat/engine/stack.py
index c9474fe8bc..2b9ae4e894 100644
--- a/heat/engine/stack.py
+++ b/heat/engine/stack.py
@@ -60,10 +60,10 @@ class Stack(Resource):
 
         return self._nested
 
-    def handle_create(self):
-        response = urllib2.urlopen(self.properties[PROP_TEMPLATE_URL])
-        child_template = json.loads(response.read())
-
+    def create_with_template(self, child_template):
+        '''
+        Handle the creation of the nested stack from a given JSON template.
+        '''
         self._nested = parser.Stack(self.stack.context,
                                     self.name,
                                     child_template,
@@ -91,6 +91,12 @@ class Stack(Resource):
         self._nested.create()
         self.instance_id_set(self._nested.id)
 
+    def handle_create(self):
+        response = urllib2.urlopen(self.properties[PROP_TEMPLATE_URL])
+        template = json.load(response)
+
+        self.create_with_template(template)
+
     def handle_delete(self):
         try:
             stack = self.nested()