Refactor rest package into separate files

1 files changed, 14 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/core/rest/SavingUsedServiceDiscovery.scala b/src/main/scala/xyz/driver/core/rest/SavingUsedServiceDiscovery.scala
new file mode 100644
index 0000000..8018bdf
--- /dev/null
+++ b/src/main/scala/xyz/driver/core/rest/SavingUsedServiceDiscovery.scala
@@ -0,0 +1,14 @@
+package xyz.driver.core.rest
+
+import xyz.driver.core.Name
+
+trait SavingUsedServiceDiscovery {
+
+  private val usedServices = new scala.collection.mutable.HashSet[String]()
+
+  def saveServiceUsage(serviceName: Name[Service]): Unit = usedServices.synchronized {
+    usedServices += serviceName.value
+  }
+
+  def getUsedServices: Set[String] = usedServices.synchronized { usedServices.toSet }
+}