/
GeoDataPatcher.scala
45 lines (32 loc) · 1.28 KB
/
GeoDataPatcher.scala
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package com.patson.init
import com.patson.Util
import com.patson.data.{AirportSource, CitySource, CountrySource}
import com.patson.model._
import scala.collection.mutable.{ArrayBuffer, ListBuffer}
import scala.concurrent.{Await, Future}
import scala.concurrent.duration.Duration
object GeoDataPatcher extends App {
import actorSystem.dispatcher
//implicit val materializer = FlowMaterializer()
mainFlow
def mainFlow() {
val incomeInfo = GeoDataGenerator.getIncomeInfo()
val getCityFuture = GeoDataGenerator.getCity(incomeInfo)
var cities = Await.result(getCityFuture, Duration.Inf)
println("FROM " + cities.length)
cities = cities ++ AdditionalLoader.loadAdditionalCities(incomeInfo)
println("TO " + cities.length)
//cities.foreach(println)
//make sure cities are saved first as we need the id for airport info
try {
// AirportSource.deleteAllAirports()
CitySource.deleteAllCitites()
CitySource.saveCities(cities)
} catch {
case e : Throwable => e.printStackTrace()
}
val airports = GeoDataGenerator.buildAirportData(GeoDataGenerator.getAirport(), GeoDataGenerator.getRunway(), cities)
GeoDataGenerator.buildCountryData(airports)
Await.result(actorSystem.terminate(), Duration.Inf)
}
}