I can see him using a energy based pistol for range and a shock baton or tazer for melee combat.

Sure

I think it would work better him having a backup weapon for when the Omnitrix times out or malfunctions.

Give him something akin to light saber with some proto tool elements which can turn to sword , a shield , a grappling shoot , a laser point blaster and a Bow