smartsearch/src/main/java/com/bowintek/practice/services/impl/WellInfoServiceImpl.java

package com.bowintek.practice.services.impl;

import com.bowintek.practice.mapper.SrSubjectMapper;
import com.bowintek.practice.mapper.WellBasicInformationMapper;
import com.bowintek.practice.mapper.cquery.WellInfoCQuery;
import com.bowintek.practice.model.WellBasicInformation;
import com.bowintek.practice.model.WellBasicInformationExample;
import com.bowintek.practice.services.service.WellInfoService;
import com.bowintek.practice.util.multipdb.DBTypeEnum;
import com.bowintek.practice.util.multipdb.SwitchDataSource;
import com.bowintek.practice.vo.SubjectVo;
import com.bowintek.practice.vo.query.WellInfoParams;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.List;

import static com.bowintek.practice.util.Constant.BOREHOLE_INTER_TYPE;
import static com.bowintek.practice.util.Constant.FACT_CURRENT_STATE;

@Component
public class WellInfoServiceImpl implements WellInfoService {

    @Autowired
    private WellBasicInformationMapper wellBasicInformationMapper;
    @Autowired
    private WellInfoCQuery wellInfoCQuery;

    @Override
    @SwitchDataSource(DBTypeEnum.POSTGRE)
    public PageInfo<HashMap<String, Object>> getList(Integer page, Integer rows, WellInfoParams params) {
        PageHelper.startPage(page, rows);

        List<HashMap<String,Object>> dataList = wellInfoCQuery.getWellInfoList(params);

        PageInfo< HashMap<String, Object>> result = new PageInfo(dataList);

        result.getList().forEach(it->{
            if(FACT_CURRENT_STATE.keySet().contains(it.get("current_state"))){
                it.put("current_state_name",FACT_CURRENT_STATE.get(it.get("current_state")));
            }
        });
        return result;
    }

    @Override
    @SwitchDataSource(DBTypeEnum.POSTGRE)
    public HashMap<String,Object> getWellInfo(String well_id){
        return wellInfoCQuery.getWellInfo(well_id);
    }

    @Override
    @SwitchDataSource(DBTypeEnum.POSTGRE)
    public List<HashMap<String,Object>> selectBoreholeInterList(String well_id){
        List<HashMap<String,Object>> dataList = wellInfoCQuery.selectBoreholeInterList(well_id);

        dataList.forEach(it->{
            if(BOREHOLE_INTER_TYPE.keySet().contains(it.get("borehole_inter_type"))){
                it.put("borehole_inter_type_name",BOREHOLE_INTER_TYPE.get(it.get("borehole_inter_type")));
            }
        });
        return dataList;
    }

    @Override
    @SwitchDataSource(DBTypeEnum.POSTGRE)
    public List<HashMap<String,Object>> selectTestHistoryList(String well_id){
        return wellInfoCQuery.selectTestHistoryList(well_id);
    }

    @Override
    @SwitchDataSource(DBTypeEnum.POSTGRE)
    public List<HashMap<String,Object>> selectAnalyticalAssaysList(String well_id){
        return wellInfoCQuery.selectAnalyticalAssaysList(well_id);
    }
}