From 5fe99752927f4f2fb460ecb5cd651089770d811f Mon Sep 17 00:00:00 2001
From: Matt Gilman <matt.c.gilman@gmail.com>
Date: Fri, 19 Jun 2015 09:57:30 -0400
Subject: [PATCH] NIFI-701: - Including controller services in downloaded
 templates.

---
 .../main/java/org/apache/nifi/web/api/dto/DtoFactory.java    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-api/src/main/java/org/apache/nifi/web/api/dto/DtoFactory.java b/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-api/src/main/java/org/apache/nifi/web/api/dto/DtoFactory.java
index 999a4a47be..54f11fbde2 100644
--- a/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-api/src/main/java/org/apache/nifi/web/api/dto/DtoFactory.java
+++ b/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-api/src/main/java/org/apache/nifi/web/api/dto/DtoFactory.java
@@ -797,6 +797,11 @@ public final class DtoFactory {
                 copySnippet.getRemoteProcessGroups().add(copy(remoteGroup));
             }
         }
+        if (originalSnippet.getControllerServices() != null) {
+            for (final ControllerServiceDTO controllerService : originalSnippet.getControllerServices()) {
+                copySnippet.getControllerServices().add(copy(controllerService));
+            }
+        }
 
         return copySnippet;
     }