JAVA 선수관리 프로그램 만들기

- Class Diagram

간단하게 클래스 다이어그램을 작성해보았다. 상속에 대한 화살표는 정확히 알겠는데 다른 흐름을 어떻게 화살표로 표현하는지는 더 공부해봐야겠다.

(앞에 +와 -는 public 인지 private 인지 구별)

 

 

- MainClass

switch문을 이용해 메뉴를 만들어놓았다.

package main;

import java.io.IOException;
import java.util.Scanner;

import dao.DaoClass;
import dto.HumanDto;

public class MainClass {

	public static void main(String[] args) throws IOException {

//		HumanDto playerArr[] = new HumanDto[2];

		DaoClass dao = new DaoClass();
		boolean exit = true;
		dao.newFile();
		
		while(exit) {
			Scanner sc = new Scanner(System.in);
			System.out.println("\n[야구 선수 정보 관리 프로그램]");
			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.print("실행할 작업을 선택해주세요. : ");
			int menuNum = sc.nextInt();
			
			switch (menuNum) {
				case 1:
					dao.insert();
					break;
				case 2:
					dao.delete();
					break;
				case 3:
					dao.select();
					break;
				case 4:
					dao.update();
					break;
				case 5:
					dao.fileWriter();
					break;
				case 6:
					dao.sorting();
					dao.swap();
					break;
				case 7:
					dao.allprint();
					break;
				case 8:
					exit = false;
					break;
				default:
					break;
			}

		}
		System.out.println("프로그램을 종료합니다.");

	}

}

 

 

- DaoClass

insert 메소드와 swap 메소드 작성할 때 헤매었다.

 insert 작성에는 if문 안에 자식Dto 클래스를 할당하면서 앞에 부모Dto 클래스도 같이 할당했는데 결과를 보니 2번 할당이 되면서 먼저 할당시킨 부모Dto 인스턴스 값들이 null로 나왔다.. 그래서 if문이 끝난 뒤 부모Dto값을 setter로 하나씩 값을 넣어주었더니 작동이 되었다.

swap에는 이중 for문안에 j값이 시작되는 부분을 i+1로 설정하지 않았다가 고생을 했다.. i값보다 항상 한 인덱스 뒤에서 부터 스왑이 이루어진다는 것을 명심하자.

package dao;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Collections;
import java.util.Scanner;

import dto.BatterDto;
import dto.HumanDto;
import dto.PitcherDto;

public class DaoClass {

	Scanner sc = new Scanner(System.in);
	private HumanDto playerArr[] = new HumanDto [30];
	private int indexCount;
	private	boolean upDown = false;
	
	
	public DaoClass() {
		indexCount = 0;
	}	
	

	//추가
	
	public void insert() {
		
		System.out.print("투수(1)/타자(2)");
		int t = sc.nextInt();
		String position = (t == 1)?"투수":"타자";
		
		System.out.print("등넘버 : ");
		int number = sc.nextInt();
		
		System.out.print("선수 이름 : ");
		String name = sc.next();
		
		System.out.print("선수 나이 : ");
		int age = sc.nextInt();
		
		System.out.print("선수 키 : ");
		double height = sc.nextDouble();
		
		
		if( t == 1 ) {
			System.out.print("win : ");
			int win = sc.nextInt();
			System.out.print("lose : ");
			int lose = sc.nextInt();
			System.out.print("defence : ");
			double defence = sc.nextDouble();
			
			
			playerArr[indexCount] = new PitcherDto (win,lose,defence);
			
		}
		else {
			System.out.print("batCount : ");
			int batCount =sc.nextInt();
			System.out.print("hit : ");
			int hit=sc.nextInt();
			System.out.print("hitAvg : ");
			double hitAvg=sc.nextDouble();
			
			playerArr[indexCount] = new BatterDto (batCount,hit,hitAvg);
			
			
		}
		
		playerArr[indexCount].setPosition(position);
		playerArr[indexCount].setNumber(number);
		playerArr[indexCount].setName(name);
		playerArr[indexCount].setAge(age);
		playerArr[indexCount].setHeight(height);
		
		indexCount++;
		
	}
	
	
	//삭제
	public void delete() {
		System.out.print("삭제할 선수 이름 : ");
		String delName = sc.next();
		
		int index = search(delName);
		
		if(index == -1) {
			System.out.println("해당 선수가 없습니다.");
			return;
		}
		
		System.out.println("\""+playerArr[index].getName()+"\" 선수의 정보를 삭제했습니다.");
		playerArr[index]=null;		
		
		
	}
	
	
	//검색
	public void select() {
		System.out.print("검색할 선수 이름을 입력하세요. : ");
		String selName = sc.next();
		
		int index = search(selName);
		if(index == -1) {
			System.out.println("해당 선수가 없습니다.");
			return;
		}
		
		System.out.println("\""+playerArr[index].getName()+"\" 선수");
		System.out.println(playerArr[index].toString());
		
	}
	
	
	//수정
	public void update() {
		System.out.print("수정할 선수 이름을 입력하세요. : ");
		String upName = sc.next();
		
		int index = search(upName);
		if(index == -1) {
			System.out.println("해당 선수가 없습니다.");
			return;
		}
		
		if(playerArr[index].getPosition().equals("투수")) {
			System.out.print("수정된 defence 값 입력 : ");
			double upDefence = sc.nextDouble();
			((PitcherDto) playerArr[index]).setDefence(upDefence);
		}
		else {
			System.out.print("수정된 hitAvg 값 입력 : ");
			double upHitAvg = sc.nextDouble();
			((BatterDto) playerArr[index]).setHitAvg(upHitAvg);
		}
		
		System.out.println("선수 정보를 수정했습니다.");
	}
	
	
	//전체선수출력
	public void allprint() {
		for (int i = 0; i < playerArr.length; i++) {
			if(playerArr[i] != null 
					) {
				System.out.println(playerArr[i].toString());
			}
		}
	}

	
	//조회
	public int search(String name) {
		int index = -1;
		for (int i = 0; i < playerArr.length; i++) {
			if(playerArr[i] != null && !playerArr[i].getName().equals("")) {
				if(playerArr[i].getName().contains(name)) {
					index = i;
					break;
				}
				
			}
		}
		return index;
	}
	
	
	//저장

	public void newFile() {
		File newFile = new File("F:\\tmp\\baseballTeam.txt");
		
		try {
			if(newFile.createNewFile()) {
			}
		}catch(IOException e) {
			e.printStackTrace();
		}
	}
	
	
	public void fileWriter() throws IOException {
		File file = new File("F:\\tmp\\baseballTeam.txt");
		
		try {
			PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));
			for (int i = 0; i < playerArr.length; i++) {
				if(playerArr[i] != null ) { 
					pw.println(playerArr[i].toString());
				}
			}
			pw.close();
			System.out.println("저장을 완료했습니다.");
			
		}catch (IOException e) {
			e.printStackTrace();
		}
		
	}
	
	
	
	//순서
	public void sorting() {

		System.out.print("등번호 정렬 오름차순(1) 내림차순(2)? ");
		int ud = sc.nextInt();
		upDown = (ud == 1) ? true : false;
		
	}

	public void swap() {
		HumanDto temp = new HumanDto();
		
		for (int i = 0; i < playerArr.length-1; i++) {
			for (int j = i + 1; j < playerArr.length; j++) {
				if(upDown == true && playerArr[i] != null && playerArr[j] != null) {
					if(playerArr[i].getNumber() > playerArr[j].getNumber()) {
						temp = playerArr[i];
						playerArr[i] = playerArr[j];
						playerArr[j] = temp;
					}
				}
				else if(upDown == false && playerArr[i] != null && playerArr[j] != null) {
					if(playerArr[i].getNumber() < playerArr[j].getNumber()) {
						temp = playerArr[i];
						playerArr[i] = playerArr[j];
						playerArr[j] = temp;
					}
				}
			}
		}
		
		System.out.println("정렬을 완료했습니다.");
	}

	
}

 

 

- HumanDto

Dto의 부모클래스 이다.

필드에 5개의 멤버 변수가 있고, 생성자와 get, set 메소드, toString 메소드를 작성해두었다.

package dto;

public class HumanDto {

	private String position;
	private int number;
	private String name;
	private int age;
	private double height;
	
	public HumanDto() {
	}

	public HumanDto(String position, int number, String name, int age, double height) {

		this.position = position;
		this.number = number;
		this.name = name;
		this.age = age;
		this.height = height;
	}

	public String getPosition() {
		return position;
	}


	public void setPosition(String position) {
		this.position = position;
	}


	public int getNumber() {
		return number;
	}


	public void setNumber(int number) {
		this.number = number;
	}


	public String getName() {
		return name;
	}


	public void setName(String name) {
		this.name = name;
	}


	public int getAge() {
		return age;
	}


	public void setAge(int age) {
		this.age = age;
	}


	public double getHeight() {
		return height;
	}


	public void setHeight(double height) {
		this.height = height;
	}

	@Override
	public String toString() {
		return "포지션=" + position + ", 등번호=" + number + ", 이름=" + name + ", 나이=" + age + ", 신장="
				+ height;
	}

}

 

 

 

- PitcherDto

HumanDto를 상속받는 자식 클래스이다.

필드에 3개의 멤버 변수가 있고, 생성자와 get, set 메소드, toString 메소드를 작성해두었다.

package dto;

public class PitcherDto extends HumanDto {

	private int win;
	private int lose;
	private double defence;
	
	public PitcherDto(int win, int lose, double defence) {
		
		super();
		this.win = win;
		this.lose = lose;
		this.defence = defence;
	}


	public int getWin() {
		return win;
	}


	public void setWin(int win) {
		this.win = win;
	}


	public int getLose() {
		return lose;
	}


	public void setLose(int lose) {
		this.lose = lose;
	}


	public double getDefence() {
		return defence;
	}


	public void setDefence(double defence) {
		this.defence = defence;
	}

	@Override
	public String toString() {
		return "PitcherDto ["+super.toString()+", win=" + win + ", lose=" + lose + ", defence=" + defence + "]";
	}
	
}

 

 

- BatterDto

HumanDto를 상속받는 자식 클래스이다.

필드에 3개의 멤버 변수가 있고, 생성자와 get, set 메소드, toString 메소드를 작성해두었다.

 

package dto;

public class BatterDto extends HumanDto {

	private int batCount;
	private int hit;
	private double hitAvg;
	
	public BatterDto() {}

	public BatterDto(int batCount, int hit, double hitAvg) {
		
		super();
		this.batCount = batCount;
		this.hit = hit;
		this.hitAvg = hitAvg;
	}


	public int getBatCount() {
		return batCount;
	}


	public void setBatCount(int batCount) {
		this.batCount = batCount;
	}


	public int getHit() {
		return hit;
	}


	public void setHit(int hit) {
		this.hit = hit;
	}


	public double getHitAvg() {
		return hitAvg;
	}


	public void setHitAvg(double hitAvg) {
		this.hitAvg = hitAvg;
	}


	@Override
	public String toString() {
		return "BatterDto ["+super.toString()+", batCount=" + batCount + ", hit=" + hit + ", hitAvg=" + hitAvg + "]";
	}
}