Added support for baseUrl

opendiffy · Dec 16, 2022 · d3c97ab · d3c97ab
1 parent 5902867
commit d3c97ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/main/scala/ai/diffy/ApiController.scala b/src/main/scala/ai/diffy/ApiController.scala
@@ -160,9 +160,9 @@ class ApiController(
   @GetMapping(path = Array("/api/1/info"))
   def getInfo(): Map[String, _] = Map(
         "name" -> settings.serviceName,
-        "candidate" -> httpServiceToMap(s"${settings.candidateHost}:${settings.candidatePort}"),
-        "primary" -> httpServiceToMap(s"${settings.primaryHost}:${settings.primaryPort}"),
-        "secondary" -> httpServiceToMap(s"${settings.secondaryHost}:${settings.secondaryPort}"),
+        "candidate" -> httpServiceToMap(settings.candidate.toString),
+        "primary" -> httpServiceToMap(settings.primary.toString),
+        "secondary" -> httpServiceToMap(settings.secondary.toString),
         "relativeThreshold" -> settings.relativeThreshold,
         "absoluteThreshold" -> settings.absoluteThreshold,
         "protocol" -> "http",

diff --git a/src/main/scala/ai/diffy/Settings.scala b/src/main/scala/ai/diffy/Settings.scala
@@ -54,5 +54,9 @@ object Downstream {
   }
 }
 sealed trait Downstream
-case class HostPort(host: String, port: Int) extends Downstream
-case class BaseUrl(baseUrl: String) extends Downstream
+case class HostPort(host: String, port: Int) extends Downstream{
+  override def toString: String = s"${host}:${port}"
+}
+case class BaseUrl(baseUrl: String) extends Downstream {
+  override def toString: String = baseUrl
+}