InfoPublishServiceImpl.java 3.88 KB
package com.bsth.service.publish.impl;

import com.bsth.entity.publish.ScanStatictic;
import com.bsth.service.publish.InfoPublishService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.*;

//@Service
public class InfoPublishServiceImpl implements InfoPublishService {

    @Autowired
    @Qualifier("infoPublishJdbcTemplate")
    private JdbcTemplate infoPublishJdbcTemplate;

    @Override
    public Map<String, Object> list(Map<String, String> map, int page, int size, String order, int direction) {
        Map<String, Object> result = new HashMap<>();
        StringBuilder query = new StringBuilder("SELECT ? rq, physicalCode physical_code, stationName station_name,count(1) `count` FROM bsth_h_scan where 1 = 1");
        List<Object> params = new ArrayList<>();
        params.add(map.get("rq"));
        for (Map.Entry<String, String> entry : map.entrySet()) {
            String key = entry.getKey(), value = entry.getValue();
            if (StringUtils.isEmpty(key) || ",page,size,order,direction,".indexOf(String.format(",%s,", key)) > -1) {
                continue;
            }
            if ("rq".equals(key)) {
                query.append(" and rq between ? and ?");
                String[] rqs = value.split("-");
                if (rqs.length != 2) {
                    throw new RuntimeException("异常的起止日期");
                }
                for (String rq : rqs) {
                    params.add(rq);
                }
            } else {
                query.append(" and ").append(key).append(" like concat(?, '%')");
                params.add(value);
            }
        }
        query.append(" GROUP BY physicalCode, stationName");
        StringBuilder count = new StringBuilder("SELECT COUNT(*) FROM (").append(query).append(") t");
        Integer total = infoPublishJdbcTemplate.queryForObject(count.toString(), params.toArray(), Integer.class);
        query.append(" ORDER BY physicalCode LIMIT ").append(page * size).append(",").append(size);
        List<ScanStatictic> data = infoPublishJdbcTemplate.query(query.toString(), params.toArray(), BeanPropertyRowMapper.newInstance(ScanStatictic.class));
        result.put("list", data);
        result.put("totalPages", total % size == 0 ? total / size - 1  : total / size);

        return result;
    }

    @Override
    public Map<String, Object> export(Map<String, String> map) {
        Map<String, Object> result = new HashMap<>();
        StringBuilder query = new StringBuilder("SELECT ? rq, physicalCode, stationName,count(1) `count` FROM bsth_h_scan where 1 = 1");
        List<Object> params = new ArrayList<>();
        params.add(map.get("rq"));
        for (Map.Entry<String, String> entry : map.entrySet()) {
            String key = entry.getKey(), value = entry.getValue();
            if ("rq".equals(key)) {
                query.append(" and rq between ? and ?");
                String[] rqs = value.split("-");
                if (rqs.length != 2) {
                    throw new RuntimeException("异常的起止日期");
                }
                for (String rq : rqs) {
                    params.add(rq);
                }
            } else {
                query.append(" and ").append(key).append(" like concat(?, '%')");
                params.add(value);
            }
        }
        query.append(" GROUP BY physicalCode, stationName");
        query.append(" ORDER BY physicalCode");
        List<Iterator<?>> iterators = new ArrayList<>();
        iterators.add(infoPublishJdbcTemplate.queryForList(query.toString(), params.toArray()).iterator());
        result.put("list", iterators);

        return result;
    }
}