From e0fb4b1e343a47294b08eae7cfdabf8ff4d5a942 Mon Sep 17 00:00:00 2001
From: badrinathpatchikolla <badrinath.patchikolla@modak.com>
Date: Mon, 22 May 2023 23:11:13 +0530
Subject: [PATCH] added the data column in http response case class

---
 .../music/of/the/ainur/almaren/http/HTTPConnector.scala   | 8 +++++---
 1 file changed, 5 insertions(+), 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
index ee058b6..2b06204 100644
--- 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 {