aboutsummaryrefslogtreecommitdiffstats
path: root/cdf/src/cdf-prop-value/cdf-util/src/main/java/org/onap/dcae/cdf/util/common/Compress.java
blob: 86ff60b66d6e4f5c8f002594dd4ffaef5899ffc6 (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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/*
    Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.  
 
    Licensed under the Apache License, Version 2.0 (the "License"); 
    you may not use this code 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.onap.dcae.cdf.util.common;

import java.util.zip.GZIPOutputStream;
import java.util.zip.ZipOutputStream;
import java.util.zip.ZipEntry;
// import java.io.InputStream;
import java.io.FileOutputStream;
import java.io.FileInputStream;
import java.io.File;
import java.io.IOException;

public class Compress {

    /**
     * Compress a file with the gzip algorithm, sending output to outFilename.
     * Based on code at http://www.java-tips.org/java-se-tips/java.util.zip/how-to-compress-a-file-in-the-gip-format.html.
     */
    public static void gzip(String inFilename, String outFilename) throws IOException {
	String tmpFilename = outFilename + ".tmp";
	try {
	    // Create the GZIP output stream
	    GZIPOutputStream out = new GZIPOutputStream(new FileOutputStream(tmpFilename));

	    // Open the input file
	    FileInputStream in = new FileInputStream(inFilename);

	    // Transfer bytes from the input file to the GZIP output stream
	    byte[] buf = new byte[4096];
	    int len;
	    while ((len = in.read(buf)) > 0) {
		out.write(buf, 0, len);
	    }
	    in.close();

	    // Complete the GZIP file
	    out.finish();
	    out.close();
	    
	    // rename .gz.tmp to .gz
	    File target = new File(outFilename);
	    if (target.exists()) target.delete();
	    File file = new File(tmpFilename);
	    boolean result = file.renameTo(target);
	    if (!result) throw new IOException("Cannot rename " + tmpFilename + " to " + outFilename);
	} catch (IOException e) {
	    // If we can't write the gzip file, remove it and pass on the exception.
	    File f = new File(outFilename);
	    f.delete();
	    throw e;
	}
    }

    /**
     * Compress a file with the gzip algorithm, sending output to filename+".gz".
     */
    public static void gzip(String filename) throws IOException {
        gzip(filename, filename + ".gz");
    }

    /**
     * Compress a file with the zip algorithm, sending output to outFilename
     * Based on code at http://www.java-tips.org/java-se-tips/java.util.zip/how-to-compress-a-file-in-the-gip-format.html.
     */
    public static void zip(String inFilename, String outFilename) throws IOException {
	String tmpFilename = outFilename + ".tmp";
	try {
	    // Create the ZIP output stream
	    ZipOutputStream out = new ZipOutputStream(new FileOutputStream(tmpFilename));
	    ZipEntry zipEntry = new ZipEntry(inFilename);
	    out.putNextEntry(zipEntry);

	    // Open the input file
	    FileInputStream in = new FileInputStream(inFilename);

	    // Transfer bytes from the input file to the ZIP output stream
	    byte[] buf = new byte[4096];
	    int len;
	    while ((len = in.read(buf)) > 0) {
		out.write(buf, 0, len);
	    }
	    in.close();

	    // Complete the ZIP file
	    out.finish();
	    out.close();
	    
	    // rename .zip.tmp to .zip
	    File target = new File(outFilename);
	    if (target.exists()) target.delete();
	    File file = new File(tmpFilename);
	    boolean result = file.renameTo(target);
	    if (!result) throw new IOException("Cannot rename " + tmpFilename + " to " + outFilename);
	} catch (IOException e) {
	    // If we can't write the zip file, remove it and pass on the exception.
	    File f = new File(outFilename);
	    f.delete();
	    throw e;
	}
    }

    /**
     * Compress a file with the gzip algorithm, sending output to filename+".zip".
     */
    public static void zip(String filename) throws IOException {
        zip(filename, filename + ".zip");
    }

    public static void main(String args[]) throws Exception {
	if (args.length == 1) {
	    gzip(args[0]);
	    zip(args[0]);
	} else {
	    System.err.println("Usage: java Compress filename");
	}
    }
}