adressevedligehold/afstandandenrute/AfstandAndenRuteTask.java

package dk.daoas.adressevedligehold.afstandandenrute;

import java.sql.Connection;
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 ags.utils.dataStructures.trees.thirdGenKD.KdTree;
import dk.daoas.adressevedligehold.ReloadHelper;
import dk.daoas.adressevedligehold.ServiceConfig;
import dk.daoas.adressevedligehold.beans.Address;
import dk.daoas.adressevedligehold.db.DBConnection;
import dk.daoas.adressevedligehold.tasks.Task;
import dk.daoas.adressevedligehold.tasks.TaskLogger;


public class AfstandAndenRuteTask extends Task {

        private TaskLogger logger = TaskLogger.getInstance();

        final static boolean VERBOSE = false;//


        static boolean test_mode;


        String distributor;
        int antalIkkeDaekkede = -1;
        
        AtomicInteger antalBeregnet = new AtomicInteger();


        ThreadPoolExecutor threadPool;

        boolean isIncremental;
        String type;


        public AfstandAndenRuteTask(String distributor, String type) {
                this.distributor = distributor;
                this.type = type;
                
                type = type.toUpperCase();
                switch(type) {
                case "FULL":
                        isIncremental = false;
                        break;
                case "INCREMENTAL":
                        isIncremental = true;
                        break;
                default:
                        throw new RuntimeException("Unknown type" + type);                      
                }
        }


        @Override
        protected void taskRun() throws Exception {
                

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

                int max_workers = ServiceConfig.getInstance().maxWorkers;        
                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(max_workers);
                
                logger.info("Starting with INCREMENTAL:" + isIncremental);


                try (Connection conn = DBConnection.getConnection() ) {
                        Database db = new Database(conn, isIncremental);
                        
                        db.prepareIncrementalSearch();

                        Queue<Address> ikkeDaekkede = db.hentAlleIkkedaekkedeAdresser(consts.getMinPostnr(), consts.getMaxPostnr() );
                        antalIkkeDaekkede = ikkeDaekkede.size();
                        logger.info("Antal ikke-daekkede: " + 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(distributor);
                                logger.info( "AlleDaekkedeAdresser.length=" + antalDaekkedeAdresser );

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

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

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

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

                                        int bucketSize = 96*12;

                                        KdTree<Address> addressTree = new KdTree<Address>( 3, bucketSize );
                                        for(Address 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(max_workers + 1);

                                for (int i=0; i<max_workers; i++) {
                                        LookupWorker worker = new LookupWorker(i, this, barrier, ikkeDaekkede, hoTrees,antalFundne,antalBeregnet,db,VERBOSE,consts);
                                        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 && this.isAborted() == false) {
                                        db.renameResultTables();
                                        
                                        ReloadHelper.triggerReload( "AfstandAndenRute/" + distributor );


                                } 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);
        }

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

        static class WorkerThreadFactory implements ThreadFactory {      
                int count = 0;   

                @Override        
                public Thread newThread(Runnable r) {    
                        return new Thread(r, "lookupWorker/" + count++);         
                }
        }


        @Override
        public String getDescription() {
                return "Afstand anden rute/" + distributor;
        }


        @Override
        public String getDetail() {
                if (isIncremental) {
                        return "Type: incremental";
                } else {
                        return "Type: full";
                }
        }


        @Override
        public double getPercentCompleted() {
                return (antalBeregnet.get() / ((float)antalIkkeDaekkede)) * 100.0;
        }
}
1	torben	2878	package dk.daoas.adressevedligehold.afstandandenrute;
2
3			import java.sql.Connection;
4			import java.util.HashMap;
5			import java.util.List;
6			import java.util.Map;
7			import java.util.Queue;
8			import java.util.Set;
9			import java.util.concurrent.CyclicBarrier;
10			import java.util.concurrent.Executors;
11			import java.util.concurrent.ThreadFactory;
12			import java.util.concurrent.ThreadPoolExecutor;
13			import java.util.concurrent.atomic.AtomicInteger;
14
15			import ags.utils.dataStructures.trees.thirdGenKD.KdTree;
16	torben	2896	import dk.daoas.adressevedligehold.ReloadHelper;
17	torben	2878	import dk.daoas.adressevedligehold.ServiceConfig;
18			import dk.daoas.adressevedligehold.beans.Address;
19			import dk.daoas.adressevedligehold.db.DBConnection;
20			import dk.daoas.adressevedligehold.tasks.Task;
21	torben	2903	import dk.daoas.adressevedligehold.tasks.TaskLogger;
22	torben	2878
23
24
25			public class AfstandAndenRuteTask extends Task {
26
27	torben	2903	private TaskLogger logger = TaskLogger.getInstance();
28	torben	2878
29	torben	2908	final static boolean VERBOSE = false;//
30	torben	2878
31
32			static boolean test_mode;
33
34
35			String distributor;
36	torben	2891	int antalIkkeDaekkede = -1;
37
38			AtomicInteger antalBeregnet = new AtomicInteger();
39	torben	2878
40
41
42			ThreadPoolExecutor threadPool;
43
44	torben	2927	boolean isIncremental;
45			String type;
46	torben	2878
47
48	torben	2927	public AfstandAndenRuteTask(String distributor, String type) {
49	torben	2878	this.distributor = distributor;
50	torben	2927	this.type = type;
51
52			type = type.toUpperCase();
53			switch(type) {
54			case "FULL":
55			isIncremental = false;
56			break;
57			case "INCREMENTAL":
58			isIncremental = true;
59			break;
60			default:
61			throw new RuntimeException("Unknown type" + type);
62			}
63	torben	2878	}
64
65
66			@Override
67			protected void taskRun() throws Exception {
68	torben	2927
69	torben	2878
70			Constants.init(distributor);
71			Constants consts = Constants.getInstance();
72
73			int max_workers = ServiceConfig.getInstance().maxWorkers;
74			if (max_workers <= 0) {
75			logger.info("!!! AUTO-DETECT MAX_WORKERS !!!");
76			int cores = Runtime.getRuntime().availableProcessors();
77			cores -= 1;//Efterlad 1 core/cpu i reserve til systemet
78
79			max_workers = Math.max(1, cores); //Dog skal der som minimum være 1 core til beregning
80
81			}
82			if (test_mode) {
83			max_workers = 1;
84			}
85			logger.info("Starting with MAX_WORKERS:" + max_workers);
86			setupThreadPool(max_workers);
87	torben	2927
88			logger.info("Starting with INCREMENTAL:" + isIncremental);
89	torben	2878
90
91	torben	2891
92	torben	2878	try (Connection conn = DBConnection.getConnection() ) {
93	torben	2927	Database db = new Database(conn, isIncremental);
94
95			db.prepareIncrementalSearch();
96	torben	2878
97			Queue<Address> ikkeDaekkede = db.hentAlleIkkedaekkedeAdresser(consts.getMinPostnr(), consts.getMaxPostnr() );
98			antalIkkeDaekkede = ikkeDaekkede.size();
99			logger.info("Antal ikke-daekkede: " + antalIkkeDaekkede );
100
101
102			boolean testRun= false;
103
104			AtomicInteger antalFundne = new AtomicInteger(0);
105
106			long beregnStart =0;
107			long start = System.currentTimeMillis();
108			int antalDaekkedeAdresser = -1;
109
110			if (testRun == false) {
111
112			logger.info("Finder postnumre");
113			Set<Short> postnumre = db.hentPostnumreCache();
114
115			// Først validerer vi BBox på alle postnummre, for at undgå fuldt stop midt i beregningen
116			for(short postnr : postnumre) { //
117			logger.info("Validerer BBox for " + postnr);
118			BoundingBox bbox = db.getBoundingbox(postnr);
119			bbox.validateBbox();
120			}
121
122
123			logger.info("Henter alle daekkede adresser");
124			antalDaekkedeAdresser = db.hentAlleDaekkedeAdresser(distributor);
125			logger.info( "AlleDaekkedeAdresser.length=" + antalDaekkedeAdresser );
126
127			Map<Short, List<Address>> addrHoList = db.getDaekkedeAdresserHO();
128
129			Map<Short, KdTree<Address>> hoTrees = new HashMap<Short,KdTree<Address>>();
130
131			for ( Map.Entry<Short, List<Address>> entry : addrHoList.entrySet() ) {
132			short ho = entry.getKey();
133			List<Address> geopoints = entry.getValue();
134
135			logger.info("Opbygger KDTree for " + ho + " - antal=" + geopoints.size() );
136
137			int bucketSize = 96*12;
138
139			KdTree<Address> addressTree = new KdTree<Address>( 3, bucketSize );
140			for(Address a: geopoints) {
141			addressTree.addPoint(a.xyz, a);
142			}
143
144			hoTrees.put(ho, addressTree);
145			}
146
147			db.resetResultTable();
148
149			beregnStart = System.currentTimeMillis();
150			logger.info("Starter beregning");
151
152			//pre-check er ok - reset tmp tabel og start søgningen
153
154			CyclicBarrier barrier = new CyclicBarrier(max_workers + 1);
155
156			for (int i=0; i<max_workers; i++) {
157	torben	2908	LookupWorker worker = new LookupWorker(i, this, barrier, ikkeDaekkede, hoTrees,antalFundne,antalBeregnet,db,VERBOSE,consts);
158	torben	2878	threadPool.submit( worker );
159			}
160
161			barrier.await(); // Afvent at workerne bliver færdige
162
163			logger.info("Calc is done - cleaning up remaining bits");
164
165			threadPool.shutdown(); //Calc is done now
166
167
168			db.saveBatch();
169
170			if (test_mode == false && this.isAborted() == false) {
171			db.renameResultTables();
172	torben	2896
173			ReloadHelper.triggerReload( "AfstandAndenRute/" + distributor );
174	torben	2878
175
176			} else {
177			logger.info( "Rename tables is disabled !!!" );
178			}
179
180
181
182			} else {
183			/// Test
184			/*
185			db.resetResultTable();
186
187
188			alleDaekkedeAdresser = db.hentAlleDaekkedeAdresser();
189			logger.info( "AlleDaekkedeAdresser.length=" + alleDaekkedeAdresser.size());
190
191			short post = (short) 2700;
192			Lookup lookup = new Lookup(post, db, threadPool);
193			lookup.doLookup();*/
194			}
195
196
197			long now = System.currentTimeMillis();
198			long elapsed = now - start ;
199			long elapsedBeregn = now - beregnStart;
200
201
202			logger.info("Fuld load done : " + formatMilliSeconds(elapsed) );
203			logger.info("Fuld load done (beregning) : " + formatMilliSeconds(elapsedBeregn) );
204			logger.info("Antal daekkede : " + antalDaekkedeAdresser );
205			logger.info("Antal ikke-daekkede : " + antalIkkeDaekkede );
206			logger.info("Heraf, antal fundne : " + antalFundne );
207
208			logger.info( String.format("Fandt adresser til : %.2f %%", (antalFundne.get() *100.0)/antalIkkeDaekkede ) );
209			//logger.info("Fandt adresser til : " + (antalFundne*100.0)/ikkeDaekkede.length + "%" );
210
211			}
212			}
213
214
215
216			static String formatMilliSeconds(long milliseconds) {
217			int mseconds = (int) milliseconds % 1000;
218			int seconds = (int) (milliseconds / 1000) % 60 ;
219			int minutes = (int) ((milliseconds / (1000*60)) % 60);
220			int hours = (int) ((milliseconds / (10006060)) % 24);
221
222			return String.format("%02d:%02d:%02d.%03d", hours, minutes, seconds, mseconds);
223			}
224
225			private void setupThreadPool(int max_workers) {
226			threadPool = (ThreadPoolExecutor) Executors.newFixedThreadPool(max_workers, new WorkerThreadFactory() );
227			}
228
229			static class WorkerThreadFactory implements ThreadFactory {
230			int count = 0;
231
232			@Override
233			public Thread newThread(Runnable r) {
234			return new Thread(r, "lookupWorker/" + count++);
235			}
236			}
237
238
239
240
241
242
243			@Override
244			public String getDescription() {
245	torben	2891	return "Afstand anden rute/" + distributor;
246	torben	2878	}
247
248
249
250			@Override
251			public String getDetail() {
252	torben	2927	if (isIncremental) {
253			return "Type: incremental";
254			} else {
255			return "Type: full";
256			}
257	torben	2878	}
258
259
260
261			@Override
262	torben	2891	public double getPercentCompleted() {
263			return (antalBeregnet.get() / ((float)antalIkkeDaekkede)) * 100.0;
264	torben	2878	}
265			}