Is there a way to do conditional group by in spark 2.1.1?

[kant kodali]

Hi All,

Is there a way to do conditional group by in spark 2.1.1? other words, I
want to do something like this

if (field1 == "foo") {
       df.groupBy(field1)
} else if (field2 == "bar")
      df.groupBy(field2)

Thanks