added the data column in http response case class

modakanalytics · May 22, 2023 · e0fb4b1 · e0fb4b1
1 parent 184918e
commit e0fb4b1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/scala/com/github/music/of/the/ainur/almaren/http/HTTPConnector.scala b/src/main/scala/com/github/music/of/the/ainur/almaren/http/HTTPConnector.scala
@@ -20,7 +20,8 @@ final case class Response(
   `__STATUS_MSG__`:Option[String] = None,
   `__ERROR__`:Option[String] = None,
   `__ELAPSED_TIME__`:Long,
-   `__URL__`:String)
+   `__URL__`:String,
+  `__DATA__`: String)
 
 final case class ResponseBatch(
   `__ID__`:Seq[String],
@@ -77,7 +78,7 @@ private[almaren] case class HTTP(
     val response = Try(requestHandler(row,session,url,headers,params ++ hiddenParams,method,connectTimeout,readTimeout))
     val elapsedTime = System.currentTimeMillis() - startTime
     val id = row.getAs[Any](Alias.IdCol).toString
-
+    val data = row.getAs[Any](Alias.DataCol).toString
     def getResponse(r: requests.Response) = Response(
       id,
       Some(r.text()),
@@ -88,7 +89,8 @@ private[almaren] case class HTTP(
         case _ => Some(r.statusMessage)
       },
       `__ELAPSED_TIME__` = elapsedTime,
-      `__URL__` = url
+      `__URL__` = url,
+      `__DATA__` = data
     )
 
     response match {