angular.jsでdirectiveからng-classを使ってcssを切り替える方法

軽くハマったのでひな形をメモしておきます

javascript

var app = angular.module("myApp",[]);

app.controller("ControllerIndex", function($scope) {
});

app.directive("csschange", function () {
  return function ($scope, element, attrs) {
    $scope.class_name = "class01 class02 class03";
  }
});

html

<div ng-controller="ControllerMain">
  <div csschange ng-class="class_name">
</div>

以上です