com.github.pjfanning.jackson.reflection.ScalaReflectionAnnotationIntrospector.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jackson-scala3-reflection-extensions_3 Show documentation

Jackson scala3 support that uses scala3-reflection to get type info

The newest version!

package com.github.pjfanning.jackson.reflection

import co.blocke.scala_reflection.RType
import co.blocke.scala_reflection.info.{ObjectInfo, ScalaClassInfo, SealedTraitInfo}
import com.fasterxml.jackson.core.Version
import com.fasterxml.jackson.databind.introspect.{Annotated, JacksonAnnotationIntrospector}
import com.fasterxml.jackson.databind.jsontype.NamedType
import org.slf4j.LoggerFactory

import scala.jdk.CollectionConverters.*
import scala.util.control.NonFatal

class ScalaReflectionAnnotationIntrospector extends JacksonAnnotationIntrospector {
  private val logger = LoggerFactory.getLogger(classOf[ScalaReflectionAnnotationIntrospector])

  override def version(): Version = JacksonModule.version

  override def findSubtypes(a: Annotated): java.util.List[NamedType] = {
    try {
      val rtype = RTypeCache.getRType(a.getRawType)
      rtype match {
        case traitInfo: SealedTraitInfo =>
          traitInfo.children
            .map(ct => new NamedType(getClass(ct)))
            .toSeq.asJava
        case classInfo: ScalaClassInfo =>
          classInfo.children
            .map(ct => new NamedType(getClass(ct)))
            .toSeq.asJava
        case _ => None.orNull
      }
    } catch {
      case NonFatal(t) => {
        logger.warn(s"Failed to findSubtypes in ${a.getRawType}: $t")
        None.orNull
      }
    }
  }

  private def getClass(rtype: RType): Class[_] = rtype match {
    case objectInfo: ObjectInfo => getCompanionObjectClass(objectInfo.infoClass)
    case rt => rt.infoClass
  }

  private def getCompanionObjectClass(cls: Class[_]): Class[_] = {
    val cn = cls.getName
    if (cn.endsWith("$")) {
      cls
    } else {
      Class.forName(cn + '$', true, Thread.currentThread().getContextClassLoader)
    }
  }
}