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

부모인스턴스는 자식 인스턴스를 할당받아서

 자식 인스턴스들을 관리할 수 있다는 개념을 적용한 예제

(전체 소스코드를 보고 싶지 않다면 daoClass에 위의 개념을 적용했으니 그부분만 보면 된다)

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.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 humanClass dtoArr[] = new humanClass[20]; private int count; private int selectIndex; public daoClass() { //프로그램 시작시 파일을 만들고 시작 fileClass.createFile(); count = 0; } 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(position, count); dtoArr[count].setNumber(number); dtoArr[count].setName(name); dtoArr[count].setAge(age); dtoArr[count].setBirth(birth); count++; } //포지션에 따른 입력 public void positionChoice(String position,int index) { //잘못 입력시 처리는 안함 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(); dtoArr[index] = new pitcherClass(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(); dtoArr[index] = new batterClass(BA, RBI, HR, SB); break; } } public void delete() { //삭제 boolean find=select(); if(find) { dtoArr[selectIndex]=null; if(selectIndex < count) { for (int i = selectIndex+1; i <= count; i++) { dtoArr[i-1] = dtoArr[i]; } } count--; 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 < count; i++) { //Null포인터 익셉션발생함 try { if (dtoArr[i].getName().equals(str)) { selectIndex = i; check = true; break; } }catch(Exception e) { e.printStackTrace(); } } if (check) { System.out.println(dtoArr[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(); dtoArr[selectIndex].setNumber(number); dtoArr[selectIndex].setName(name); dtoArr[selectIndex].setAge(age); dtoArr[selectIndex].setBirth(birth); System.out.println("수정 완료!!"); }else { System.out.println("수정할 데이터가 없습니다."); } } public void instanceOfInsert() { //할당받은 인스턴스에 따른 positionChoice 메소드 처리 if( dtoArr[selectIndex] instanceof batterClass ) //할당받은것이 타자 인스턴스이면 positionChoice("타자", selectIndex); else if( dtoArr[selectIndex] instanceof pitcherClass ) //할당받은것이 투수 인스턴스이면 positionChoice("투수", selectIndex); } public void dataPrint() { if(count==0) { System.out.println("출력할 데이터가 없습니다."); return; } for (int i = 0; i< count; i++) { System.out.println(dtoArr[i].toString()); } } public void fileSave() { //문자열로 된 데이터들을 파일에 덮어 씌우기(저장) String strArr[]=allDataToString(); fileClass.writeFile(strArr); } public String[] allDataToString() { //모든 데이터를 토큰을 붙여 문자열로 만드는 메소드 String strArr[]=new String[count]; for (int i = 0; i < strArr.length; i++) { //포지션-넘버-이름-나이-생년월일 + 포지션별 정보 token이용해서 붙이기 strArr[i]=dtoArr[i].getNumber()+ "-" +dtoArr[i].getName()+"-"+dtoArr[i].getAge()+"-"+dtoArr[i].getBirth(); if( dtoArr[i] instanceof batterClass ) { //할당받은것이 타자 인스턴스이면 batterClass bcls = (batterClass)dtoArr[i]; strArr[i]=bcls.getPosition()+"-"+ strArr[i]+"-"+bcls.getBA()+"-"+bcls.getRBI()+"-"+bcls.getHR()+"-"+bcls.getSB(); } else if( dtoArr[i] instanceof pitcherClass ) { //할당받은것이 투수 인스턴스이면 pitcherClass pcls = (pitcherClass)dtoArr[i]; strArr[i]= pcls.getPosition()+"-"+ strArr[i]+"-"+pcls.getVictories()+"-"+pcls.getERA()+"-"+pcls.getStrikeout()+"-"+pcls.getSave(); } } return strArr; } public void readfile() { //데이터 받아옴 String data[]=fileClass.readFile(); for (int i = 0; i < data.length; i++) { String strSplits[] = data[i].split("-"); if (strSplits[0].equals("타자")) { dtoArr[i] = new batterClass(strSplits[5],strSplits[6],strSplits[7],strSplits[8]); dtoArr[i].setNumber(strSplits[1]); dtoArr[i].setName(strSplits[2]); dtoArr[i].setAge(strSplits[3]); dtoArr[i].setBirth(strSplits[4]); } else if (strSplits[0].equals("투수")) { dtoArr[i] = new pitcherClass(strSplits[5],strSplits[6],strSplits[7],strSplits[8]); dtoArr[i].setNumber(strSplits[1]); dtoArr[i].setName(strSplits[2]); dtoArr[i].setAge(strSplits[3]); dtoArr[i].setBirth(strSplits[4]); } if(dtoArr[i]!=null) System.out.println(dtoArr[i].toString()); } } public void sort() { //정렬 sortClass sort = new sortClass(dtoArr); sort.init(); sort.input(); sort.sorting(); for (int i = 0; i< dtoArr.length; i++) { if(dtoArr[i]!=null) System.out.println(dtoArr[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 bA, String rBI, String hR, String sB) { super(); 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 victories, String eRA, String strikeout, String save) { super(); 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; } System.out.println(info); 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.Scanner; import dto.batterClass; import dto.humanClass; import dto.pitcherClass; public class sortClass { humanClass number[]; //값저장 배열 int N; //입력 받는 변수. boolean upDown;//1 올림 -1 내림 public sortClass(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.length - 1; i++) { // 타자일때 if (number[i] instanceof batterClass) { double BA_i = Double.parseDouble(((batterClass) number[i]).getBA()); for (int j = 0; j < number.length-1-i; j++) { if (number[j] instanceof batterClass) { double BA_j = Double.parseDouble(((batterClass) number[j]).getBA()); if (upDown) { // 올림// 올림차순 sort if (BA_i > BA_j) { System.out.println("BA_i = "+ BA_i + " BA_j= "+BA_j ); swap(i, j); } } else { // 내림 // 내림차순 sort if (BA_i < BA_j) { swap(i, j); } } } } } // 투수일때 else if (number[i] instanceof pitcherClass) { int victory_i = Integer.parseInt(((pitcherClass) number[i]).getVictories()); for (int j = 0; j < number.length-1-i; j++) { if (number[j] instanceof pitcherClass) { int victory_j = Integer.parseInt(((pitcherClass) number[j]).getVictories()); if (upDown) { // 올림// 올림차순 sort if (victory_i > victory_j) { swap(i, j); } } else { // 내림 // 내림차순 sort if (victory_i < victory_j) { swap(i, j); } } } } } } } public void swap(int i,int j) { humanClass tmp = number[i]; number[i] = number[j]; number[j] = tmp; } public void resultPrint() { for (int i = 0; i < number.length; i++) { System.out.print(number[i] + " "); } } }