实验4 基于文件的简易记账程序

package cn.edu.qtech; import java.io.*; import java.util.*; public class ConsumptionRecord { File f=new File("D:\\record.txt"); ArrayList<String> records=new ArrayList<String>(); //String records[]=new String[100]; public void readFromFile() { try{ FileReader inOne=new FileReader(f); BufferedReader inTwo= new BufferedReader(inOne); String s=null; while((s=inTwo.readLine())!=null) { // inTwo.read(); // inTwo.readLine(); records.add(s); //inTwo.read(); //System.out.println(s); } inOne.close(); inTwo.close(); } catch(IOException e) { System.out.println(e); } } public void writeToFile(String oneRecord){ if(f.exists()) { FileWriter outOne=null; try{ outOne=new FileWriter("D:\\record.txt",true); BufferedWriter outTwo= new BufferedWriter(outOne); // for(String str:oneRecord) // {outTwo.write(str); // } // readFromFile(); records.add(oneRecord); Iterator<String> iter=records.iterator(); while(iter.hasNext()){ String te=iter.next(); } outTwo.write(oneRecord); outTwo.newLine(); //outTwo.newLine(); outTwo.close(); outOne.close(); } catch(IOException e) { System.out.println(e); } } else { try{ FileWriter outOne=new FileWriter(f); BufferedWriter outTwo= new BufferedWriter(outOne); //for(String str:oneRecord) // {outTwo.write(str); // } outTwo.newLine(); outTwo.write(oneRecord); outTwo.newLine(); outTwo.close(); outOne.close(); } catch(IOException e) { System.out.println(e); } // try{f.createNewFile();} // catch(IOException e){System.out.println(e);} } } public void makeRecords(){ String s1,s2,s3; String s; System.out.println("请输入消费日期，格式XXXX-XX-XX,-1结束"); Scanner hi=new Scanner(System.in); s1=hi.next(); String yi="-1"; while(!s1.equals(yi)){ System.out.println("请输入消费条目名称："); Scanner ha=new Scanner(System.in); s2=ha.next(); System.out.println("请输入消费钱数："); Scanner he=new Scanner(System.in); s3=he.next(); s=s1+" "+s2+" "+s3; writeToFile(s); System.out.println("请输入消费日期，格式XXXX-XX-XX,-1结束"); s1=hi.next(); } } public void searchData(){ int x=0; while(x!=-1){ String s4; int flag=0; System.out.println("请输入要查询的消费条目名称,-1结束"); Scanner hi=new Scanner(System.in); s4=hi.next(); if(s4.equals("-1")){x=-1;} if(x!=-1){ Iterator<String> iter=records.iterator(); while(iter.hasNext()){ String te=iter.next(); // s5=te.substring(11, 14); if( te.contains(s4)) { System.out.println("消费条目"+s4+"信息"); System.out.println(te); flag=1; } } if(flag==0){ System.out.println("您要查找的消费条目不存在！");} } } } public static void main(String[] args) { // TODO Auto-generated method stub int n; ConsumptionRecord cr=new ConsumptionRecord(); System.out.println("请输入操作类型：1.添加新消费记录； 2.查询消费条目信息；-1退出"); Scanner hi=new Scanner(System.in); n=hi.nextInt(); while(n!=-1){ if(n==1){ cr.makeRecords(); } cr.readFromFile(); if(n==2){ cr.searchData(); }System.out.println("请输入操作类型：1.添加新消费记录； 2.查询消费条目信息；-1退出"); n=hi.nextInt(); } } }