这段代码如何优化

Question

这段代码如何优化呢？

var sortFlag0 = false;

var sortFlag1 = false;

var sortFlag2 = false;

$scope.sortData = function (id) {

    console.log(id);

    if (id == 0) {

        sortFlag0 = !sortFlag0;

        if (sortFlag0) { //升序

            $("#sortImg" + id).attr("src", "../../img/shsort.png");

        } else if (!sortFlag0) { //降序

            $("#sortImg" + id).attr("src", "../../img/jsort.png");

        }

    } else if (id == 1) {

        sortFlag1 = !sortFlag1;

        if (sortFlag1) { //升序

            $("#sortImg" + id).attr("src", "../../img/shsort.png");

        } else if (!sortFlag1) { //降序

            $("#sortImg" + id).attr("src", "../../img/jsort.png");

        }

    } else if (id == 2) {

        sortFlag2 = !sortFlag2;

        if (sortFlag2) { //升序

            $("#sortImg" + id).attr("src", "../../img/shsort.png");

        } else if (!sortFlag2) { //降序

            $("#sortImg" + id).attr("src", "../../img/jsort.png");

        }

    }

}