OpenConcerto

Dépôt officiel du code source de l'ERP OpenConcerto
sonarqube

svn://code.openconcerto.org/openconcerto

Rev

Rev 94 | Rev 177 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 94 Rev 156
1 
/*
 1 
/*
2 
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 2 
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
3 
 *
3 
 *
4 
 * Copyright 2011 OpenConcerto, by ILM Informatique. All rights reserved.
 4 
 * Copyright 2011 OpenConcerto, by ILM Informatique. All rights reserved.
5 
 *
5 
 *
6 
 * The contents of this file are subject to the terms of the GNU General Public License Version 3
 6 
 * The contents of this file are subject to the terms of the GNU General Public License Version 3
7 
 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a
 7 
 * only ("GPL"). You may not use this file except in compliance with the License. You can obtain a
8 
 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific
 8 
 * copy of the License at http://www.gnu.org/licenses/gpl-3.0.html See the License for the specific
9 
 * language governing permissions and limitations under the License.
 9 
 * language governing permissions and limitations under the License.
10 
 *
10 
 *
11 
 * When distributing the software, include this License Header Notice in each file.
 11 
 * When distributing the software, include this License Header Notice in each file.
12 
 */
 12 
 */
13
13
14 
 package org.openconcerto.utils;
 14 
 package org.openconcerto.utils;
15 
 
 15 
 
16 
import java.io.BufferedOutputStream;
 - 
 
17 
import java.io.BufferedWriter;
 16 
import java.io.BufferedWriter;
18 
import java.io.ByteArrayOutputStream;
 17 
import java.io.ByteArrayOutputStream;
19 
import java.io.File;
 18 
import java.io.File;
20 
import java.io.FileOutputStream;
 - 
 
21 
import java.io.IOException;
 19 
import java.io.IOException;
22 
import java.io.InputStream;
 20 
import java.io.InputStream;
23 
import java.io.OutputStream;
 21 
import java.io.OutputStream;
24 
import java.io.OutputStreamWriter;
 22 
import java.io.OutputStreamWriter;
25 
import java.io.Writer;
 23 
import java.io.Writer;
26 
import java.nio.charset.Charset;
 24 
import java.nio.charset.Charset;
- 
 
 25 
import java.nio.file.Files;
- 
 
 26 
import java.nio.file.StandardCopyOption;
- 
 
 27 
import java.util.logging.Level;
27 
 
 28 
 
28 
public class StreamUtils {
 29 
public class StreamUtils {
29 
 
 30 
 
30 
    public static final OutputStream NULL_OS = new OutputStream() {
 31 
    public static final OutputStream NULL_OS = new OutputStream() {
31 
        @Override
 32 
        @Override
32 
        public void write(int b) throws IOException {
 33 
        public void write(int b) throws IOException {
33 
            // ignore
 34 
            // ignore
34 
        }
 35 
        }
35 
 
 36 
 
36 
        public void write(byte b[], int off, int len) throws IOException {
 37 
        public void write(byte b[], int off, int len) throws IOException {
37 
            if (b == null)
 38 
            if (b == null)
38 
                throw new NullPointerException();
 39 
                throw new NullPointerException();
39 
            // ignore
 40 
            // ignore
40 
        }
 41 
        }
41 
    };
 42 
    };
42 
 
 43 
 
43 
    /**
 44 
    /**
44 
     * Verbatim copy an entry from input to output stream.
 45 
     * Verbatim copy an entry from input to output stream.
45 
     *
46 
     *
46 
     * @param in the source.
 47 
     * @param in the source.
47 
     * @param out the destination.
 48 
     * @param out the destination.
48 
     * @throws IOException if an error occurs while reading or writing.
 49 
     * @throws IOException if an error occurs while reading or writing.
49 
     */
 50 
     */
50 
    public static void copy(InputStream in, OutputStream out) throws IOException {
 51 
    public static void copy(InputStream in, OutputStream out) throws IOException {
- 
 
 52 
        // TODO use in.transferTo(out) in Java 9
51 
        copy(in, out, 512 * 1024);
 53 
        copy(in, out, 512 * 1024);
52 
    }
 54 
    }
53 
 
 55 
 
54 
    public static void copy(InputStream in, OutputStream out, final int bufferSize) throws IOException {
 56 
    public static void copy(InputStream in, OutputStream out, final int bufferSize) throws IOException {
55 
        copy(in, out, bufferSize, -1);
 57 
        copy(in, out, bufferSize, -1);
56 
    }
 58 
    }
57 
 
 59 
 
58 
    public static long copy(InputStream in, OutputStream out, final int bufferSize, final long length) throws IOException {
 60 
    public static long copy(InputStream in, OutputStream out, final int bufferSize, final long length) throws IOException {
- 
 
 61 
        if (bufferSize < 1)
- 
 
 62 
            throw new IllegalArgumentException("Buffer size too small : " + bufferSize);
59 
        final byte[] buffer = new byte[bufferSize];
 63 
        final byte[] buffer = new byte[bufferSize];
60 
        long totalCount = 0;
 64 
        long totalCount = 0;
61 
        final boolean copyAll = length < 0;
 65 
        final boolean copyAll = length < 0;
62 
        while (copyAll || totalCount < length) {
 66 
        while (copyAll || totalCount < length) {
63 
            final long toRead = copyAll ? buffer.length : Math.min(length - totalCount, buffer.length);
 67 
            final long toRead = copyAll ? buffer.length : Math.min(length - totalCount, buffer.length);
64 
            // since buffer.length is an int
 68 
            // since buffer.length is an int
65 
            assert 0 <= toRead && toRead <= Integer.MAX_VALUE;
 69 
            assert 0 < toRead && toRead <= Integer.MAX_VALUE;
66 
            final int count = in.read(buffer, 0, (int) toRead);
 70 
            final int count = in.read(buffer, 0, (int) toRead);
67 
            if (count == -1)
 71 
            if (count <= 0) {
- 
 
 72 
                // like Files.copy(InputStream, OutputStream), stop if reading 0 bytes
- 
 
 73 
                if (count == 0)
- 
 
 74 
                    Log.get().log(Level.WARNING, "", new IllegalStateException("read() returned 0 for " + in));
68 
                break;
 75 
                break;
- 
 
 76 
            }
69 
            totalCount += count;
 77 
            totalCount += count;
70 
            out.write(buffer, 0, count);
 78 
            out.write(buffer, 0, count);
71 
        }
 79 
        }
72 
        // < if end of stream
 80 
        // < if end of stream
73 
        assert copyAll || totalCount <= length;
 81 
        assert copyAll || totalCount <= length;
74 
        return totalCount;
 82 
        return totalCount;
75 
    }
 83 
    }
76 
 
 84 
 
77 
    public static void copy(InputStream ins, File out) throws IOException {
 85 
    public static void copy(InputStream ins, File out) throws IOException {
78 
        // buffered since read() in copy(InputStream, OutputStream) may return 1 byte at a time
 - 
 
79 
        final OutputStream ous = new BufferedOutputStream(new FileOutputStream(out));
 86 
        Files.copy(ins, out.toPath(), StandardCopyOption.REPLACE_EXISTING);
80 
        try {
 - 
 
81 
            copy(ins, ous);
 - 
 
82 
        } finally {
 - 
 
83 
            ous.close();
 - 
 
84 
        }
 - 
 
85 
    }
 87 
    }
86 
 
 88 
 
87 
    /**
 89 
    /**
88 
     * Read until the end of the stream is reached. NOTE : since this method didn't create the
 90 
     * Read until the end of the stream is reached. NOTE : since this method didn't create the
89 
     * stream, it doesn't close it.
 91 
     * stream, it doesn't close it.
90 
     *
92 
     *
91 
     * @param ins the stream to read from.
 93 
     * @param ins the stream to read from.
92 
     * @return the bytes.
 94 
     * @return the bytes.
93 
     * @throws IOException if an error occurs.
 95 
     * @throws IOException if an error occurs.
94 
     */
 96 
     */
95 
    public static byte[] read(final InputStream ins) throws IOException {
 97 
    public static byte[] read(final InputStream ins) throws IOException {
96 
        final ByteArrayOutputStream out = new ByteArrayOutputStream(512);
 98 
        final ByteArrayOutputStream out = new ByteArrayOutputStream(512);
97 
        copy(ins, out);
 99 
        copy(ins, out);
98 
        return out.toByteArray();
 100 
        return out.toByteArray();
99 
    }
 101 
    }
100 
 
 102 
 
101 
    public static void writeln(final String s, final OutputStream out) throws IOException {
 103 
    public static void writeln(final String s, final OutputStream out) throws IOException {
102 
        write(s + "\n", out);
 104 
        write(s + "\n", out);
103 
    }
 105 
    }
104 
 
 106 
 
105 
    public static void write(final String s, final OutputStream out) throws IOException {
 107 
    public static void write(final String s, final OutputStream out) throws IOException {
106 
        write(s, out, StringUtils.UTF8);
 108 
        write(s, out, StringUtils.UTF8);
107 
    }
 109 
    }
108 
 
 110 
 
109 
    public static void write(final String s, final OutputStream out, Charset charset) throws IOException {
 111 
    public static void write(final String s, final OutputStream out, Charset charset) throws IOException {
110 
        out.write(s.getBytes(charset));
 112 
        out.write(s.getBytes(charset));
111 
    }
 113 
    }
112 
 
 114 
 
113 
    /**
 115 
    /**
114 
     * Wrap the output stream into a writer, and write the XML declaration.
 116 
     * Wrap the output stream into a writer, and write the XML declaration.
115 
     *
117 
     *
116 
     * @param outs an output stream.
 118 
     * @param outs an output stream.
117 
     * @return a writer with the same encoding as the XML.
 119 
     * @return a writer with the same encoding as the XML.
118 
     * @throws IOException if an error occurs.
 120 
     * @throws IOException if an error occurs.
119 
     */
 121 
     */
120 
    public static BufferedWriter createXMLWriter(OutputStream outs) throws IOException {
 122 
    public static BufferedWriter createXMLWriter(OutputStream outs) throws IOException {
121 
        return new BufferedWriter(createXMLUnbufferedWriter(outs));
 123 
        return new BufferedWriter(createXMLUnbufferedWriter(outs));
122 
    }
 124 
    }
123 
 
 125 
 
124 
    public static Writer createXMLUnbufferedWriter(OutputStream outs) throws IOException {
 126 
    public static Writer createXMLUnbufferedWriter(OutputStream outs) throws IOException {
125 
        // see http://www.w3.org/TR/REC-xml/#sec-guessing
 127 
        // see http://www.w3.org/TR/REC-xml/#sec-guessing
126 
        // don't use UTF-8 BOM as Java does not support it :
 128 
        // don't use UTF-8 BOM as Java does not support it :
127 
        // http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=4508058
 129 
        // http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=4508058
128 
        final Writer res = new OutputStreamWriter(outs, StringUtils.UTF8);
 130 
        final Writer res = new OutputStreamWriter(outs, StringUtils.UTF8);
129 
        res.write("<?xml version='1.0' encoding='UTF-8' ?>\n");
 131 
        res.write("<?xml version='1.0' encoding='UTF-8' ?>\n");
130 
        return res;
 132 
        return res;
131 
    }
 133 
    }
132 
}
 134 
}