thoerup/traininfoservice/RequestPlotter.java

package dk.thoerup.traininfoservice;

import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import dk.thoerup.traininfoservice.banedk.TimeoutMap;

public class RequestPlotter extends HttpServlet {
        private static final long serialVersionUID = 1L;

        static final Logger log = Logger.getLogger(RequestPlotter.class.getName());

        static final String KML = "application/vnd.google-earth.kml";
        static final String KMZ = "application/vnd.google-earth.kmz";
        
        Map<String,String> cache = new TimeoutMap<String,String>(2*60*1000);

        class RequestPosition {
                public String ip;
                public Date time;
                public String lat;
                public String lng;
        }


        boolean isGz(String fileStr) {
                return fileStr.substring(fileStr.length() - 3).equals(".gz");
        }

        protected List<RequestPosition> getRequestsFromFileWorker(boolean multiple) throws IOException{
                List<RequestPosition> positions = new ArrayList<RequestPosition>();

                try {
                        String files_single[] = {"/var/log/apache2/app_access.log"};
                        String files_multi[] = {"/var/log/apache2/app_access.log.2.gz", "/var/log/apache2/app_access.log.1", "/var/log/apache2/app_access.log"};
                        
                        String files[];
                        
                        if (multiple == false) {
                                files = files_single;
                        } else {
                                files = files_multi;
                        }

                        SimpleDateFormat df = new SimpleDateFormat("dd/MMM/yyyy:HH:mm:ss");
                        
                        for (String fileStr : files ) {
                                File f = new File(fileStr);
                                if ( !f.exists() ) {
                                        continue;
                                }
                                
                                FileInputStream fis = new FileInputStream(fileStr);
                                
                                InputStream input;
                                if ( isGz(fileStr)) {
                                        input = new GZIPInputStream(fis);
                                } else {
                                        input = fis;
                                }
                                
                                BufferedReader in = new BufferedReader( new InputStreamReader(input) );

        
                                String line;
                                while ( (line=in.readLine()) != null) {
                                        if (line.indexOf("LocateStation") == -1 ){
                                                continue;
                                        }
        
                                        if (line.indexOf("latitude") == -1 ) {
                                                continue;
                                        }
                                        RequestPosition pos = new RequestPosition();
        
                                        String toks[] = line.split(" ");
                                        pos.ip = toks[0];
        
                                        pos.time = df.parse( toks[3].replace("[", "") );
        
                                        String argpart = toks[6].split("\\?")[1];
        
                                        String args[] = argpart.split("&");
        
                                        pos.lat = args[0].split("=")[1];
                                        pos.lng = args[1].split("=")[1];
        
                                        positions.add(pos);
                                }
                                in.close();
                                input.close();
                                fis.close();
                        }
                } catch (ParseException pe) {
                        log.log(Level.SEVERE, "parseException", pe);
                        throw new IOException(pe);
                } catch (IOException e) {
                        log.log(Level.SEVERE, "getKml()", e);
                        throw e;
                }

                return positions;
        }

        protected String formatXml(List<RequestPosition> list) {
                StringBuilder sb = new StringBuilder();

                sb.append( "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" );
                sb.append( "<kml xmlns=\"http://www.opengis.net/kml/2.2\">\n" );
                sb.append( "<Document>\n" );            
                
                
                sb.append( " <Style id=\"red\">\n" );
                sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/red-circle.png</href></Icon></IconStyle>\n" );
                sb.append( " </Style>\n");

                sb.append( " <Style id=\"yellow\">\n" );
                sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/ylw-circle.png</href></Icon></IconStyle>\n" );
                sb.append( " </Style>\n" );

                sb.append( " <Style id=\"green\">\n" );
                sb.append( "  <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/grn-circle.png</href></Icon></IconStyle>\n" );
                sb.append( " </Style>\n\n" );
                
/*
                String overlay = 
              " <ScreenOverlay>" +
                " <name>Absolute Positioning: Top left</name>" +
                " <description>Absolute Positioning: Top left</description>" +
                " <visibility>1</visibility>" +
                "  <Icon>" +
                "   <href>http://code.google.com/apis/kml/documentation/top_left.jpg</href>" +
                "  </Icon>" +
                "  <overlayXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
                "  <screenXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
                "  <rotationXY x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
                "  <size x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
              " </ScreenOverlay>" ;
                
                sb.append(overlay);
        */      
                

                final String STYLE_GREEN = "green";
                final String STYLE_YELLOW = "yellow";
                final String STYLE_RED = "red";

                String oldstyle = null;
                Date now = new Date();
                for(RequestPosition current : list) {
                        
                        String style;
                        long timediff = now.getTime() - current.time.getTime();
                        
                        if ( timediff < (3*60*60*1000) ) {
                                style = STYLE_RED;
                        } else if ( timediff < (24*60*60*1000)) {
                                style = STYLE_YELLOW;
                        } else {
                                style = STYLE_GREEN;
                        }
                        
                        if (  !style.equals(oldstyle) ) {
                                if (oldstyle != null)                           
                                        sb.append( "</Folder>\n");
                                sb.append( "<Folder>\n");
                                sb.append( " <name>" ).append(style).append("</name>\n");
                                sb.append( " <open>0</open>\n" );
                                
                                oldstyle = style;
                        }
                        
                        sb.append( " <Placemark>\n" );
                        sb.append( "  <styleUrl>#" + style + "</styleUrl>\n" );
                        sb.append( "  <description>IP=" + current.ip + "  Time=" + current.time + "</description>\n" );
                        sb.append( "  <Point><coordinates>" + current.lng + "," + current.lat + ",0</coordinates></Point>\n" );
                        sb.append( " </Placemark>\n" );                 
                }
                sb.append( "</Folder>\n" );
                sb.append( "</Document>\n" );
                sb.append( "</kml>\n" );

                return sb.toString(); 
        }
        
        protected String getRequestsFromFile(boolean multiple) throws IOException {             
                String kmlData = null;
                String key;
                
                if (multiple == false)  {
                        key = "kmldata";
                } else {
                        key = "kmldata-multi";
                }
                                        
                kmlData = cache.get(key);
                        
                if (kmlData == null) {
                        kmlData = formatXml( getRequestsFromFileWorker(multiple) );
                        cache.put(key, kmlData);
                        kmlData += "<!-- from source -->";                      
                } else {
                        kmlData += "<!-- cached -->";
                }
                                        
                return kmlData;
        }
        
        boolean enabled(String param) {
                if (param == null || param.equals("")) {
                        return false;
                }
                
                int p = 0;
                try { 
                        p = Integer.parseInt(param);
                } catch (Exception e) {}
                
                return (p != 0);
        }

        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
                
                boolean multiple = enabled( req.getParameter("multi") );
                
                String kmlData = getRequestsFromFile(multiple);
                
                if ( enabled(req.getParameter("zip")) ) {

                        ByteArrayOutputStream baos = new ByteArrayOutputStream();

                        ZipOutputStream zip = new ZipOutputStream(baos);
                        zip.putNextEntry( new ZipEntry("trains.kml") );
                        zip.write( kmlData.getBytes() );
                        zip.closeEntry();
                        zip.close();

                        byte bytes[] = baos.toByteArray();

                        resp.setContentType(KMZ);
                        resp.setContentLength( bytes.length );
                        resp.getOutputStream().write(bytes);

                } else {
                        resp.setContentType(KML);
                        resp.getWriter().print( kmlData );
                }
        }
}
1	torben	795	package dk.thoerup.traininfoservice;
2
3			import java.io.BufferedReader;
4	torben	798	import java.io.ByteArrayOutputStream;
5	torben	809	import java.io.File;
6	torben	795	import java.io.FileInputStream;
7			import java.io.IOException;
8	torben	809	import java.io.InputStream;
9	torben	795	import java.io.InputStreamReader;
10	torben	800	import java.text.ParseException;
11			import java.text.SimpleDateFormat;
12	torben	799	import java.util.ArrayList;
13	torben	800	import java.util.Date;
14	torben	799	import java.util.List;
15	torben	802	import java.util.Map;
16	torben	796	import java.util.logging.Level;
17			import java.util.logging.Logger;
18	torben	809	import java.util.zip.GZIPInputStream;
19	torben	798	import java.util.zip.ZipEntry;
20			import java.util.zip.ZipOutputStream;
21	torben	795
22			import javax.servlet.ServletException;
23			import javax.servlet.http.HttpServlet;
24			import javax.servlet.http.HttpServletRequest;
25			import javax.servlet.http.HttpServletResponse;
26
27	torben	802	import dk.thoerup.traininfoservice.banedk.TimeoutMap;
28
29	torben	795	public class RequestPlotter extends HttpServlet {
30			private static final long serialVersionUID = 1L;
31	torben	799
32	torben	796	static final Logger log = Logger.getLogger(RequestPlotter.class.getName());
33	torben	799
34	torben	798	static final String KML = "application/vnd.google-earth.kml";
35			static final String KMZ = "application/vnd.google-earth.kmz";
36	torben	802
37			Map<String,String> cache = new TimeoutMap<String,String>(2601000);
38	torben	799
39			class RequestPosition {
40			public String ip;
41	torben	800	public Date time;
42	torben	799	public String lat;
43			public String lng;
44			}
45
46
47	torben	809	boolean isGz(String fileStr) {
48			return fileStr.substring(fileStr.length() - 3).equals(".gz");
49			}
50	torben	799
51	torben	809	protected List<RequestPosition> getRequestsFromFileWorker(boolean multiple) throws IOException{
52	torben	799	List<RequestPosition> positions = new ArrayList<RequestPosition>();
53
54	torben	795	try {
55	torben	809	String files_single[] = {"/var/log/apache2/app_access.log"};
56			String files_multi[] = {"/var/log/apache2/app_access.log.2.gz", "/var/log/apache2/app_access.log.1", "/var/log/apache2/app_access.log"};
57
58			String files[];
59
60			if (multiple == false) {
61			files = files_single;
62			} else {
63			files = files_multi;
64			}
65	torben	799
66	torben	800	SimpleDateFormat df = new SimpleDateFormat("dd/MMM/yyyy:HH:mm:ss");
67	torben	809
68			for (String fileStr : files ) {
69			File f = new File(fileStr);
70			if ( !f.exists() ) {
71	torben	795	continue;
72			}
73	torben	809
74			FileInputStream fis = new FileInputStream(fileStr);
75
76			InputStream input;
77			if ( isGz(fileStr)) {
78			input = new GZIPInputStream(fis);
79			} else {
80			input = fis;
81			}
82
83			BufferedReader in = new BufferedReader( new InputStreamReader(input) );
84	torben	799
85	torben	809
86			String line;
87			while ( (line=in.readLine()) != null) {
88			if (line.indexOf("LocateStation") == -1 ){
89			continue;
90			}
91
92			if (line.indexOf("latitude") == -1 ) {
93			continue;
94			}
95			RequestPosition pos = new RequestPosition();
96
97			String toks[] = line.split(" ");
98			pos.ip = toks[0];
99
100			pos.time = df.parse( toks[3].replace("[", "") );
101
102			String argpart = toks[6].split("\\?")[1];
103
104			String args[] = argpart.split("&");
105
106			pos.lat = args[0].split("=")[1];
107			pos.lng = args[1].split("=")[1];
108
109			positions.add(pos);
110	torben	795	}
111	torben	809	in.close();
112			input.close();
113			fis.close();
114	torben	795	}
115	torben	800	} catch (ParseException pe) {
116			log.log(Level.SEVERE, "parseException", pe);
117			throw new IOException(pe);
118	torben	799	} catch (IOException e) {
119	torben	796	log.log(Level.SEVERE, "getKml()", e);
120	torben	799	throw e;
121	torben	795	}
122	torben	799
123			return positions;
124	torben	795	}
125
126	torben	799	protected String formatXml(List<RequestPosition> list) {
127			StringBuilder sb = new StringBuilder();
128
129			sb.append( "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" );
130			sb.append( "<kml xmlns=\"http://www.opengis.net/kml/2.2\">\n" );
131			sb.append( "<Document>\n" );
132	torben	800
133
134			sb.append( " <Style id=\"red\">\n" );
135			sb.append( " <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/red-circle.png</href></Icon></IconStyle>\n" );
136			sb.append( " </Style>\n");
137	torben	799
138	torben	800	sb.append( " <Style id=\"yellow\">\n" );
139			sb.append( " <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/ylw-circle.png</href></Icon></IconStyle>\n" );
140			sb.append( " </Style>\n" );
141
142			sb.append( " <Style id=\"green\">\n" );
143			sb.append( " <IconStyle><Icon><href>http://maps.google.com/mapfiles/kml/paddle/grn-circle.png</href></Icon></IconStyle>\n" );
144	torben	801	sb.append( " </Style>\n\n" );
145
146			/*
147			String overlay =
148			" <ScreenOverlay>" +
149			" <name>Absolute Positioning: Top left</name>" +
150			" <description>Absolute Positioning: Top left</description>" +
151			" <visibility>1</visibility>" +
152			" <Icon>" +
153			" <href>http://code.google.com/apis/kml/documentation/top_left.jpg</href>" +
154			" </Icon>" +
155			" <overlayXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
156			" <screenXY x=\"0\" y=\"1\" xunits=\"fraction\" yunits=\"fraction\"/>" +
157			" <rotationXY x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
158			" <size x=\"0\" y=\"0\" xunits=\"fraction\" yunits=\"fraction\"/>" +
159			" </ScreenOverlay>" ;
160
161			sb.append(overlay);
162			*/
163
164	torben	800
165	torben	805	final String STYLE_GREEN = "green";
166			final String STYLE_YELLOW = "yellow";
167			final String STYLE_RED = "red";
168	torben	800
169	torben	805	String oldstyle = null;
170	torben	800	Date now = new Date();
171	torben	799	for(RequestPosition current : list) {
172	torben	800
173			String style;
174			long timediff = now.getTime() - current.time.getTime();
175
176			if ( timediff < (36060*1000) ) {
177	torben	805	style = STYLE_RED;
178	torben	800	} else if ( timediff < (246060*1000)) {
179	torben	805	style = STYLE_YELLOW;
180	torben	800	} else {
181	torben	805	style = STYLE_GREEN;
182	torben	800	}
183
184	torben	805	if ( !style.equals(oldstyle) ) {
185			if (oldstyle != null)
186			sb.append( "</Folder>\n");
187			sb.append( "<Folder>\n");
188			sb.append( " <name>" ).append(style).append("</name>\n");
189	torben	806	sb.append( " <open>0</open>\n" );
190	torben	805
191			oldstyle = style;
192			}
193
194	torben	799	sb.append( " <Placemark>\n" );
195	torben	805	sb.append( " <styleUrl>#" + style + "</styleUrl>\n" );
196	torben	799	sb.append( " <description>IP=" + current.ip + " Time=" + current.time + "</description>\n" );
197			sb.append( " <Point><coordinates>" + current.lng + "," + current.lat + ",0</coordinates></Point>\n" );
198			sb.append( " </Placemark>\n" );
199			}
200	torben	805	sb.append( "</Folder>\n" );
201			sb.append( "</Document>\n" );
202	torben	799	sb.append( "</kml>\n" );
203
204			return sb.toString();
205			}
206	torben	809
207			protected String getRequestsFromFile(boolean multiple) throws IOException {
208			String kmlData = null;
209			String key;
210	torben	802
211	torben	809	if (multiple == false) {
212			key = "kmldata";
213			} else {
214			key = "kmldata-multi";
215			}
216
217			kmlData = cache.get(key);
218
219	torben	803	if (kmlData == null) {
220	torben	809	kmlData = formatXml( getRequestsFromFileWorker(multiple) );
221			cache.put(key, kmlData);
222	torben	803	kmlData += "<!-- from source -->";
223	torben	802	} else {
224	torben	803	kmlData += "<!-- cached -->";
225	torben	802	}
226	torben	809
227			return kmlData;
228			}
229	torben	810
230			boolean enabled(String param) {
231			if (param == null \|\| param.equals("")) {
232			return false;
233			}
234
235			int p = 0;
236			try {
237			p = Integer.parseInt(param);
238			} catch (Exception e) {}
239
240			return (p != 0);
241			}
242	torben	799
243	torben	809	@Override
244			protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
245	torben	802
246	torben	810	boolean multiple = enabled( req.getParameter("multi") );
247	torben	809
248			String kmlData = getRequestsFromFile(multiple);
249
250	torben	810	if ( enabled(req.getParameter("zip")) ) {
251	torben	799
252	torben	798	ByteArrayOutputStream baos = new ByteArrayOutputStream();
253	torben	799
254	torben	798	ZipOutputStream zip = new ZipOutputStream(baos);
255			zip.putNextEntry( new ZipEntry("trains.kml") );
256	torben	803	zip.write( kmlData.getBytes() );
257	torben	798	zip.closeEntry();
258			zip.close();
259	torben	799
260	torben	798	byte bytes[] = baos.toByteArray();
261	torben	799
262	torben	798	resp.setContentType(KMZ);
263			resp.setContentLength( bytes.length );
264			resp.getOutputStream().write(bytes);
265	torben	799
266	torben	798	} else {
267			resp.setContentType(KML);
268	torben	803	resp.getWriter().print( kmlData );
269	torben	798	}
270	torben	795	}
271			}