Merge pull request #2 from Making-Sense-Info/v1.0.1

Add check endpoint

Author: NicoLaval

pom.xml

@@ -9,7 +9,7 @@
     </parent>
     <groupId>fr.insee</groupId>
     <artifactId>trevas-lab</artifactId>
-    <version>1.0.0</version>
+    <version>1.1.0</version>
     <name>Trevas Lab</name>
     <description>VTL API calling Trevas</description>
 

src/main/java/fr/insee/trevas/lab/controller/TrevasLabController.java

@@ -44,6 +44,19 @@ public class TrevasLabController {
     @Autowired
     private ObjectMapper objectMapper;
 
+    @GetMapping("check")
+    public ResponseEntity<String> check() {
+        try (SparkSession sparkSession = sparkEngine.buildSparkSession()) {
+            if (sparkSession != null && !sparkSession.sparkContext().isStopped()) {
+                return ResponseEntity.ok("Spark Session is available and running.");
+            } else {
+                return ResponseEntity.status(500).body("Spark Session is not available.");
+            }
+        } catch (Exception e) {
+            return ResponseEntity.status(500).body("Error checking Spark Session: " + e.getMessage());
+        }
+    }
+
     @PostMapping("/connect")
     public ResponseEntity<EditVisualize> getDataFromConnector(
             Authentication auth,

src/main/java/fr/insee/trevas/lab/service/SparkEngine.java

@@ -36,7 +36,7 @@ public class SparkEngine {
     @Autowired
     private ObjectMapper objectMapper;
 
-    private SparkSession buildSparkSession() {
+    public SparkSession buildSparkSession() {
         SparkConf conf = Utils.loadSparkConfig(System.getenv("SPARK_CONF_DIR"));
         conf.set("spark.driver.allowMultipleContexts", "true");
         // Note: all the dependencies are required for deserialization.