Revert "Container-jersey2: Scala code replaced with Java"

4 files changed, 239 insertions, 0 deletions

diff --git a/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ComponentGraphProvider.scala b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ComponentGraphProvider.scala
new file mode 100644
index 00000000000..cabde3680a4
--- /dev/null
+++ b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ComponentGraphProvider.scala
@@ -0,0 +1,40 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.servlet.jersey
+
+import javax.inject.Singleton
+
+import com.yahoo.container.di.config.{ResolveDependencyException, RestApiContext}
+import com.yahoo.container.jaxrs.annotation.Component
+import org.glassfish.hk2.api.{ServiceHandle, Injectee, InjectionResolver}
+
+/**
+ * Resolves jdisc container components for jersey 2 components.
+ * Similar to Gjoran's ComponentGraphProvider for jersey 1.
+ * @author tonytv
+ */
+@Singleton //jersey2 requirement: InjectionResolvers must be in the Singleton scope
+class ComponentGraphProvider(injectables: Traversable[RestApiContext.Injectable]) extends InjectionResolver[Component] {
+  override def resolve(injectee: Injectee, root: ServiceHandle[_]): AnyRef = {
+    val wantedClass = injectee.getRequiredType match {
+      case c: Class[_] => c
+      case unsupported => throw new UnsupportedOperationException("Only classes are supported, got " + unsupported)
+    }
+
+    val componentsWithMatchingType = injectables.filter{ injectable =>
+        wantedClass.isInstance(injectable.instance) }
+
+    val injectionDescription =
+      s"class '$wantedClass' to inject into Jersey resource/provider '${injectee.getInjecteeClass}')"
+
+    if (componentsWithMatchingType.size > 1)
+      throw new ResolveDependencyException(s"Multiple components found of $injectionDescription: " +
+        componentsWithMatchingType.map(_.id).mkString(","))
+
+    componentsWithMatchingType.headOption.map(_.instance).getOrElse {
+      throw new ResolveDependencyException(s"Could not find a component of $injectionDescription.")
+    }
+  }
+
+  override def isMethodParameterIndicator: Boolean = true
+  override def isConstructorParameterIndicator: Boolean = true
+}
diff --git a/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyApplication.scala b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyApplication.scala
new file mode 100644
index 00000000000..eea41003984
--- /dev/null
+++ b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyApplication.scala
@@ -0,0 +1,16 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.servlet.jersey
+
+import javax.ws.rs.core.Application
+
+import scala.collection.JavaConverters._
+
+/**
+ * @author tonytv
+ */
+class JerseyApplication(resourcesAndProviderClasses: Set[Class[_]]) extends Application {
+  private val classes: java.util.Set[Class[_]] = resourcesAndProviderClasses.asJava
+
+  override def getClasses = classes
+  override def getSingletons = super.getSingletons
+}
diff --git a/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyServletProvider.scala b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyServletProvider.scala
new file mode 100644
index 00000000000..f0eff54dc16
--- /dev/null
+++ b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/JerseyServletProvider.scala
@@ -0,0 +1,109 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.servlet.jersey
+
+import java.io.{IOException, InputStream}
+
+import com.fasterxml.jackson.databind.ObjectMapper
+import com.fasterxml.jackson.datatype.jdk8.Jdk8Module
+import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule
+import com.fasterxml.jackson.jaxrs.json.JacksonJaxbJsonProvider
+import com.yahoo.container.di.componentgraph.Provider
+import com.yahoo.container.di.config.RestApiContext
+import com.yahoo.container.di.config.RestApiContext.BundleInfo
+import com.yahoo.container.jaxrs.annotation.Component
+import com.yahoo.container.servlet.jersey.util.ResourceConfigUtil.registerComponent
+import org.eclipse.jetty.servlet.ServletHolder
+import org.glassfish.hk2.api.{InjectionResolver, TypeLiteral}
+import org.glassfish.hk2.utilities.Binder
+import org.glassfish.hk2.utilities.binding.AbstractBinder
+import org.glassfish.jersey.media.multipart.MultiPartFeature
+import org.glassfish.jersey.server.ResourceConfig
+import org.glassfish.jersey.servlet.ServletContainer
+import org.objectweb.asm.ClassReader
+
+import scala.collection.JavaConverters._
+import scala.util.control.Exception
+
+
+/**
+ * @author tonytv
+ */
+class JerseyServletProvider(restApiContext: RestApiContext) extends Provider[ServletHolder] {
+  private val jerseyServletHolder = new ServletHolder(new ServletContainer(resourceConfig(restApiContext)))
+
+  private def resourceConfig(restApiContext: RestApiContext) = {
+    val resourceConfig = ResourceConfig.forApplication(
+      new JerseyApplication(resourcesAndProviders(restApiContext.getBundles.asScala)))
+
+    registerComponent(resourceConfig, componentInjectorBinder(restApiContext))
+    registerComponent(resourceConfig, jacksonDatatypeJdk8Provider)
+    resourceConfig.register(classOf[MultiPartFeature])
+
+    resourceConfig
+  }
+
+  def resourcesAndProviders(bundles: Traversable[BundleInfo]) =
+    (for {
+      bundle <- bundles.view
+      classEntry <- bundle.getClassEntries.asScala
+      className <- detectResourceOrProvider(bundle.classLoader, classEntry)
+    } yield loadClass(bundle.symbolicName, bundle.classLoader, className)).toSet
+
+
+  def detectResourceOrProvider(bundleClassLoader: ClassLoader, classEntry: String): Option[String] = {
+    using(getResourceAsStream(bundleClassLoader, classEntry)) { inputStream =>
+      val visitor = ResourceOrProviderClassVisitor.visit(new ClassReader(inputStream))
+      visitor.getJerseyClassName
+    }
+  }
+
+  private def getResourceAsStream(bundleClassLoader: ClassLoader, classEntry: String) = {
+    bundleClassLoader.getResourceAsStream(classEntry) match {
+      case null => throw new RuntimeException(s"No entry $classEntry in bundle $bundleClassLoader")
+      case stream => stream
+    }
+
+  }
+
+  def using[T <: InputStream, R](stream: T)(f: T => R): R = {
+    try {
+      f(stream)
+    } finally {
+      Exception.ignoring(classOf[IOException]) {
+        stream.close()
+      }
+    }
+  }
+
+  def loadClass(bundleSymbolicName: String, classLoader: ClassLoader, className: String) = {
+    try {
+      classLoader.loadClass(className)
+    } catch {
+      case e: Exception => throw new RuntimeException(s"Failed loading class $className from bundle $bundleSymbolicName", e)
+    }
+  }
+
+  def componentInjectorBinder(restApiContext: RestApiContext): Binder = {
+    val componentGraphProvider = new ComponentGraphProvider(restApiContext.getInjectableComponents.asScala)
+    val componentAnnotationType = new TypeLiteral[InjectionResolver[Component]] {}
+
+    new AbstractBinder {
+      override def configure() {
+        bind(componentGraphProvider).to(componentAnnotationType)
+      }
+    }
+  }
+
+  def jacksonDatatypeJdk8Provider: JacksonJaxbJsonProvider = {
+    val provider = new JacksonJaxbJsonProvider()
+    provider.setMapper(
+      new ObjectMapper()
+        .registerModule(new Jdk8Module)
+        .registerModule(new JavaTimeModule))
+    provider
+  }
+
+  override def get() = jerseyServletHolder
+  override def deconstruct() {}
+}
+
diff --git a/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ResourceOrProviderClassVisitor.scala b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ResourceOrProviderClassVisitor.scala
new file mode 100644
index 00000000000..c015f11360e
--- /dev/null
+++ b/container-jersey2/src/main/scala/com/yahoo/container/servlet/jersey/ResourceOrProviderClassVisitor.scala
@@ -0,0 +1,74 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.servlet.jersey
+
+import javax.ws.rs.Path
+import javax.ws.rs.ext.Provider
+
+import org.objectweb.asm.{ClassVisitor, Opcodes, Type, AnnotationVisitor, ClassReader}
+
+
+/**
+ * @author tonytv
+ */
+class ResourceOrProviderClassVisitor private () extends ClassVisitor(Opcodes.ASM5) {
+  private var className: String = null
+  private var isPublic: Boolean = false
+  private var isAbstract = false
+
+  private var isInnerClass: Boolean = false
+  private var isStatic: Boolean = false
+
+  private var isAnnotated: Boolean = false
+
+  def getJerseyClassName: Option[String] = {
+    if (isJerseyClass) Some(getClassName)
+    else None
+  }
+
+  def isJerseyClass: Boolean = {
+    isAnnotated && isPublic && !isAbstract &&
+      (!isInnerClass || isStatic)
+  }
+
+  def getClassName = {
+    assert (className != null)
+    Type.getObjectType(className).getClassName
+  }
+
+  override def visit(version: Int, access: Int, name: String, signature: String, superName: String, interfaces: Array[String]) {
+    isPublic = ResourceOrProviderClassVisitor.isPublic(access)
+    className = name
+    isAbstract =  ResourceOrProviderClassVisitor.isAbstract(access)
+  }
+
+  override def visitInnerClass(name: String, outerName: String, innerName: String, access: Int) {
+    assert (className != null)
+
+    if (name == className) {
+      isInnerClass = true
+      isStatic = ResourceOrProviderClassVisitor.isStatic(access)
+    }
+  }
+
+  override def visitAnnotation(desc: String, visible: Boolean): AnnotationVisitor = {
+    isAnnotated |= ResourceOrProviderClassVisitor.annotationClassDescriptors(desc)
+    null
+  }
+}
+
+
+object ResourceOrProviderClassVisitor {
+  val annotationClassDescriptors = Set(classOf[Path], classOf[Provider]) map Type.getDescriptor
+
+  def isPublic = isSet(Opcodes.ACC_PUBLIC) _
+  def isStatic = isSet(Opcodes.ACC_STATIC) _
+  def isAbstract = isSet(Opcodes.ACC_ABSTRACT) _
+
+  private def isSet(bits: Int)(access: Int): Boolean = (access & bits) == bits
+
+  def visit(classReader: ClassReader): ResourceOrProviderClassVisitor = {
+    val visitor = new ResourceOrProviderClassVisitor
+    classReader.accept(visitor, ClassReader.SKIP_DEBUG | ClassReader.SKIP_CODE | ClassReader.SKIP_FRAMES)
+    visitor
+  }
+}