daoas/adressevedligehold/AbstractAddressSource.java

package dk.daoas.adressevedligehold;

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.util.DeduplicateHelper;

public abstract class AbstractAddressSource implements AddressSource {
        
        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() {
                // TODO Auto-generated method stub
                return file.getName();
        }
        
        @Override //AutoCloseable
        public void close() throws Exception {
                System.out.println("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) {
                        System.out.println("Error on closing " + e.getMessage() );
                }       
        }
        
        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) {
                                System.out.println("Error cleaning up resources");
                        }
                        
                        throw e; // Re-throw     
                }               
        }
}
1	package dk.daoas.adressevedligehold;
2
3	import java.io.BufferedReader;
4	import java.io.IOException;
5	import java.io.InputStream;
6	import java.io.InputStreamReader;
7	import java.nio.charset.Charset;
8	import java.util.List;
9
10	import org.apache.commons.fileupload.FileItem;
11
12	import com.google.common.base.Splitter;
13
14	import dk.daoas.adressevedligehold.util.DeduplicateHelper;
15
16	public abstract class AbstractAddressSource implements AddressSource {
17
18	protected DeduplicateHelper<String> dirigeringsCache = new DeduplicateHelper<String>();
19
20	protected FileItem file;
21
22	protected InputStream is;
23	protected InputStreamReader isr;
24	protected BufferedReader br;
25
26	protected int lineCount = 0;
27
28	public AbstractAddressSource(FileItem file) {
29	this.file = file;
30	}
31
32	@Override
33	public String getFilename() {
34	// TODO Auto-generated method stub
35	return file.getName();
36	}
37
38	@Override //AutoCloseable
39	public void close() throws Exception {
40	System.out.println("Closing BK after lines " + lineCount);
41	try {
42	if (br != null)
43	br.close();
44	if (isr != null)
45	isr.close();
46	if (is != null)
47	is.close();
48
49	file.delete();
50
51	} catch (Exception e) {
52	System.out.println("Error on closing " + e.getMessage() );
53	}
54	}
55
56	protected void validatNoHeaderLine(int exptectedFieldCount, char seperator) throws IOException {
57	try (
58	InputStream is1 = file.getInputStream();
59	InputStreamReader isr1 = new InputStreamReader(is1, Charset.forName("ISO-8859-1") );
60	BufferedReader br1 = new BufferedReader(isr1)
61	) {
62	String line = br1.readLine();
63
64	if (line == null) {
65	throw new IOException("Can't read 1st line - is file empty?");
66	}
67
68	List<String> parts = Splitter.on(seperator).splitToList(line);
69	if (parts.size() != exptectedFieldCount) {
70	throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + exptectedFieldCount);
71	}
72	}
73
74
75	is = file.getInputStream();
76	isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
77	br = new BufferedReader(isr);
78	}
79
80	protected void validateWithHeader(int expectedFieldCount, char seperator) throws IOException {
81	try {
82	is = file.getInputStream();
83	isr = new InputStreamReader(is, Charset.forName("ISO-8859-1") );
84	br = new BufferedReader(isr);
85
86	String line = br.readLine();
87	if (line == null) {
88	throw new IOException("Can't read 1st line - is file empty?");
89	}
90
91	List<String> parts = Splitter.on(seperator).splitToList(line);
92	if (parts.size() != expectedFieldCount) {
93	throw new IOException("Not enough fields in CSV file. Found " + parts.size() + ", expected " + expectedFieldCount);
94	}
95
96
97
98	} catch (Exception e) {
99	try {
100	br.close();
101	isr.close();
102	is.close();
103	} catch (Exception e2) {
104	System.out.println("Error cleaning up resources");
105	}
106
107	throw e; // Re-throw
108	}
109	}
110	}