find by regex

RegexMatch.scala

package org.snippets

object RegexMatch {

  val LOCATION_JU_MAIN_CAT_PREFIX = "a"
  val LOCATION_JU_WAP_CAT_PREFIX = "b"
  val LOCATION_JU_TAOBAO_CAT_PREFIX = "c"
  val LOCATION_JU_TMALL_CAT_PREFIX = "d"

  val juCatRE = s"$LOCATION_JU_MAIN_CAT_PREFIX([0-9]+)".r
  val wapCatRE = s"$LOCATION_JU_WAP_CAT_PREFIX([0-9]+)".r
  val taobaoCatRE = s"$LOCATION_JU_TAOBAO_CAT_PREFIX([0-9]+)".r
  val tmallCatRE = s"$LOCATION_JU_TMALL_CAT_PREFIX([0-9]+)".r

  def unapply(location: String): Option[Long] =
    location match {
      case juCatRE(catId) => Some(catId.toLong)
      case wapCatRE(catId) => Some(catId.toLong)
      case taobaoCatRE(catId) => Some(catId.toLong)
      case tmallCatRE(catId) => Some(catId.toLong)
    }

  def fnd(location: String): Option[Long] =
    Seq(juCatRE, wapCatRE, taobaoCatRE, tmallCatRE)
      .flatMap(_.unapplySeq(location))
      .headOption
      .map(_.head.toLong)

  def main(args: Array[String]) {
    assert(RegexMatch.unapply("a123").contains(123))
    assert(RegexMatch.fnd("a123").contains(123))
    assert(RegexMatch.fnd("b3").contains(3))
    assert(RegexMatch.fnd("b").isEmpty)
    assert(RegexMatch.fnd("a123b").isEmpty)
  }
}