openwiki: DumpExcelHandler

import cn.hutool.core.convert.Convert;
import cn.hutool.core.io.FileUtil;
import cn.hutool.poi.excel.ExcelWriter;
import com.iqibai.gamedata.model.ResultUser;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Component;

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

@Component
@Slf4j
public class DumpExcelHandler {

    @Autowired
    private MongoTemplate mongoTemplate;

    public void processExcel(){

        String filePrefix = "/data/work/user/redis-user";
        for (int i = 1; i <= 4; i++) {
            String file = filePrefix + i +".txt";
            List<String> userIds = FileUtil.readLines(file,"utf-8");
            List<ResultUser> resultUsers = new ArrayList<>();
            for(String userId:userIds){
                Integer userIdInt = Convert.toInt(userId,0);
                if(userIdInt == 0){
                    continue;
                }
                Query findQuery = new Query(Criteria.where("userId").is(userIdInt));
                ResultUser findUser = mongoTemplate.findOne(findQuery,ResultUser.class);
                if(findUser == null){
                    continue;
                }
                resultUsers.add(findUser);
            }
            ExcelWriter excelWriter = new ExcelWriter(filePrefix+i+".xls");
            excelWriter.write(resultUsers);
            excelWriter.flush();
            log.info("end process {}",file);

        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 当发现某一个孩子身上的问题让你觉得不得不面对,甚至必须要采取措施给予纠正帮扶的时候,我们需要的不是解决问题的方法,...
    华妈2008阅读 409评论 0 0
  • 佛法对感情的看法是什么?这个每一个宗教,不是讲爱就是讲博爱,要不然就是讲慈悲。那么我们要从人的最底处的地方...
    周肖阅读 567评论 0 0
  • 绕着小池塘,溜达了一圈,仔细挑选了几朵我认为最漂亮的荷花,小心翼翼的把茶叶放入花心。此时已经是晚霞减退,弯弯的月牙...
    ad8d261a83e4阅读 249评论 2 0

友情链接更多精彩内容