DataFrame.transform - Spark Function Composition - Functions post refactor

ExampleTransformLibRefactor.scala

def sumAmounts(by: Column*): DataFrame => DataFrame =
  df => df.groupBy(by: _*).agg(sum(col("amount")))

def extractPayerBeneficiary(columnName: String): DataFrame => DataFrame =
  df =>
    df.withColumn(
      s"${columnName}_payer",
      regexp_extract(
        col(columnName),
        "paid by ([A-Z])",
        1
      )
    ).withColumn(
      s"${columnName}_beneficiary",
      regexp_extract(
        col(columnName),
        "to ([A-Z])",
        1
      )
    )