1 files changed, 105 insertions, 0 deletions
diff --git a/catalog-core/catalog-mgr/src/main/java/org/openo/commontosca/catalog/model/common/TemplateUtils.java b/catalog-core/catalog-mgr/src/main/java/org/openo/commontosca/catalog/model/common/TemplateUtils.java
new file mode 100644
index 00000000..945cc92a
--- /dev/null
+++ b/catalog-core/catalog-mgr/src/main/java/org/openo/commontosca/catalog/model/common/TemplateUtils.java
@@ -0,0 +1,105 @@
+/**
+ * Copyright 2016 ZTE Corporation.
+ *
+ * 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.
+ */
+package org.openo.commontosca.catalog.model.common;
+
+import org.openo.commontosca.catalog.db.exception.CatalogResourceException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.BufferedInputStream;
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipFile;
+import java.util.zip.ZipInputStream;
+
+public class TemplateUtils {
+  private static final Logger logger = LoggerFactory.getLogger(TemplateUtils.class);
+  
+  /**
+   * @param zipFileName
+   * @param zipEntryName
+   * @return
+   * @throws CatalogResourceException
+   */
+  public static String readStringFromZipFile(String zipFileName, String zipEntryName) throws CatalogResourceException {
+    String[] lines = readFromZipFile(zipFileName, zipEntryName);
+    StringBuffer sb = new StringBuffer();
+    for (String line : lines) {
+      sb.append(line).append(System.lineSeparator());
+    }
+    return sb.toString();
+  }
+  
+  /**
+   * @param zipFileName
+   * @param zipEntryName
+   * @return
+   * @throws CatalogResourceException
+   */
+  @SuppressWarnings("resource")
+  public static String[] readFromZipFile(String zipFileName, String zipEntryName)
+      throws CatalogResourceException {
+    ZipInputStream zipIns = null;
+    BufferedReader zipEntryBr = null;
+    try {
+      ZipFile zipFile = new ZipFile(zipFileName);
+      
+      zipIns = new ZipInputStream(new BufferedInputStream(new FileInputStream(zipFileName)));
+      ZipEntry zipEntry;
+      while ((zipEntry = zipIns.getNextEntry()) != null) {
+        if (zipEntryName.equals(zipEntry.getName())
+            || (zipEntryName.replaceAll("/", File.separator)).equals(zipEntry.getName())) {
+          zipEntryBr = new BufferedReader(new InputStreamReader(zipFile.getInputStream(zipEntry)));
+          List<String> lineList = new ArrayList<>();
+          String line;
+          while ((line = zipEntryBr.readLine()) != null) {
+            lineList.add(line);
+          }
+          
+          return lineList.toArray(new String[0]);
+        }
+      }
+    } catch (IOException e) {
+      throw new CatalogResourceException("Parse Tosca Meta Fail.", e);
+    } finally {
+      closeStreamAndReader(zipIns, zipEntryBr);
+    }
+    return new String[0];
+  }
+  
+  private static void closeStreamAndReader(ZipInputStream zin, BufferedReader br) {
+    if (br != null) {
+      try {
+        br.close();
+      } catch (IOException e1) {
+        logger.error("Buffered reader close failed !");
+      }
+    }
+    if (zin != null) {
+      try {
+        zin.closeEntry();
+      } catch (IOException e2) {
+        logger.error("Zip inputStream close failed !");
+      }
+    }
+  }
+}