(22.10.26)Java 프로그래밍: Stream 실습 -입력스트림, 출력스트림

텍스트 파일에 5인의 정보를 저장한다

각행의 포맷은 번호|이름|전화

자바 스트림을 이용하여 텍스트 파일을 읽어들일때 특정인의 이름을 확인하여 이름과 전화번호를 화면에 표시해보세요.

Main 메소드만 이용하기

 

문서내용

 

 

Main 클래스 코드

import java.io.FileReader;
import java.io.BufferedReader;
public class StreamMain 
{
    
    public static void main(String[] args) 
    {
        try {
            FileReader fr = new FileReader("D:/java_test/info.txt");
            BufferedReader br = new BufferedReader(fr);
            String line = null;
            while((line=br.readLine())!=null) {
                String[] token = line.split("\\|");
                
                if(token[1].equals("해린")) {
                    System.out.printf("%s\t%s", token[1],token[2]);    
                }
            }
            br.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

콘솔 값:

해린 01066665555

 

 

파일 하드디스크  →   소스 스트림:FileReader  →   필터스트림:BufferedReader  →  메모리(RAM)   →   표준출력스트림:out →모니터

             

---

파일을 읽은 다음 메모리에 있는 데이터를 모니터가 아닌 다른 파일로 보내기(텍스트 파일 복사)

입력 스트림:

BufferedReader br = new BufferedReader(new FileReader("D:/java_test/info.txt"));

출력 스트림:

FileWriter fw = new FileWriter("D:/java_test/info2.txt");

 

실행 결과

텍스트 파일이 복사되어 info2.txt 파일이 새로 생긴걸 볼 수 있다.

 

하드디스크(파일)  →   소스 스트림:FileReader  →   필터스트림:BufferedReader  →  메모리(RAM)   →    싱크스트림: FileWriter → 하드디스크(파일)

---

자바가 운영체제 따라서 알아서 개행문자를 쓰게끔하는 필터 스트림을 넣어준 파일 복사하기

PrintWriter : 출력용 필터 스트림: 운영체제에 상관없이 (윈도우:"\n")없이도 개행문자를 넣어서 출력해준다.

 

 

복사된 문서

info3.txt 파일이 만들어졌다.

 

하드디스크(파일)  →   소스 스트림:FileReader  →   필터스트림:BufferedReader  →  메모리(RAM)   →  필터스트림:PrintWriter →  싱크스트림: FileWriter → 하드디스크(파일)

---

키보드에서 사원정보를 입력하여 파일(emps.txt)에 저장하려고 한다.

사원번호(empno), 이름(ename), 부서번호(depno), 입사일(hiredate)

사원 정보를 Emp 클래스의 객체에 저장하여 그 데이터가 파일에 저장되도록한다.

사원정보를 파일로부터 읽어올 때로 emp 클래스의 객체에  저장하여 화면 표시되도록한다.

 

Emp 클래스 코드

public class Emp 
{
    private int empno;
    private String ename;
    private int deptno;
    private String hiredate;
    
    public Emp() {}
    public Emp(int empno, String ename, int deptno, String hiredate) {
        this.empno = empno;
        this.ename = ename;
        this.deptno = deptno;
        this.hiredate = hiredate;
    }
    public Emp(String[] info)
    {
        this.empno = Integer.valueOf(info[0]);
        this.ename = info[1];
        this.deptno = Integer.valueOf(info[2]);
        this.hiredate = info[3];
    }
    @Override
    public boolean equals(Object obj) {
        Emp other = (Emp) obj;
        return this.empno==other.empno;
    }
    @Override
    public String toString() {
        return String.format("%d\t%s\t%d\t%s", 
                empno,ename,deptno,hiredate);
    }
    public int getEmpno() {
        return empno;
    }
    public void setEmpno(int empno) {
        this.empno = empno;
    }
    public String getEname() {
        return ename;
    }
    public void setEname(String ename) {
        this.ename = ename;
    }
    public int getDeptno() {
        return deptno;
    }
    public void setDeptno(int deptno) {
        this.deptno = deptno;
    }
    public String getHiredate() {
        return hiredate;
    }
    public void setHiredate(String hiredate) {
        this.hiredate = hiredate;
    }
}

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
 
public class StreamMain 
{
    public static void main(String[] args) 
    {
        boolean go = true;
        Menu menu = null;
        while(go)
        {
            menu = menu();
            switch(menu)
            {
            case LIST: printList(list());    break;
            case ADD:  add(getInput());        break;
            case FIND: printEmp(find());    break;
            case UPDATE: update();break;
            case DELETE: delete(); break;
            case EXIT: go=false; break;
            default: System.out.println("메뉴입력 오류");
            }
        }
        
        System.out.println("프로그램 종료");
    }
    static enum Menu { LIST, ADD, FIND, UPDATE, DELETE, EXIT};
    static ArrayList<Emp> empList = new ArrayList<>();
    
    private static Menu menu()
    {
        String mlist = "목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):";
        Scanner kbd = new Scanner(System.in);
        System.out.print(mlist);
        String m = kbd.nextLine().strip();
        Menu menu = null;
        switch(m)
        {
        case "s": menu = Menu.LIST;      break;
        case "a": menu = Menu.ADD;      break;
        case "f": menu = Menu.FIND;      break;
        case "u": menu = Menu.UPDATE;      break;
        case "d": menu = Menu.DELETE;      break;
        case "x": menu = Menu.EXIT;      break;
        default:  menu = null;
        }
        return menu;
    }
    
    private static Emp getInput()
    {
        Scanner kbd = new Scanner(System.in);
        System.out.print("사번:");
        int empno = kbd.nextInt();
        kbd.nextLine();
        
        System.out.print("이름:");
        String ename = kbd.nextLine().strip();
        
        System.out.print("부서번호:");
        int deptno = kbd.nextInt();
        kbd.nextLine();
        
        System.out.print("입사일:");
        String hiredate = kbd.nextLine().strip();
        
        return new Emp(empno,ename,deptno,hiredate);
    }
    
    private static void add(Emp emp)
    {
        try {
            // 파일의 마지막에 이어쓰기할 때는 아래처럼 append 모드를 true로 설정해야 한다
            PrintWriter pw = new PrintWriter(new FileWriter("D:/test/emp3.txt", true));
            String line = String.format("%d|%s|%d|%s",
                    emp.getEmpno(), emp.getEname(),emp.getDeptno(), emp.getHiredate());
            pw.println(line);
            pw.close(); 
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
    private static ArrayList<Emp> list()
    {
        try {
            BufferedReader br = new BufferedReader(new FileReader("D:/test/emp3.txt"));
            String line = null;
            ArrayList<Emp> list = new ArrayList<>();
            while((line=br.readLine())!=null)
            {
                list.add(new Emp(line.split("\\|")));
            }
            br.close();
            return list;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
    
    private static void printList(ArrayList<Emp> list)
    {
        for(int i=0;i<list.size();i++) 
        {
            System.out.println(list.get(i));
        }
    }
    
    private static Emp find()
    {
        Scanner kbd = new Scanner(System.in);
        System.out.print("검색할 사원의 번호:");
        int empno = kbd.nextInt();
        kbd.nextLine();
        
        Emp kwd = new Emp();
        kwd.setEmpno(empno);
        ArrayList<Emp> list = list();
        if(list.contains(kwd))
        {
            return list.get(list.indexOf(kwd));
        }
        return null;
    }
    
    private static void update()
    {
        Scanner kbd = new Scanner(System.in);
        System.out.print("수정할 사원의 번호:");
        int empno = kbd.nextInt();
        kbd.nextLine();
        
        Emp kwd = new Emp();
        kwd.setEmpno(empno);
        ArrayList<Emp> list = list();
        if(list.contains(kwd))
        {
            System.out.print("새 부서번호:");
            int deptno = kbd.nextInt();
            kbd.nextLine();
            
            list.get(list.indexOf(kwd)).setDeptno(deptno);
            // 기존 파일에 덮어쓰기
            overwrite(list);
            System.out.println("부서번호 변경 성공");
        }else {
            System.out.println("해당 번호의 사원은 없음");
        }
    }
    
    private static void delete()
    {
        Scanner kbd = new Scanner(System.in);
        System.out.print("삭제할 사원의 번호:");
        int empno = kbd.nextInt();
        kbd.nextLine();
        
        Emp kwd = new Emp();
        kwd.setEmpno(empno);
        ArrayList<Emp> list = list();
        if(list.contains(kwd))
        {
            list.remove(list.indexOf(kwd));
            overwrite(list);
            System.out.println("사원정보 삭제 성공");
        }else {
            System.out.println("해당 번호의 사원은 없음");
        }
    }
    
    private static void overwrite(ArrayList<Emp> list)
    {
        try {
            PrintWriter pw = new PrintWriter(new FileWriter("D:/test/emp3.txt"));
            Emp emp = null;
            for(int i=0;i<list.size();i++)
            {
                emp = list.get(i);
                String line = String.format("%d|%s|%d|%s",
                    emp.getEmpno(), emp.getEname(),emp.getDeptno(), emp.getHiredate());
                pw.println(line);
            }
            pw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    private static void printEmp(Emp emp)
    {
        System.out.println(emp);
    }
}

메인 메소드엔 기능별로 메소드를 나누어 코딩하였다.

 

1. list() 메소드

입력 스트림인 FileReader와 BufferedReader를 이용하여 문서에 있는 글자들을 |을 기준으로 나누어 문자열을 만들어 ArrayList 에 넣는다. 이후 printList()메소드에 리턴한 list를 파라미터로 받아 출력해낸다.

 

2. add() 메소드

먼저 getInput() 메소드를 통해 키보드에서 받아온 사원번호, 이름, 부서번호, 입사일 속성들을 Emp 객체를 생성해 저장하여 리턴해 준다. add() 메소드는 리턴된 Emp 객체를 받아서 출력스트림 FileWriter, PrintWriter을 통해 지정해준 파일에 정보를 저장한다. 문자열 포맷을 작성하여 파라미터로 받아온 Emp 객체의 값을 |로 나누어 각각 저장해준다.

 

3. find() 메소드

검색할 사원번호를 받아서 새로운 Emp 객체를 생성해 사원번호 값에 저장해준다.(객체간의 비교를 위해서) 리스트 메소드를 호출하여 문서에 있던  속성들을 ArrayList로 저장한 list에 검색한 번호를 가진 객체가 있는지 확인해준다. contains()메소드가 list의 원소마다 kwd와 하나하나 비교를 하고 이때 오버라이드된 equals() 메소드 기능을 쓴다. kwd와 같은게 있다면 해당 번호를 갖고 있는 인덱스를 가진 ArrayList의 Emp 객체를 리턴해주고 printEmp()메소드를 통해 출력한다. toString() 메소드를 오버라이드 해줬기 때문에 객체의 주소가 콘솔에 나오는 것이 아닌 문자열 포맷에 맞게 출력된다. 

 

4. update() 메소드

find()메소드와 동일하게 찾을 번호를 Emp객체에 저장해주고 ArrayList통해서 찾는다. 찾았다면 새 부서번호를 받아서 사원번호가 있는 리스트에 새로 받아온 부서번호로 수정해준다. 메모리에서만 수정한 것이지 파일에서 수정된 것이 아니므로 overwrite()메소드를 통해 수정된 정보를 파일에도 동일하게 저장해주도록한다.

 

5. delete() 메소드

update() 메소드 방식과 똑같이 리스트에서 사원번호가 있는 Emp객체 찾아서 리스트 한행을 지우고 overwrite()메소드를 통해 문서에서도 똑같이 삭제 시켜준다.