package com.ks.lib.common.util;

import org.springframework.data.domain.Sort;

import java.util.ArrayList;
import java.util.List;

public class MongoDBQueryUtil {


    /**
     * 转换sql语句的排序
     *
     * @param sortList
     * @return
     */
    public static Sort.Order[] convertSQLSort(List<String> sortList) {
        List<Sort.Order> sList = new ArrayList<>();

        for (String sort : sortList) {
            String key = "";
            if (sort.trim().contains(" ")) {
                key = sort.trim().split(" ")[0];
            } else {
                key = sort.trim();
            }

            if (sort.toLowerCase().endsWith(" desc")) {
                sList.add(Sort.Order.desc(key));
            } else {
                sList.add(Sort.Order.asc(key));
            }
        }

        Sort.Order[] sorts = new Sort.Order[sortList.size()];
        sList.toArray(sorts);
        return sorts;
    }

}