这段代码如何优化

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");
        }
    }
}

Answer 1

var sortFlag=[false,false,false];
$scope.sortData = function (id) {
    sortFlag[id] = !sortFlag[id];
    if (sortFlag[id]) { //升序
        $("#sortImg" + id).attr("src", "../../img/shsort.png");
    } else if (!sortFlag[id]) { //降序
        $("#sortImg" + id).attr("src", "../../img/jsort.png");
    }
}

//三目运算
var sortFlag = [false, false, false];
$scope.sortData = function (id) {
    sortFlag[id] = !sortFlag[id];
    $("#sortImg" + id).attr("src", "../../img/" + (sortFlag[id] ? 'sh' : 'j') + "sort.png");
}