这段代码如何优化呢？

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