[자바] nio 를 이용한 소켓 클라부분 (GUI 연동)

시험도 끝났겠다 다시 자바 서버에 대해서 포스팅을 해보도록 하겠습니다. :D

우선 클라 화면부터 보시겠습니다.

서버에서도 말해드렸듯이 기능은

1. 자기자신에게만 서버에게 보내고 받는 기능

2. 자신이 원하는 상대에게 보낼수 있는 기능

3. 서버에 접속한 모든 클라한테 보낼수 있는 기능.

4. 연결 해제

그외 다른 기능이 필요하다 싶으면, 댓글로 남겨주시길 바랍니다.

쓰디쓴 피드백 감사하겠습니다.

그럼 소스를 보도록 하겠습니다.

package JSieunPlatform;
/*
 * 2017_04_11
 * Copyright 2017. J.sieun all rights reserved.
 * 마음껏 사용하도 되나, 제이름은 적어주시길 바랍니다.
 * */
import java.awt.BorderLayout;
import java.awt.Dimension;
import java.awt.GridLayout;
import java.awt.Toolkit;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.AdjustmentEvent;
import java.awt.event.AdjustmentListener;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

import javax.swing.BorderFactory;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollBar;
import javax.swing.JScrollPane;
import javax.swing.JTextField;
import javax.swing.JTextPane;
import javax.swing.text.SimpleAttributeSet;
import javax.swing.text.StyledDocument;

public class JClient extends JFrame implements ActionListener, AdjustmentListener, ItemListener, Runnable, KeyListener {
 private JLabel nameLabel;
 private JButton conectButton, disconectButton, packetButton;
 private JButton loginButton, whisperButton;

 // 상태창 변수들
 private JLabel stateLabel;
 private JScrollBar scrollBar;
 private JTextPane statePane;
 private SimpleAttributeSet attriState;
 private StyledDocument docState;
 // 메시지 입력 변수들
 private JTextField editText;
 private JTextField nameText;
 private JButton sendButton;

 // 소켓채널
 private Selector _Selector = null;
 private SocketChannel _Client = null;
 private InetSocketAddress connectAddress;
 private String _address = "127.0.0.1";
 private int _port = 7273;
 private boolean _bStart = true;

 // RecvBuf
 private ByteBuffer _RecvBuf;
 // 패킷
 private JPacket packet = null;
 // 유저 이름
 private String _UserName = "";

 public JClient() {
  super("J.Sieun Client");
  connectAddress = new InetSocketAddress(_address, _port);
  packet = new JPacket();
  _RecvBuf = ByteBuffer.allocate(128);
 }

 public void initialize() {
  // 제목
  JPanel mainPanel = new JPanel();
  nameLabel = new JLabel("Test_Client");
  mainPanel.add(nameLabel);

  // 버튼들
  JPanel buttonPanel = new JPanel(new GridLayout(8, 1));

  conectButton = new JButton("연결");
  conectButton.addActionListener(this);
  disconectButton = new JButton("연결해제");
  disconectButton.addActionListener(this);
  loginButton = new JButton("로그인");
  loginButton.addActionListener(this);
  whisperButton = new JButton("귓속말");
  whisperButton.addActionListener(this);
  packetButton = new JButton("패킷전송");
  packetButton.addActionListener(this);

  buttonPanel.add(conectButton);
  buttonPanel.add(disconectButton);
  buttonPanel.add(loginButton);
  buttonPanel.add(whisperButton);
  buttonPanel.add(packetButton);
  // 상태창
  JPanel statePanel = new JPanel(new BorderLayout());
  stateLabel = new JLabel("\tState Message\t", JLabel.CENTER);

  statePane = new JTextPane();
  statePane.setBorder(BorderFactory.createRaisedBevelBorder());
  statePane.setEditable(false);

  JScrollPane scroll = new JScrollPane(statePane, JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED,
    JScrollPane.HORIZONTAL_SCROLLBAR_NEVER);
  scroll.setOpaque(false);

  scrollBar = scroll.getVerticalScrollBar();
  docState = statePane.getStyledDocument();
  statePanel.add(stateLabel, "North");
  statePanel.add(scroll, "Center");
  // 메시지 입력창
  JPanel msgPanel = new JPanel();
  editText = new JTextField(15);
  nameText = new JTextField(8);
  sendButton = new JButton("보내기");
  sendButton.addActionListener(this);
  msgPanel.add(nameText);
  msgPanel.add(editText);
  msgPanel.add(sendButton);

  // 최종 추가
  getContentPane().add(mainPanel, "North");
  getContentPane().add(statePanel, "Center");
  getContentPane().add(msgPanel, "South");
  getContentPane().add(buttonPanel, "East");

  setDefaultCloseOperation(EXIT_ON_CLOSE);
  Toolkit kit = getToolkit();
  Dimension dmen = kit.getScreenSize();
  setLocation((int) ((dmen.width - getSize().width) / 2), (int) ((dmen.height - getSize().height) / 2));
  setSize(700, 400);
  setResizable(false);
  setVisible(true);
 }

 @Override
 public void itemStateChanged(ItemEvent arg0) {
  // TODO Auto-generated method stub

 }

 @Override
 public void adjustmentValueChanged(AdjustmentEvent arg0) {
  // TODO Auto-generated method stub

 }

 @Override
 public void actionPerformed(ActionEvent e) {
  // TODO Auto-generated method stub
  Object o = e.getSource();
  if (o.equals(conectButton)) {
   try {
    Connect();
   } catch (Exception ex) {
    stateMsg("서버연결 에러 : " + ex.getMessage());
   }
   conectButton.setEnabled(false);
  }
  if(o.equals(disconectButton))
  {
   packet.Clear();
   packet.Type(JPacket.DISCONNECT);
   packet.Add(_UserName);
   Send(packet);
   try {
    //종료
    _Client.close();
   } catch (IOException e1) {
    // TODO Auto-generated catch block
    e1.printStackTrace();
   }
   stateMsg("연결을 종료합니다. \n");
   disconectButton.setEnabled(false);
  }
  if (o.equals(packetButton)) {
   
   packet.Clear();
   packet.Type(JPacket.TESTPACKET);
   packet.Add("패킷테스트");
   packet.Add(1001001);
   
   Send(packet);
  }
  if (o.equals(sendButton)) {
   packet.Clear();
   packet.Type(JPacket.CHAT);
   packet.Add(_UserName);
   packet.Add("" + editText.getText());
   editText.setText("");
   Send(packet);
  }
  if (o.equals(loginButton)) {
   packet.Clear();
   packet.Type(JPacket.LOGIN);
   packet.Add("" + nameText.getText());
   _UserName = nameText.getText();
   nameText.setText("");
   nameLabel.setText("" + _UserName + " 님");
   Send(packet);
   loginButton.setEnabled(false);
  }
  if (o.equals(whisperButton)) {
   packet.Clear();
   packet.Type(JPacket.WHISPER);
   packet.Add(_UserName);
   packet.Add("" + nameText.getText());
   packet.Add("" + editText.getText());
   nameText.setText("");
   editText.setText("");
   Send(packet);
  }
 }

 @Override
 public void keyPressed(KeyEvent e) {
  /*
   * // TODO Auto-generated method stub if(e.getKeyCode() ==
   * KeyEvent.VK_ENTER) { packet.Clear(); packet.Type(JPacket.CHAT);
   * packet.Add(""+editText.getText()); editText.setText("");
   * Send(packet); }
   */
 }

 @Override
 public void keyTyped(KeyEvent e) {
  // TODO Auto-generated method stub
  int keycode = e.getKeyCode();
 }

 @Override
 public void keyReleased(KeyEvent e) {
  // TODO Auto-generated method stub
 }

 private void Connect() {
  try {
   _Client = SocketChannel.open();
   _Client.configureBlocking(false);
   // 요청연결
   _Client.connect(connectAddress);
   stateMsg("연결완료 \n");
   Thread thread = new Thread(this);
   thread.start();
  } catch (Exception e) {
   System.out.println("Connect Error : " + e.getMessage());
  }
 }

 public void Send(JPacket data) {
  try {
   _Client.write(data.getBuf());
   // Thread.sleep(10);
  } catch (Exception e) {
   System.out.println("Send Error : " + e.getMessage());
  }

 }

 public void run() {
  try {
   _Selector = Selector.open();
   _Client.register(_Selector, SelectionKey.OP_CONNECT);
  } catch (Exception ex) {
   stateMsg("Run Error : " + ex.getMessage() + "\n");
  }
  while (_bStart) {
   try {
    _Selector.select();
    Iterator<?> keys = _Selector.selectedKeys().iterator();
    while (keys.hasNext()) {
     SelectionKey key = (SelectionKey) keys.next();
     keys.remove();
     if (key.isConnectable()) {
      if (_Client.finishConnect()) {
       key.interestOps(SelectionKey.OP_READ);
      } else {
       key.cancel();
      }
     } else if (key.isReadable()) {
      _Client.read(_RecvBuf);
      RecvEvt(_RecvBuf);
     }
    }
   } catch (Exception e) {
    stateMsg("Recv Error :" + e.getMessage() + "\n");
    this._bStart = false;
   }
  }
 }

 public void RecvEvt(ByteBuffer _RecvBuf) {
  _RecvBuf.flip();
  packet.Clear();
  short Type = _RecvBuf.getShort();
  switch (Type) {
  case JPacket.LOGIN :
   stateMsg("서버와 접속이 완료되었습니다. \n");
   _RecvBuf.clear();
   break;
  case JPacket.LOGININFOR :
   stateMsg("*****로그인 정보*****\n");
   LoginInfor(_RecvBuf);
   _RecvBuf.clear();
   break;
  case JPacket.DISCONNECT :
   String _DisConnName = packet.RecvOut(_RecvBuf);
   stateMsg(_DisConnName + "님이 로그아웃하였습니다. \n");
   _RecvBuf.clear();
   break;
  case JPacket.CHAT :
   String _ChatName = packet.RecvOut(_RecvBuf);
   String _ChatMsg = packet.RecvOut(_RecvBuf);
   stateMsg("[" + _ChatName + "] 님의 말 : " + _ChatMsg + "\n");
   _RecvBuf.clear();
   break;
  case JPacket.WHISPER :
   String Whisper_Name = packet.RecvOut(_RecvBuf);
   String Whisper_Msg = packet.RecvOut(_RecvBuf);
   stateMsg("[" + Whisper_Name + "] 님의 귓속말 : " + Whisper_Msg + "\n");
   _RecvBuf.clear();
   break;
  case JPacket.TESTPACKET :
   String TestPacket = packet.RecvOut(_RecvBuf);
   String TestPacket1 = packet.RecvOut(_RecvBuf);
   stateMsg("패킷테스트 : " + TestPacket + " :::: " + TestPacket1 + "\n");
   _RecvBuf.clear();
   break;
  default:
   _RecvBuf.clear();
   break;
  }
 }

 // 상태메시지
 public void stateMsg(String msg) {
  try {
   attriState = new SimpleAttributeSet();
   statePane.setCaretPosition(docState.getEndPosition().getOffset() - 1);
   docState.insertString(statePane.getCaretPosition(), msg, attriState);
   scrollBar.addAdjustmentListener(this);
  } catch (Exception e) {
   String _errorMsg = "*상태메시지 에러발생!!*";
   JOptionPane.showMessageDialog(null, _errorMsg, "JServer", JOptionPane.INFORMATION_MESSAGE);
  }
 }

 // 로그인 정보
 public void LoginInfor(ByteBuffer _RecvBuf) {
  String _strSize = packet.RecvOut(_RecvBuf);
  int _Size = Integer.parseInt(_strSize);
  for (int i = 0; i < _Size; i++) {
   stateMsg((i  + 1 )+ "  " + packet.RecvOut(_RecvBuf) + " 님 ");
  }
  stateMsg("\n");
 }

 public static void main(String[] args) {
  // TODO Auto-generated method stub
  JClient test = new JClient();
  test.initialize();
 }

}

애초에 클라이언트는 서버에 접속하기만 하면 되기때문에 그렇게 많이 여러가지 것들을 배제한 상태입니다.

그럼 소스 해석을 해보도록 하겠습니다.

1. initilize () 일명 초기화 메소드

이건 그냥 GUI를 사용하는 것이기 때문에 패스 하도록 하겠습니다.

그외 버튼 이벤트도 패스 하도록 하겠습니다.

2. Connect() 메소드 ☆☆☆☆★

이 메소드를 한마디로 표현하자면 서버와의 연결을 시작하는 것으로 연결할 SocketChannel 을 Open 하고,

논블로킹 상태로 만듭니다.

3. Send()메소드 ☆☆☆☆★

이 메소드를 한마디로 표현하면 서버에게 보내는 메소드를 뜻합니다. write 메소드를 이용하여 보내고자하는

"패킷" 을 바이너리 형식 즉, Buffer 형식으로 변환해서 보내는 것이 되겠습니다.

4. run() 쓰레드 시작지점 ☆☆★★★

이부분은 쓰레드가 계속해서 실행이 되면서 서버에게 받을 데이터를 읽는 즉, 계속해서 클라는 "대기" 상태 입니다. 저는 이 run 부분을 없애고 클라가 서버에게 데이터를 전송할때에만 가능하도록 구현해보고싶었지만, 제 능력밖인거 같습니다. 아마 비동기 자바 소켓은 가능할것으로 보입니다.

아무튼 소스를 해석하자면  셀렉터를 생성하고 서버에게 데이터가 왔을시 Iterator 를 통하여 그 데이터를

_RecvBuf 라는 ByteBuffer 에 데이터를 삽입하고 그것을 RecvEvt 라는 메소드를 통해 이벤트를 처리하는 부분이 되겠습니다.

5. RecvEvt 메소드 ☆☆★★★

이메소드 같은 경우에는 서버로부터 받은 "패킷"을 자신이 원하는 값으로 변환해주는 메소드가 되겠습니다.

_RecvBuf 의 postion 값을 0 으로 해야 하기 때문에 "flip" 메소드를 사용했습니다. 그후

저같은 경우에는 패킷을 주고받을때

Type

(2byte)

Data(.....)

이런형식이기 때문에 처음에 2byte 를 Type값입니다.

그 Type값에 따라 스위치(Switch)문에 이벤트를 실행하도록 설정했습니다.

#그후에는 Data 값입니다.

*스위치문(Switch) 문을 통하여 이벤트를 다 끝마치면 _RecvBuf 는 초기화(clear())를 해줘야 합니다. *

그렇지 않으면, 현재 클라는 계속해서 대기 상태이기 때문에 이전에 받은 데이터 값이 계속해서 진행되고 원치 않는 값들이 나오는 에러를 보실수 있습니다.

나머지 부분은 서버 클라에서 그렇게 중요한 부분이 아니기 때문에 이상 포스팅을 마치도록 하겠습니다.

감사합니다.

제가 만든 소스이니 마음껏 사용하셔도 되지만, 소스 맨위에 이름은 변경하지 말아주셨으면 합니다.

기타 궁금하신 내용이 있으시면 댓글로 남겨주시길 바랍니다 :D