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