daoas/fulddaekning/LookupMain.java

package dk.daoas.fulddaekning;

import java.io.File;
import java.io.FileReader;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.Set;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.logging.FileHandler;
import java.util.logging.Logger;
import java.util.logging.SimpleFormatter;

import ags.utils.dataStructures.trees.thirdGenKD.KdTree;


public class LookupMain {


        static final String CONFIG_FILENAME = "fulddaekning.properties";

        static boolean use_osrm;
        static boolean test_mode;
        static SafeProperties conf;

        static int max_workers;
        static boolean verbose;

        static String trigger_url;

        static String distributor;

        final static Logger logger = Logger.getLogger( LookupMain.class.toString() );

        public static ProgressBar progressBar;


        static ThreadPoolExecutor threadPool;

        private static void setupThreadPool() {  
                threadPool = (ThreadPoolExecutor) Executors.newFixedThreadPool(max_workers, new WorkerThreadFactory() );         
        }


        public static void main(String[] args) throws Exception {

                //Setup j.u.l Logger
                Logger root = Logger.getLogger("");
                FileHandler fhandler = new FileHandler("fulddaekning.log"); // Ingen max størrelse, ingen rotation og ingen append
                fhandler.setFormatter( new SimpleFormatter() );
                root.addHandler( fhandler );


                File confFile = new File( CONFIG_FILENAME );
                if (! confFile.exists() ) {
                        logger.warning("Config file not found: " + CONFIG_FILENAME);
                        System.exit(1);
                }

                int antalIkkeDaekkede = -1;
                
                conf = new SafeProperties();
                conf.load( new FileReader(confFile) );


                verbose = Boolean.parseBoolean( conf.getSafeProperty("VERBOSE") );
                logger.info("Starting with VERBOSE:" + verbose);

                test_mode = Boolean.parseBoolean( conf.getSafeProperty("TEST") );
                logger.info("Starting with TEST:" + test_mode);

                use_osrm = Boolean.parseBoolean( conf.getSafeProperty("USE_OSRM") );
                logger.info("Starting with USE_OSRM:" + use_osrm);

                distributor = conf.getSafeProperty("DISTRIBUTOR");
                distributor = distributor.toUpperCase();
                logger.info("Starting for DISTRIBUTOR:" + distributor);

                trigger_url = conf.getSafeProperty("TRIGGER_URL");

                Constants.init(distributor);
                Constants consts = Constants.getInstance();

                max_workers = Integer.parseInt( conf.getSafeProperty("MAX_WORKERS") );   
                if (max_workers <= 0) {  
                        logger.info("!!! AUTO-DETECT MAX_WORKERS !!!");  
                        int cores = Runtime.getRuntime().availableProcessors();  
                        cores -= 1;//Efterlad 1 core/cpu i reserve til systemet  

                        max_workers = Math.max(1, cores); //Dog skal der som minimum være 1 core til beregning  

                }        
                if (test_mode) {
                        max_workers = 1;
                }
                logger.info("Starting with MAX_WORKERS:" + max_workers);         
                setupThreadPool();

                
                Database db = new Database(conf);
                Queue<Adresse> ikkeDaekkede = db.hentAlleIkkedaekkedeAdresser(consts.getMinPostnr(), consts.getMaxPostnr() );
                antalIkkeDaekkede = ikkeDaekkede.size();
                logger.info("Antal ikke-daekkede: " + antalIkkeDaekkede );

                progressBar = new ProgressBar( antalIkkeDaekkede );


                boolean testRun= false;

                AtomicInteger antalFundne = new AtomicInteger(0);

                long beregnStart =0;
                long start = System.currentTimeMillis();
                int antalDaekkedeAdresser = -1;

                if (testRun == false) {

                        logger.info("Finder postnumre");
                        Set<Short> postnumre = db.hentPostnumreCache();

                        // Først validerer vi BBox på alle postnummre, for at undgå fuldt stop midt i beregningen
                        for(short postnr : postnumre) { //
                                logger.info("Validerer BBox for " + postnr);
                                BoundingBox bbox = db.getBoundingbox(postnr);
                                bbox.validateBbox();
                        }


                        logger.info("Henter alle daekkede adresser");
                        antalDaekkedeAdresser = db.hentAlleDaekkedeAdresser();
                        logger.info( "AlleDaekkedeAdresser.length=" + antalDaekkedeAdresser );

                        Map<Short, List<Adresse>> addrHoList = db.getDaekkedeAdresserHO();

                        Map<Short, KdTree<Adresse>> hoTrees = new HashMap<Short,KdTree<Adresse>>();

                        for ( Map.Entry<Short, List<Adresse>> entry : addrHoList.entrySet() ) {
                                short ho = entry.getKey();
                                List<Adresse> geopoints = entry.getValue();

                                logger.info("Opbygger KDTree for " + ho + " - antal=" + geopoints.size() );

                                int bucketSize = 96*12;

                                KdTree<Adresse> addressTree = new KdTree<Adresse>( 3, bucketSize );
                                for(Adresse a: geopoints) {
                                        addressTree.addPoint(a.xyz, a);
                                }
                                
                                hoTrees.put(ho, addressTree);                           
                        }

                        db.resetResultTable();

                        beregnStart = System.currentTimeMillis();
                        logger.info("Starter beregning");

                        //pre-check er ok - reset tmp tabel og start søgningen
                        
                        CyclicBarrier barrier = new CyclicBarrier(LookupMain.max_workers + 1);
                        
                        for (int i=0; i<max_workers; i++) {
                                LookupWorker worker = new LookupWorker(i, barrier, ikkeDaekkede, hoTrees,antalFundne,db,verbose,consts, progressBar);
                                threadPool.submit( worker );
                        }
                        
                        barrier.await(); // Afvent at workerne bliver færdige
                        
                        logger.info("Calc is done - cleaning up remaining bits");
                        
                        threadPool.shutdown(); //Calc is done now


                        db.saveBatch();

                        if (test_mode == false) {
                                db.renameResultTables();

                                logger.info("Calling trigger URL");
                                String url = trigger_url + "&trigger=AfstandAndenRute/" + distributor;

                                HttpUtil.getContentString(url, 500);

                        } else {
                                logger.info( "Rename tables is disabled !!!" );
                        }


                } else {
                        /// Test
                        /*
                        db.resetResultTable();


                        alleDaekkedeAdresser = db.hentAlleDaekkedeAdresser();
                        logger.info( "AlleDaekkedeAdresser.length=" + alleDaekkedeAdresser.size());

                        short post = (short) 2700;                      
                        Lookup lookup = new Lookup(post, db, threadPool);
                        lookup.doLookup();*/                                                            
                }


                long now = System.currentTimeMillis();
                long elapsed = now - start ;
                long elapsedBeregn = now - beregnStart;


                logger.info("Fuld load done : " + formatMilliSeconds(elapsed) );
                logger.info("Fuld load done (beregning) : " + formatMilliSeconds(elapsedBeregn) );
                logger.info("Antal daekkede : " + antalDaekkedeAdresser );
                logger.info("Antal ikke-daekkede : " + antalIkkeDaekkede );
                logger.info("Heraf, antal fundne : " + antalFundne );

                logger.info( String.format("Fandt adresser til : %.2f %%", (antalFundne.get() *100.0)/antalIkkeDaekkede ) );
                //logger.info("Fandt adresser til : " + (antalFundne*100.0)/ikkeDaekkede.length + "%" );
        }


        static String formatMilliSeconds(long milliseconds) {
                int mseconds = (int) milliseconds % 1000;
                int seconds = (int) (milliseconds / 1000) % 60 ;
                int minutes = (int) ((milliseconds / (1000*60)) % 60);
                int hours   = (int) ((milliseconds / (1000*60*60)) % 24);

                return String.format("%02d:%02d:%02d.%03d", hours, minutes, seconds, mseconds);
        }


        static class WorkerThreadFactory implements ThreadFactory {      
                int count = 0;   

                @Override        
                public Thread newThread(Runnable r) {    
                        return new Thread(r, "lookupWorker/" + count++);         
                }
        }
}
1	torben	2241	package dk.daoas.fulddaekning;
2
3			import java.io.File;
4			import java.io.FileReader;
5	torben	2597	import java.util.HashMap;
6	torben	2585	import java.util.List;
7	torben	2597	import java.util.Map;
8	torben	2711	import java.util.Queue;
9	torben	2328	import java.util.Set;
10	torben	2711	import java.util.concurrent.CyclicBarrier;
11			import java.util.concurrent.Executors;
12			import java.util.concurrent.ThreadFactory;
13			import java.util.concurrent.ThreadPoolExecutor;
14			import java.util.concurrent.atomic.AtomicInteger;
15	torben	2241	import java.util.logging.FileHandler;
16			import java.util.logging.Logger;
17			import java.util.logging.SimpleFormatter;
18
19	torben	2744	import ags.utils.dataStructures.trees.thirdGenKD.KdTree;
20	torben	2707
21	torben	2708
22	torben	2744
23	torben	2241	public class LookupMain {
24
25
26			static final String CONFIG_FILENAME = "fulddaekning.properties";
27	torben	2711
28	torben	2753	static boolean use_osrm;
29	torben	2736	static boolean test_mode;
30	torben	2241	static SafeProperties conf;
31	torben	2711
32	torben	2241	static int max_workers;
33			static boolean verbose;
34	torben	2711
35	torben	2771	static String trigger_url;
36
37	torben	2241	static String distributor;
38
39			final static Logger logger = Logger.getLogger( LookupMain.class.toString() );
40	torben	2259
41	torben	2797	public static ProgressBar progressBar;
42	torben	2711
43	torben	2797
44	torben	2711	static ThreadPoolExecutor threadPool;
45
46			private static void setupThreadPool() {
47			threadPool = (ThreadPoolExecutor) Executors.newFixedThreadPool(max_workers, new WorkerThreadFactory() );
48			}
49
50
51
52	torben	2241	public static void main(String[] args) throws Exception {
53	torben	2711
54	torben	2241	//Setup j.u.l Logger
55			Logger root = Logger.getLogger("");
56			FileHandler fhandler = new FileHandler("fulddaekning.log"); // Ingen max størrelse, ingen rotation og ingen append
57			fhandler.setFormatter( new SimpleFormatter() );
58			root.addHandler( fhandler );
59	torben	2711
60
61
62
63	torben	2241	File confFile = new File( CONFIG_FILENAME );
64			if (! confFile.exists() ) {
65			logger.warning("Config file not found: " + CONFIG_FILENAME);
66			System.exit(1);
67			}
68	torben	2711
69	torben	2715	int antalIkkeDaekkede = -1;
70
71	torben	2241	conf = new SafeProperties();
72			conf.load( new FileReader(confFile) );
73	torben	2711
74
75
76	torben	2241	verbose = Boolean.parseBoolean( conf.getSafeProperty("VERBOSE") );
77			logger.info("Starting with VERBOSE:" + verbose);
78	torben	2711
79	torben	2736	test_mode = Boolean.parseBoolean( conf.getSafeProperty("TEST") );
80			logger.info("Starting with TEST:" + test_mode);
81	torben	2711
82	torben	2753	use_osrm = Boolean.parseBoolean( conf.getSafeProperty("USE_OSRM") );
83			logger.info("Starting with USE_OSRM:" + use_osrm);
84
85	torben	2241	distributor = conf.getSafeProperty("DISTRIBUTOR");
86			distributor = distributor.toUpperCase();
87			logger.info("Starting for DISTRIBUTOR:" + distributor);
88	torben	2711
89	torben	2771	trigger_url = conf.getSafeProperty("TRIGGER_URL");
90
91	torben	2241	Constants.init(distributor);
92			Constants consts = Constants.getInstance();
93	torben	2711
94			max_workers = Integer.parseInt( conf.getSafeProperty("MAX_WORKERS") );
95			if (max_workers <= 0) {
96			logger.info("!!! AUTO-DETECT MAX_WORKERS !!!");
97			int cores = Runtime.getRuntime().availableProcessors();
98			cores -= 1;//Efterlad 1 core/cpu i reserve til systemet
99
100			max_workers = Math.max(1, cores); //Dog skal der som minimum være 1 core til beregning
101
102			}
103	torben	2742	if (test_mode) {
104			max_workers = 1;
105			}
106	torben	2711	logger.info("Starting with MAX_WORKERS:" + max_workers);
107			setupThreadPool();
108
109	torben	2717
110			Database db = new Database(conf);
111			Queue<Adresse> ikkeDaekkede = db.hentAlleIkkedaekkedeAdresser(consts.getMinPostnr(), consts.getMaxPostnr() );
112			antalIkkeDaekkede = ikkeDaekkede.size();
113			logger.info("Antal ikke-daekkede: " + antalIkkeDaekkede );
114
115	torben	2797	progressBar = new ProgressBar( antalIkkeDaekkede );
116	torben	2717
117
118	torben	2797
119	torben	2241	boolean testRun= false;
120	torben	2711
121			AtomicInteger antalFundne = new AtomicInteger(0);
122
123	torben	2707	long beregnStart =0;
124	torben	2241	long start = System.currentTimeMillis();
125	torben	2597	int antalDaekkedeAdresser = -1;
126	torben	2711
127	torben	2241	if (testRun == false) {
128	torben	2711
129	torben	2241	logger.info("Finder postnumre");
130	torben	2575	Set<Short> postnumre = db.hentPostnumreCache();
131	torben	2711
132	torben	2241	// Først validerer vi BBox på alle postnummre, for at undgå fuldt stop midt i beregningen
133	torben	2575	for(short postnr : postnumre) { //
134	torben	2241	logger.info("Validerer BBox for " + postnr);
135			BoundingBox bbox = db.getBoundingbox(postnr);
136			bbox.validateBbox();
137			}
138	torben	2263
139	torben	2711
140	torben	2263	logger.info("Henter alle daekkede adresser");
141	torben	2597	antalDaekkedeAdresser = db.hentAlleDaekkedeAdresser();
142			logger.info( "AlleDaekkedeAdresser.length=" + antalDaekkedeAdresser );
143	torben	2711
144	torben	2707	Map<Short, List<Adresse>> addrHoList = db.getDaekkedeAdresserHO();
145	torben	2711
146	torben	2707	Map<Short, KdTree<Adresse>> hoTrees = new HashMap<Short,KdTree<Adresse>>();
147	torben	2711
148	torben	2707	for ( Map.Entry<Short, List<Adresse>> entry : addrHoList.entrySet() ) {
149	torben	2597	short ho = entry.getKey();
150	torben	2707	List<Adresse> geopoints = entry.getValue();
151	torben	2711
152	torben	2597	logger.info("Opbygger KDTree for " + ho + " - antal=" + geopoints.size() );
153	torben	2711
154	torben	2746	int bucketSize = 96*12;
155
156			KdTree<Adresse> addressTree = new KdTree<Adresse>( 3, bucketSize );
157	torben	2744	for(Adresse a: geopoints) {
158			addressTree.addPoint(a.xyz, a);
159			}
160
161	torben	2597	hoTrees.put(ho, addressTree);
162			}
163	torben	2711
164	torben	2585	db.resetResultTable();
165	torben	2711
166	torben	2707	beregnStart = System.currentTimeMillis();
167	torben	2585	logger.info("Starter beregning");
168	torben	2711
169	torben	2241	//pre-check er ok - reset tmp tabel og start søgningen
170
171	torben	2711	CyclicBarrier barrier = new CyclicBarrier(LookupMain.max_workers + 1);
172	torben	2585
173	torben	2711	for (int i=0; i<max_workers; i++) {
174	torben	2797	LookupWorker worker = new LookupWorker(i, barrier, ikkeDaekkede, hoTrees,antalFundne,db,verbose,consts, progressBar);
175	torben	2711	threadPool.submit( worker );
176	torben	2241	}
177
178	torben	2711	barrier.await(); // Afvent at workerne bliver færdige
179	torben	2714
180	torben	2719	logger.info("Calc is done - cleaning up remaining bits");
181
182	torben	2714	threadPool.shutdown(); //Calc is done now
183	torben	2711
184
185	torben	2585	db.saveBatch();
186	torben	2711
187	torben	2741	if (test_mode == false) {
188	torben	2241	db.renameResultTables();
189	torben	2771
190			logger.info("Calling trigger URL");
191			String url = trigger_url + "&trigger=AfstandAndenRute/" + distributor;
192
193			HttpUtil.getContentString(url, 500);
194
195	torben	2241	} else {
196			logger.info( "Rename tables is disabled !!!" );
197			}
198
199	torben	2711
200
201	torben	2241	} else {
202			/// Test
203	torben	2585	/*
204	torben	2241	db.resetResultTable();
205	torben	2422
206	torben	2711
207	torben	2422	alleDaekkedeAdresser = db.hentAlleDaekkedeAdresser();
208	torben	2585	logger.info( "AlleDaekkedeAdresser.length=" + alleDaekkedeAdresser.size());
209	torben	2422
210	torben	2575	short post = (short) 2700;
211			Lookup lookup = new Lookup(post, db, threadPool);
212	torben	2585	lookup.doLookup();*/
213	torben	2241	}
214	torben	2260
215	torben	2711
216	torben	2241	long now = System.currentTimeMillis();
217			long elapsed = now - start ;
218	torben	2707	long elapsedBeregn = now - beregnStart;
219	torben	2585
220	torben	2711
221	torben	2241	logger.info("Fuld load done : " + formatMilliSeconds(elapsed) );
222	torben	2707	logger.info("Fuld load done (beregning) : " + formatMilliSeconds(elapsedBeregn) );
223	torben	2597	logger.info("Antal daekkede : " + antalDaekkedeAdresser );
224	torben	2715	logger.info("Antal ikke-daekkede : " + antalIkkeDaekkede );
225	torben	2587	logger.info("Heraf, antal fundne : " + antalFundne );
226	torben	2711
227	torben	2715	logger.info( String.format("Fandt adresser til : %.2f %%", (antalFundne.get() *100.0)/antalIkkeDaekkede ) );
228	torben	2688	//logger.info("Fandt adresser til : " + (antalFundne*100.0)/ikkeDaekkede.length + "%" );
229	torben	2241	}
230
231
232	torben	2711
233	torben	2241	static String formatMilliSeconds(long milliseconds) {
234			int mseconds = (int) milliseconds % 1000;
235			int seconds = (int) (milliseconds / 1000) % 60 ;
236			int minutes = (int) ((milliseconds / (1000*60)) % 60);
237			int hours = (int) ((milliseconds / (10006060)) % 24);
238	torben	2711
239	torben	2241	return String.format("%02d:%02d:%02d.%03d", hours, minutes, seconds, mseconds);
240	torben	2260	}
241	torben	2711
242
243			static class WorkerThreadFactory implements ThreadFactory {
244			int count = 0;
245
246			@Override
247			public Thread newThread(Runnable r) {
248			return new Thread(r, "lookupWorker/" + count++);
249			}
250			}
251	torben	2241	}