야구선수(투수와타자) 관리 프로그램(상속의 개념 응용)-ArrayList

앞에 포스팅 한 예제(배열로 데이터 관리)를

(ArrayList로 데이터 관리)로 바꾼 프로그램

mainClass

package main; import java.util.Scanner; import dao.daoClass; import file.fileClass; public class mainClass { public static void main(String[] args) { int choice; daoClass dao = new daoClass(); while(true) { Scanner sc = new Scanner(System.in); System.out.println("====================================="); System.out.println("1.선수 정보 입력"); System.out.println("2.선수 정보 삭제"); System.out.println("3.선수 정보 검색"); System.out.println("4.선수 정보 수정"); System.out.println("5.센수 데이터 모두 출력"); System.out.println("6.파일에 저장"); System.out.println("7.파일 데이터 불러오기"); System.out.println("8.정렬하기"); System.out.println("9.종료"); System.out.println("어느 작업을 하시겠습니까? >>"); choice = sc.nextInt(); switch(choice) { case 1://선수 정보 입력 dao.insert(); break; case 2://선수 정보 삭제 dao.delete(); break; case 3://선수 정보 검색 dao.select(); break; case 4://선수 정보 수정 dao.update(); break; case 5://데이터 모두 출력 dao.dataPrint(); break; case 6://파일에 저장 dao.fileSave(); break; case 7://파일 데이터 불러오기 dao.readfile(); break; case 8://정렬 dao.sort(); break; default: System.exit(0); } } } }

daoClass

package dao; import java.util.ArrayList; import java.util.Scanner; import dto.batterClass; import dto.humanClass; import dto.pitcherClass; import file.fileClass; import sort.sortClass; public class daoClass { Scanner sc = new Scanner(System.in); private final int INSERT = 0; private final int DELETE = 1; private final int SELECT = 3; private final int UPDATE = 2; //private humanClass dtoArr[] = new humanClass[20]; //배열대신 ArrayList로 ArrayList<humanClass> dtolist = new ArrayList<humanClass>(); //private int count; private int selectIndex; public daoClass() { //프로그램 시작시 파일을 만들고 시작 fileClass.createFile(); } public void insert() { System.out.print("번호: "); String number = sc.next(); System.out.print("이름: "); String name = sc.next(); System.out.print("나이: "); String age = sc.next(); System.out.print("생년월일: "); String birth = sc.next(); System.out.print("포지션(투수or타자): "); String position = sc.next(); positionChoice(number,name,age,birth ,position, INSERT); } //포지션에 따른 입력 public void positionChoice(String number, String name, String age, String birth, String position,int funcSelect) { //잘못 입력시 처리는 안함 switch(position) { case "투수": //승, 평균자책, 탈삼진, 세이브 System.out.print("승: "); String victories = sc.next(); System.out.print("평균 자책점: "); String ERA = sc.next(); System.out.print("탈삼진: "); String strikeout = sc.next(); System.out.print("세이브: "); String save = sc.next(); if(funcSelect == INSERT) dtolist.add(new pitcherClass(number, name, age, birth, victories, ERA, strikeout, save)); else if(funcSelect == UPDATE) dtolist.set(selectIndex,new pitcherClass(number, name, age, birth, victories, ERA, strikeout, save)); break; case "타자": // 타율, 타점, 홈런, 도루 System.out.print("타율: "); String BA = sc.next(); System.out.print("타점: "); String RBI = sc.next(); System.out.print("홈런: "); String HR = sc.next(); System.out.print("도루: "); String SB = sc.next(); if(funcSelect == INSERT) dtolist.add(new batterClass( number, name, age, birth, BA, RBI, HR, SB)); else if(funcSelect == UPDATE) dtolist.set(selectIndex,new batterClass( number, name, age, birth, BA, RBI, HR, SB)); break; } } public void delete() { //삭제 boolean find=select(); if(find) { dtolist.remove(selectIndex); System.out.println("삭제 완료!!"); }else System.out.println("삭제할 데이터가 없습니다."); } public boolean select() { //검색 String str; boolean check = false; System.out.print("검색할 이름: "); str = sc.next(); for (int i = 0; i < dtolist.size(); i++) { //Null포인터 익셉션발생함 humanClass hcls = dtolist.get(i); try { if (hcls.getName().equals(str)) { selectIndex = i; check = true; break; } }catch(Exception e) { e.printStackTrace(); } } if (check) { System.out.println(dtolist.get(selectIndex).toString()); return true; } else { System.out.println("해당 데이터는 없습니다."); return false; } } public void update() { //수정 boolean find = select(); if(find) { System.out.print("번호: "); String number = sc.next(); System.out.print("이름: "); String name = sc.next(); System.out.print("나이: "); String age = sc.next(); System.out.print("생년월일: "); String birth = sc.next(); instanceOfInsert(number,name,age,birth); System.out.println("수정 완료!!"); }else { System.out.println("수정할 데이터가 없습니다."); } } public void instanceOfInsert(String number, String name, String age, String birth) { //할당받은 인스턴스에 따른 positionChoice 메소드 처리 if( dtolist.get(selectIndex) instanceof batterClass ) //할당받은것이 타자 인스턴스이면 positionChoice(number, name, age, birth,"타자", UPDATE); else if( dtolist.get(selectIndex) instanceof pitcherClass ) //할당받은것이 투수 인스턴스이면 positionChoice(number, name, age, birth,"투수", UPDATE); } public void dataPrint() { if(dtolist.size()==0) { System.out.println("출력할 데이터가 없습니다."); return; } for (int i = 0; i< dtolist.size(); i++) { System.out.println(dtolist.get(i).toString()); } } public void fileSave() { //문자열로 된 데이터들을 파일에 덮어 씌우기(저장) String strArr[]=allDataToString(); fileClass.writeFile(strArr); } public String[] allDataToString() { //모든 데이터를 토큰을 붙여 문자열로 만드는 메소드 String strArr[]=new String[dtolist.size()]; for (int i = 0; i < strArr.length; i++) { //포지션-넘버-이름-나이-생년월일 + 포지션별 정보 token이용해서 붙이기 humanClass hcls = dtolist.get(i); strArr[i]=hcls.getNumber()+ "-" +hcls.getName()+"-"+hcls.getAge()+"-"+hcls.getBirth(); if( hcls instanceof batterClass ) { //할당받은것이 타자 인스턴스이면 batterClass bcls = (batterClass)hcls; strArr[i]=bcls.getPosition()+"-"+ strArr[i]+"-"+bcls.getBA()+"-"+bcls.getRBI()+"-"+bcls.getHR()+"-"+bcls.getSB(); } else if( hcls instanceof pitcherClass ) { //할당받은것이 투수 인스턴스이면 pitcherClass pcls = (pitcherClass)hcls; strArr[i]= pcls.getPosition()+"-"+ strArr[i]+"-"+pcls.getVictories()+"-"+pcls.getERA()+"-"+pcls.getStrikeout()+"-"+pcls.getSave(); } } return strArr; } public void readfile() { //데이터 받아옴 String data[]=fileClass.readFile(); dtolist.clear();// list 초기화 for (int i = 0; i < data.length; i++) { String strSplits[] = data[i].split("-"); if (strSplits[0].equals("타자")) { dtolist.add( new batterClass(strSplits[1],strSplits[2], strSplits[3],strSplits[4],strSplits[5],strSplits[6],strSplits[7],strSplits[8])); } else if (strSplits[0].equals("투수")) { dtolist.add( new pitcherClass(strSplits[1],strSplits[2], strSplits[3],strSplits[4],strSplits[5],strSplits[6],strSplits[7],strSplits[8])); } if(dtolist.get(i)!=null) System.out.println(dtolist.get(i).toString()); } } public void sort() { //정렬 sortClass sort = new sortClass(dtolist); sort.init(); sort.input(); sort.sorting(); for (int i = 0; i< dtolist.size(); i++) { if(dtolist.get(i)!=null) System.out.println(dtolist.get(i).toString()); } } }

dto( humanClass )

package dto; public class humanClass { private String number; private String name; private String age; private String birth; public humanClass() { } public humanClass(String number, String name, String age, String birth) { this.number = number; this.name = name; this.age = age; this.birth = birth; } public String getNumber() { return number; } public void setNumber(String number) { this.number = number; } public String getName() { return name; } public void setName(String name) { this.name = name; } public String getAge() { return age; } public void setAge(String age) { this.age = age; } public String getBirth() { return birth; } public void setBirth(String birth) { this.birth = birth; } @Override public String toString() { return "번호=" + number + ", 이름=" + name + ", 나이=" + age + ", 생년월일=" + birth + ", "; } }

dto( batterClass )

package dto; public class batterClass extends humanClass{ //타자 // 타율, 타점, 홈런, 도루 private final String position = "타자"; private String BA;// Batting Average 타율 private String RBI; //Run Batted In 타점 private String HR; // Home Run홈런 private String SB;// Stolen Bases 도루 public batterClass() { } public batterClass(String number, String name, String age, String birth, String bA, String rBI, String hR, String sB) { super(number,name,age,birth); BA = bA; RBI = rBI; HR = hR; SB = sB; } public String getBA() { return BA; } public void setBA(String bA) { BA = bA; } public String getRBI() { return RBI; } public void setRBI(String rBI) { RBI = rBI; } public String getHR() { return HR; } public void setHR(String hR) { HR = hR; } public String getSB() { return SB; } public void setSB(String sB) { SB = sB; } public String getPosition() { return position; } @Override public String toString() { return "[("+position+") "+super.toString()+"타율=" + BA + ", 타점=" + RBI + ", 홈런=" + HR + ", 도루=" + SB + "]"; } }

dto( pitcherClass )

package dto; public class pitcherClass extends humanClass { //투수 //승, 평균자책, 탈삼진, 세이브 private final String position = "투수"; private String victories; //승 private String ERA; //평균자책점(Earned Run Average) private String strikeout; // 탈삼진(모든 아웃을 탈삼진이라고 함) private String save; //세이브 public pitcherClass() { } public pitcherClass(String number, String name, String age, String birth, String victories, String eRA, String strikeout, String save) { super(number, name, age, birth); this.victories = victories; ERA = eRA; this.strikeout = strikeout; this.save = save; } public String getVictories() { return victories; } public void setVictories(String victories) { this.victories = victories; } public String getERA() { return ERA; } public void setERA(String eRA) { ERA = eRA; } public String getStrikeout() { return strikeout; } public void setStrikeout(String strikeout) { this.strikeout = strikeout; } public String getSave() { return save; } public void setSave(String save) { this.save = save; } public String getPosition() { return position; } @Override public String toString() { return "[("+position+") "+super.toString()+"승=" + victories + ", 평균자책점=" + ERA + ", 탈삼진=" + strikeout + ", 세이브=" + save + "]"; } }

fileClass

package file; import java.io.BufferedReader; import java.io.BufferedWriter; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.PrintWriter; public class fileClass { public static void createFile() { String filename="baseBall_Member"; while(true) { boolean b=findedFile(filename); if(b) { System.out.println("같은 파일이 존재합니다. 파일 생성하지 못했습니다."); return; } File newFile = new File("d:\\tmp\\"+filename+".txt"); try { if(newFile.createNewFile()) { System.out.println("파일을 생성하였습니다."); break; } } catch (Exception e) { System.out.println("예외가 발생하였음.. 파일 생성 실패"); } } } public static void writeFile(String info[]) { //덮어씌우면서 써지는 부분이기 때문에 dao에있는 모든 dto를 넘겨주고 그것을 그대로 파일에 덮어씌우는 메소드 String filename="baseBall_Member"; File writeFile = new File("d:\\tmp\\"+filename+".txt"); //파일이 있는지 없는지 알아보기 boolean b = findedFile(filename); if(!b) { System.out.println("파일이 존재하지 않습니다."); return; } b = checkBeforeWriteFile(writeFile); if(!b) { System.out.println("파일에 기입할 수 없습니다."); return; } try { PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(writeFile))); for (int i = 0; i < info.length; i++) { pw.println(info[i]); } pw.close(); } catch (Exception e) { e.printStackTrace(); } System.out.println("파일에 데이터를 작성하였습니다."); } public static String[] readFile() { String datas[] = null; String filename="baseBall_Member"; File readFile = new File("d:\\tmp\\" + filename + ".txt"); boolean b = checkBeforeReadFile(readFile); if(!b) { System.out.println("파일이 존재하지않거나 읽을수가 없습니다."); return null; } try { BufferedReader br = new BufferedReader(new FileReader(readFile)); // 배열의 갯수를 조사 int count =0; String str; while((str =br.readLine())!=null){ count++; } br.close(); //갯수만큼 데이터를 할당(배열) System.out.println("데이터의 갯수= "+count); datas = new String[count]; //데이터를 읽고 배열에 저장 int w= 0; br = new BufferedReader(new FileReader(readFile)); while((str = br.readLine()) != null) { datas[w] = str; w++; } br.close(); } catch (Exception e) { e.printStackTrace(); } //읽어 들인 데이터를 출력 for (int i = 0; i < datas.length; i++) { System.out.println("datas[" + i + "] = " + datas[i]); } return datas; } public static void plusWriteFile(String info) { String filename="baseBall_Member"; File writeFile = new File("d:\\tmp\\"+filename+".txt"); //파일이 있는지 없는지 알아보기 boolean b = findedFile(filename); if(!b) { System.out.println("파일이 존재하지 않습니다."); return; } b = checkBeforeWriteFile(writeFile); if(!b) { System.out.println("파일에 기입할 수 없습니다."); return; } try { PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(writeFile,true))); pw.write(info); pw.close(); } catch (Exception e) { e.printStackTrace(); } } public static void deleteFile() { String filename="baseBall_Member"; File deleteFile = new File("d:\\tmp\\" + filename + ".txt"); boolean b = deleteFile.exists(); if(!b) { System.out.println("삭제할 파일이 존재하지 않습니다."); return; }else { if(deleteFile.delete()) { System.out.println("삭제 성공"); }else { System.out.println("파일을 삭제할 수 없습니다."); } } } public static boolean checkBeforeReadFile(File f) { if(f.exists()) {//존재 여부 if(f.isFile()&&f.canRead()) {//파일인지 아닌지 And 읽기 가능 여부 return true; } } return false; } public static boolean checkBeforeWriteFile(File f) { if(f.canWrite()) { //FileWrite는 없으면 생성해서 작업하기 때문에 //read처럼 f.exists()를 할 필요가 없다. return true; } return false; } public static boolean findedFile(String fname) { File dirFile = new File("d:\\tmp"); String fileList[] = dirFile.list(); boolean b = false; for (int i = 0; i < fileList.length; i++) { if(fileList[i].equals(fname +".txt")==true) { b=true; break; } } return b; } }

sortClass

package sort; import java.util.ArrayList; import java.util.Scanner; import dto.batterClass; import dto.humanClass; import dto.pitcherClass; public class sortClass { ArrayList<humanClass> number; //값저장 배열 int N; //입력 받는 변수. boolean upDown;//1 올림 -1 내림 public sortClass(ArrayList<humanClass> number) { this.number = number; } public void init() { upDown=true; } public void input() { Scanner sc = new Scanner(System.in); // 입력 System.out.println("무엇으로 정렬할 거니? "); System.out.println("1.타율(타자) / 2.승(투수) "); N = sc.nextInt(); System.out.print("올림?내림?(올림:1, 내림 :-1): "); int num = sc.nextInt(); if(num==1) upDown=true; else upDown=false; } public void sorting() { for (int i = 0; i < number.size()-1 ; i++) { // 타자일때 if (number.get(i) instanceof batterClass) { double BA_i = Double.parseDouble(((batterClass) number.get(i)).getBA()); for (int j = i+1; j < number.size(); j++) { if (number.get(j) instanceof batterClass) { double BA_j = Double.parseDouble(((batterClass) number.get(j)).getBA()); System.out.println("i="+i); System.out.println("j="+j); System.out.println("BA_i = "+ BA_i + " BA_j= "+BA_j ); if (upDown) { // 올림// 올림차순 sort if (BA_i > BA_j) { swap(i, j); BA_i=Double.parseDouble(((batterClass) number.get(i)).getBA()); //BA_j = Double.parseDouble(((batterClass) number.get(j)).getBA()); } } else { // 내림 // 내림차순 sort if (BA_i < BA_j) { swap(i, j); BA_i=Double.parseDouble(((batterClass) number.get(i)).getBA()); //BA_j = Double.parseDouble(((batterClass) number.get(j)).getBA()); } } } } } // 투수일때 else if (number.get(i) instanceof pitcherClass) { int victory_i = Integer.parseInt(((pitcherClass) number.get(i)).getVictories()); for (int j = i+i; j < number.size(); j++) { if (number.get(j) instanceof pitcherClass) { int victory_j = Integer.parseInt(((pitcherClass) number.get(j)).getVictories()); if (upDown) { // 올림// 올림차순 sort if (victory_i > victory_j) { swap(i, j); victory_i = Integer.parseInt(((pitcherClass) number.get(i)).getVictories()); } } else { // 내림 // 내림차순 sort if (victory_i < victory_j) { swap(i, j); victory_i = Integer.parseInt(((pitcherClass) number.get(i)).getVictories()); } } } } } } } public void swap(int i,int j) { //객체 스왑이 안됨 humanClass tmp = number.get(i); number.set(i,number.get(j)); number.set(j, tmp); /* if(number[i] instanceof batterClass) { batterClass b_i = (batterClass)number[i]; batterClass b_j = (batterClass)number[j]; batterClass tmp = new batterClass(); tmp.setAge(b_i.getAge()); tmp.setBirth(b_i.getBirth()); tmp.setName(b_i.getName()); tmp.setNumber(b_i.getNumber()); tmp.setHR(b_i.getHR()); tmp.setBA(b_i.getBA()); tmp.setRBI(b_i.getRBI()); tmp.setSB(b_i.getSB()); b_i.setAge(b_j.getAge()); b_i.setBirth(b_j.getBirth()); b_i.setName(b_j.getName()); b_i.setNumber(b_j.getNumber()); b_i.setBA(b_j.getBA()); b_i.setHR(b_j.getHR()); b_i.setRBI(b_j.getRBI()); b_i.setSB(b_j.getSB()); b_j.setAge(tmp.getAge()); b_j.setBirth(tmp.getBirth()); b_j.setName(tmp.getName()); b_j.setNumber(tmp.getNumber()); b_j.setBA(tmp.getBA()); b_j.setHR(tmp.getHR()); b_j.setRBI(tmp.getRBI()); b_j.setSB(tmp.getSB()); } else if(number[i] instanceof pitcherClass) { pitcherClass p_i = (pitcherClass)number[i]; pitcherClass p_j = (pitcherClass)number[j]; pitcherClass tmp = new pitcherClass(); tmp.setAge(p_i.getAge()); tmp.setBirth(p_i.getBirth()); tmp.setName(p_i.getName()); tmp.setNumber(p_i.getNumber()); tmp.setERA(p_i.getERA()); tmp.setSave(p_i.getSave()); tmp.setStrikeout(p_i.getStrikeout()); tmp.setVictories(p_i.getVictories()); p_i.setAge(p_j.getAge()); p_i.setBirth(p_j.getBirth()); p_i.setName(p_j.getName()); p_i.setNumber(p_j.getNumber()); p_i.setERA(p_j.getERA()); p_i.setSave(p_j.getSave()); p_i.setStrikeout(p_j.getStrikeout()); p_i.setVictories(p_j.getVictories()); p_j.setAge(p_i.getAge()); p_j.setBirth(p_i.getBirth()); p_j.setName(p_i.getName()); p_j.setNumber(p_i.getNumber()); p_j.setERA(p_i.getERA()); p_j.setSave(p_i.getSave()); p_j.setStrikeout(p_i.getStrikeout()); p_j.setVictories(p_i.getVictories()); } */ } public void resultPrint() { for (int i = 0; i < number.size(); i++) { System.out.print(number.get(i) + " "); } } }