src/main/scala/xyz/driver/core/rest/ServiceDiscovery.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

package xyz.driver.core.rest

import xyz.driver.core.Name

trait ServiceDiscovery {

  def discover[T <: Service](serviceName: Name[Service]): T
}

object ServiceDiscovery {
  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 }
  }

  class NoServiceDiscovery extends ServiceDiscovery with SavingUsedServiceDiscovery {

    def discover[T <: Service](serviceName: Name[Service]): T =
      throw new IllegalArgumentException(s"Service with name $serviceName is unknown")
  }
}