I have requirement to Route the assignment to a Users WL or WB based on some condition

Approach 1 : Make change in the flow and based on a decision shape route it two different assignment (with one configured to WL and other to WB)

Approach 2: Keep only one assignment shape and write a new router activity and handle the logic over there in the activity .

Which do you think is a better approach.

Thanks in Advance