Fix arg name extraction with _ and $

spotify · Dec 4, 2019 · a161263 · a161263
1 parent 3f73209
commit a161263
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/scala/com/spotify/scio/AnnotationTypeInjector.scala b/src/main/scala/com/spotify/scio/AnnotationTypeInjector.scala
@@ -35,7 +35,7 @@ import scala.collection.mutable
 object AnnotationTypeInjector {
   private val Log = Logger.getInstance(classOf[AnnotationTypeInjector])
   private val CaseClassArgs = """case\s+class\s+[^(]+\((.*)\).*""".r
-  private val TypeArg = """[a-zA-Z0-9]+\s*:\s*[a-zA-Z0-9._]+([\[(](.*?)[)\]]+)?""".r
+  private val TypeArg = """[a-zA-Z0-9_$]+\s*:\s*[a-zA-Z0-9._$]+([\[(](.*?)[)\]]+)?""".r
   private val AlertEveryMissedXInvocations = 5
 
   def getApplyPropsSignature(caseClasses: Option[String]): Seq[String] =