aboutsummaryrefslogtreecommitdiffstats
path: root/catalog-core/catalog-mgr/src/main/java/org/openo/commontosca/catalog/model/common/TemplateUtils.java
blob: 945cc92ac384417f830f46fdf45150dd0ff0cae9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
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 !");
      }
    }
  }
}