package io.renren.utils;

import io.renren.utils.xss.SQLFilter;
import org.springframework.util.StringUtils;

import java.util.LinkedHashMap;
import java.util.Map;

/**
 * 查询参数
 *
 * @author chenshun
 * @email sunlightcs@gmail.com
 * @date 2017-03-14 23:15
 */
public class Query extends LinkedHashMap<String, Object> {
    private static final long serialVersionUID = 1L;
    //当前页码
    private int page = 1;
    //每页条数
    private int limit = 10;

    public Query(Map<String, Object> params) {
        this.putAll(params);


        //分页参数
        if (!StringUtils.isEmpty(params.get("page"))) {
            this.page = Integer.parseInt(params.get("page").toString());
        }
        if (!StringUtils.isEmpty(params.get("limit"))) {
            this.limit = Integer.parseInt(params.get("limit").toString());
        }
        this.put("offset", (page - 1) * limit);
        this.put("page", page);
        this.put("limit", limit);

        //防止SQL注入（因为sidx、order是通过拼接SQL实现排序的，会有SQL注入风险）
        if (!StringUtils.isEmpty(params.get("sidx"))) {
            String sidx = params.get("sidx").toString();
            this.put("sidx", SQLFilter.sqlInject(sidx));
        }
        if (!StringUtils.isEmpty(params.get("order"))) {
            String order = params.get("order").toString();
            this.put("order", SQLFilter.sqlInject(order));
        }
    }


    public int getPage() {
        return page;
    }

    public void setPage(int page) {
        this.page = page;
    }

    public int getLimit() {
        return limit;
    }

    public void setLimit(int limit) {
        this.limit = limit;
    }
}
