small change to ensure thread-safety from first inspection

databrickslabs · Nov 27, 2024 · e62c017 · e62c017
1 parent da10dc5
commit e62c017
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/main/scala/com/databricks/labs/mosaic/core/raster/gdal/MosaicRasterGDAL.scala b/src/main/scala/com/databricks/labs/mosaic/core/raster/gdal/MosaicRasterGDAL.scala
@@ -738,13 +738,9 @@ object MosaicRasterGDAL extends RasterReader {
             case Some(value) if (value) => OF_READONLY
             case _ => OF_THREAD_SAFE
         }
-        driverShortName match {
-            case Some(driverShortName) =>
-                val drivers = new JVector[String]()
-                drivers.add(driverShortName)
-                gdal.OpenEx(path, openFlags, drivers)
-            case None                  => gdal.Open(path, openFlags)
-        }
+        val drivers = new JVector[String]()
+        drivers.add(driverShortName.getOrElse(gdal.IdentifyDriver(path).getShortName))
+        gdal.OpenEx(path, OF_RASTER | openFlags, drivers)
     }
 
     /**