有限状態マシンを使っているときに特定の状態に変化したときに呼ばれたい関数をすっきりと書くにはどのように記述するのがベストでしょうか

　void OnStateEnter ()　{
　
　}

という関数を特定の状態の時に呼び出してほしい用に記述したい
Attributeか何かをうまく使って指示できませんかね

[StateIs(Running)]
　void OnStateEnter ()　{
　
　}

こんな感じで