11package org .prebid .server .hooks .modules .greenbids .real .time .data .config ;
22
33import io .netty .handler .codec .http .HttpResponseStatus ;
4+ import io .vertx .core .file .FileSystem ;
45import io .vertx .core .http .HttpClientRequest ;
56import io .vertx .core .http .HttpClientResponse ;
67import com .maxmind .db .Reader ;
2627
2728public class DatabaseReaderFactory implements Initializable {
2829
30+ private static final Logger logger = LoggerFactory .getLogger (DatabaseReaderFactory .class );
31+
2932 private final GreenbidsRealTimeDataProperties properties ;
3033
3134 private final Vertx vertx ;
3235
3336 private final AtomicReference <DatabaseReader > databaseReaderRef = new AtomicReference <>();
3437
35- private static final Logger logger = LoggerFactory . getLogger ( DatabaseReaderFactory . class ) ;
38+ private final FileSystem fileSystem ;
3639
3740 public DatabaseReaderFactory (
3841 GreenbidsRealTimeDataProperties properties , Vertx vertx ) {
3942 this .properties = properties ;
4043 this .vertx = vertx ;
44+ this .fileSystem = vertx .fileSystem ();
4145 }
4246
4347 @ Override
@@ -56,10 +60,9 @@ private Future<DatabaseReader> downloadAndExtract() {
5660 }
5761
5862 private Future <Void > downloadFile (String downloadUrl , String tmpPath ) {
59- return vertx . fileSystem () .open (tmpPath , new OpenOptions ())
63+ return fileSystem .open (tmpPath , new OpenOptions ())
6064 .compose (tmpFile -> sendHttpRequest (downloadUrl )
6165 .compose (response -> response .pipeTo (tmpFile ))
62- .eventually (v -> tmpFile .close ())
6366 .onFailure (error -> logger .error (
6467 "Failed to download file from {} to {}." , downloadUrl , tmpPath , error )));
6568 }
@@ -68,7 +71,7 @@ private Future<HttpClientResponse> sendHttpRequest(String url) {
6871 final RequestOptions options = new RequestOptions ()
6972 .setFollowRedirects (true )
7073 .setMethod (HttpMethod .GET )
71- .setTimeout (properties .timeoutMs )
74+ .setTimeout (properties .getTimeoutMs () )
7275 .setAbsoluteURI (url );
7376
7477 return vertx .createHttpClient ().request (options )
@@ -109,8 +112,8 @@ private DatabaseReader extractMMDB(String tarGzPath) {
109112 }
110113 }
111114
112- private Future < Void > removeFile (String filePath ) {
113- return vertx . fileSystem () .delete (filePath )
115+ private void removeFile (String filePath ) {
116+ fileSystem .delete (filePath )
114117 .onFailure (err -> logger .error ("Failed to remove file {}" , filePath , err ));
115118 }
116119
0 commit comments