Repeat execution of java method using values in spark dataset

Question

I have a method that contains if else statements to run code based on a string value that is passed to a method.

private Dataset getData(Dataset