package com.wedo.bvision.schedule; import java.io.File;import java.io.FileWriter;import java.io.IOException;import java.io.PrintWriter; import org.apache.commons.io.FileUtils;import org.apache.commons.io.LineIterator;import org.apache.commons.lang.StringUtils;import org.junit.Test;import org.slf4j.Logger;import org.slf4j.LoggerFactory; /** * Created by wedo on 25/9/2017. */public class BugGender { private final static Logger logger = LoggerFactory.getLogger(BugGender.class); @Test public void test(){ LineIterator it; String filePath = "C:\\Users\\usr\\Desktop\\杨嘴小学.csv"; File dirFile = new File(filePath); String fileUpdate = "C:\\Users\\usr\\Desktop\\genderUpate.txt"; File updateFile = new File(fileUpdate); if (!updateFile.exists()){ try { updateFile.createNewFile(); } catch (IOException e) { e.printStackTrace(); } } logger.info("Begin import file:{}", filePath); try { it = FileUtils.lineIterator(dirFile, "UTF-8"); PrintWriter pw = new PrintWriter(new FileWriter(fileUpdate,true),true); if (it.hasNext()) { String firstLine = it.nextLine(); if (!firstLine.startsWith("name")) { logger.error("Invalid line, please follow template, add first line"); } } while (it.hasNext()) { try { String line = it.nextLine(); String checkUserInfoUpadate = CheckUserInfohandleLine(line); String userProfileUpdate = UserProfilehandleLine(line); pw.write(checkUserInfoUpadate); pw.write(userProfileUpdate); }catch (Exception e){ e.printStackTrace(); } } pw.close(); } catch (IOException e) { e.printStackTrace(); } } private String CheckUserInfohandleLine(String line) { String[] elements = line.split(","); //性别 if (StringUtils.isNotEmpty(elements[3]) && elements[3].indexOf("女")>-1) { return "update sight_check_user_info set gendar = 2 WHERE campaign_sequence=‘"+elements[0].trim()+"‘;\r\n"; } return "update sight_check_user_info set gendar = 1 WHERE campaign_sequence=‘"+elements[0].trim()+"‘;\r\n"; } private String UserProfilehandleLine(String line) { String[] elements = line.split(","); String userId = elements[1].trim(); //性别 if (StringUtils.isNotEmpty(elements[3]) && elements[3].indexOf("女")>-1) { return "update user_profile set gender = 2 WHERE user_id=‘"+userId+"‘;\r\n"; } return "update user_profile set gender = 1 WHERE user_id=‘"+userId+"‘;\r\n"; }}
时间: 2024-10-10 08:31:44