streaming.core.datasource.SourceTypeRegistry.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of streamingpro-mlsql-spark_2.4 Show documentation

The newest version!

package streaming.core.datasource

import scala.collection.JavaConverters._
import scala.collection.mutable

object SourceTypeRegistry {
  private val registry = new java.util.concurrent.ConcurrentHashMap[String, mutable.Set[SourceInfo]]()

  def register(name: String, obj: SourceInfo) = {

    if (!registry.containsKey(name)) {
      registry.put(name, new mutable.HashSet[SourceInfo]())
    }
    registry.get(name) += obj
  }

  def sources = {
    registry.keys().asScala.toSeq
  }
}