Dont even need a stun, need a slowdown mechanic or a stamina hit when you are hit.
Perhaps make that a function of armor.
Lighter armor less of a stamina hit than heavier armor, thus giving a balance to the two.
Take more health damage but dont lose stamina when hit, or take more...