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 !");
}
}
}
}
|