adressevedligehold/coveragefileupload/AbstractAddressSource.java

package dk.daoas.adressevedligehold.fileupload;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.util.List;

import org.apache.commons.fileupload.FileItem;

import com.google.common.base.Splitter;

import dk.daoas.adressevedligehold.tasks.TaskLogger;
import dk.daoas.adressevedligehold.util.DeduplicateHelper;

public abstract class AbstractAddressSource implements AddressSource {
        
        private TaskLogger logger = TaskLogger.getInstance();
        
        protected DeduplicateHelper<String> dirigeringsCache = new DeduplicateHelper<String>();
        
        protected FileItem file;
        
        protected InputStream is;
        protected InputStreamReader isr;
        protected BufferedReader br;
        
        protected int lineCount = 0;

        public AbstractAddressSource(FileItem file) {
                this.file = file;
        }
        
        @Override
        public String getFilename() {
                return file.getName();
        }
        
        @Override //AutoCloseable
        public void close() throws Exception {
                logger.info("Closing BK after lines " + lineCount);
                try {
                        if (br != null)
                                br.close();
                        if (isr != null)
                                isr.close();
                        if (is != null)
                                is.close();

                        file.delete();
                        
                } catch (Exception e) {
                        logger.warning("Error on closing ", e );
                }       
        }
        /**
         * Reads the first line, validates that there are the expected number of fields, and then
         * re-initilze the inputstream - this is for file formats that does not use a header line.
         * 
         * @param exptectedFieldCount
         * @param seperator
         * @throws IOException
         */
        protected void validatNoHeaderLine(int exptectedFieldCount, char seperator) throws IOException {
                try (           
                                InputStream is1 = file.getInputStream();
                                InputStreamReader isr1 = new InputStreamReader(is1, Charset.forName("ISO-8859-1") );
                                BufferedReader br1 = new BufferedReader(isr1)
                        ) {
                        String line = br1.readLine();
                        
                        if (line == null) {
                                throw new IOException("Can't read 1st line - is file empty?");
                        }       
                        
                        List<String> parts = Splitter.on(seperator).splitToList(line);
                        if (parts.size() != exptectedFieldCount) {
                                throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + exptectedFieldCount);
                        }
                }

                
                is = file.getInputStream();
                isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
                br = new BufferedReader(isr);           
        }
        
        
        protected void validateWithHeader(int expectedFieldCount, char seperator) throws IOException {
                try {
                        is = file.getInputStream();
                        isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
                        br = new BufferedReader(isr);
                        
                        String line = br.readLine();
                        if (line == null) {
                                throw new IOException("Can't read 1st line - is file empty?");
                        }                       
                        
                        List<String> parts = Splitter.on(seperator).splitToList(line);
                        if (parts.size() != expectedFieldCount) {
                                throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + expectedFieldCount);
                        }


                } catch (Exception e) {
                        try {
                                br.close();
                                isr.close();
                                is.close();
                        } catch (Exception e2) {
                                logger.warning("Error cleaning up resources", e2);
                        }
                        
                        throw e; // Re-throw     
                }               
        }
}
1	torben	2947	package dk.daoas.adressevedligehold.fileupload;
2	torben	2868
3			import java.io.BufferedReader;
4	torben	2869	import java.io.IOException;
5	torben	2868	import java.io.InputStream;
6			import java.io.InputStreamReader;
7	torben	2869	import java.nio.charset.Charset;
8			import java.util.List;
9	torben	2868
10			import org.apache.commons.fileupload.FileItem;
11
12	torben	2869	import com.google.common.base.Splitter;
13
14	torben	2904	import dk.daoas.adressevedligehold.tasks.TaskLogger;
15	torben	2868	import dk.daoas.adressevedligehold.util.DeduplicateHelper;
16
17			public abstract class AbstractAddressSource implements AddressSource {
18
19	torben	2904	private TaskLogger logger = TaskLogger.getInstance();
20
21	torben	2868	protected DeduplicateHelper<String> dirigeringsCache = new DeduplicateHelper<String>();
22
23			protected FileItem file;
24
25			protected InputStream is;
26			protected InputStreamReader isr;
27			protected BufferedReader br;
28
29			protected int lineCount = 0;
30
31			public AbstractAddressSource(FileItem file) {
32			this.file = file;
33			}
34
35			@Override
36			public String getFilename() {
37			return file.getName();
38			}
39
40			@Override //AutoCloseable
41			public void close() throws Exception {
42	torben	2904	logger.info("Closing BK after lines " + lineCount);
43	torben	2868	try {
44			if (br != null)
45			br.close();
46			if (isr != null)
47			isr.close();
48			if (is != null)
49			is.close();
50
51			file.delete();
52
53			} catch (Exception e) {
54	torben	2904	logger.warning("Error on closing ", e );
55	torben	2868	}
56			}
57	torben	2870	/**
58			* Reads the first line, validates that there are the expected number of fields, and then
59			* re-initilze the inputstream - this is for file formats that does not use a header line.
60			*
61			* @param exptectedFieldCount
62			* @param seperator
63			* @throws IOException
64			*/
65	torben	2869	protected void validatNoHeaderLine(int exptectedFieldCount, char seperator) throws IOException {
66			try (
67			InputStream is1 = file.getInputStream();
68			InputStreamReader isr1 = new InputStreamReader(is1, Charset.forName("ISO-8859-1") );
69			BufferedReader br1 = new BufferedReader(isr1)
70			) {
71			String line = br1.readLine();
72
73			if (line == null) {
74			throw new IOException("Can't read 1st line - is file empty?");
75			}
76
77			List<String> parts = Splitter.on(seperator).splitToList(line);
78			if (parts.size() != exptectedFieldCount) {
79			throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + exptectedFieldCount);
80			}
81			}
82
83
84			is = file.getInputStream();
85			isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
86			br = new BufferedReader(isr);
87			}
88
89	torben	2870
90	torben	2869	protected void validateWithHeader(int expectedFieldCount, char seperator) throws IOException {
91			try {
92			is = file.getInputStream();
93			isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
94			br = new BufferedReader(isr);
95
96			String line = br.readLine();
97			if (line == null) {
98			throw new IOException("Can't read 1st line - is file empty?");
99			}
100
101			List<String> parts = Splitter.on(seperator).splitToList(line);
102			if (parts.size() != expectedFieldCount) {
103			throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + expectedFieldCount);
104			}
105
106
107
108			} catch (Exception e) {
109			try {
110			br.close();
111			isr.close();
112			is.close();
113			} catch (Exception e2) {
114	torben	2904	logger.warning("Error cleaning up resources", e2);
115	torben	2869	}
116
117			throw e; // Re-throw
118			}
119			}
120	torben	2868	}