For Air Hockey, the mallet on puck collision should act like the collision of a pool cue stick with the white cue ball.
See this label in my doc:
It should be that simple, and you should not need any ball collision procedure since you don't have freely rolling balls, only the single puck.