update socket (#183)

realDuYuanChao · web-flow · commit 19363ecabbd9 · 2021-09-12T17:03:13.000+08:00
diff --git a/src/main/java/com/examplehub/basics/network/SocketClient.java b/src/main/java/com/examplehub/basics/network/SocketClient.java
@@ -0,0 +1,37 @@
+package com.examplehub.basics.network;
+
+import java.io.*;
+import java.net.Socket;
+import java.nio.charset.StandardCharsets;
+import java.util.Scanner;
+
+public class SocketClient {
+  public static void main(String[] args) throws IOException {
+    Socket clientSocket = new Socket("127.0.0.1", 6666);
+    try (InputStream inputStream = clientSocket.getInputStream()) {
+      try (OutputStream outputStream = clientSocket.getOutputStream()) {
+        handle(inputStream, outputStream);
+      }
+    }
+    clientSocket.close();
+    System.out.println("disconnected.");
+  }
+  private static void handle(InputStream inputStream, OutputStream outputStream) throws IOException {
+    var writer = new BufferedWriter(new OutputStreamWriter(outputStream, StandardCharsets.UTF_8));
+    var reader = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8));
+    Scanner scanner = new Scanner(System.in);
+    System.out.println("[server] " + reader.readLine());
+    for (;;) {
+      System.out.print(">>> "); // 打印提示
+      String s = scanner.nextLine(); // 读取一行输入
+      writer.write(s + System.lineSeparator());
+//      writer.newLine();
+      writer.flush();
+      String resp = reader.readLine();
+      System.out.println("<<< " + resp);
+      if (resp.equals("bye")) {
+        break;
+      }
+    }
+  }
+}
diff --git a/src/main/java/com/examplehub/basics/network/SocketServer.java b/src/main/java/com/examplehub/basics/network/SocketServer.java
@@ -0,0 +1,57 @@
+package com.examplehub.basics.network;
+
+import java.io.*;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.nio.charset.StandardCharsets;
+
+public class SocketServer {
+  static class ServerHandler implements Runnable {
+    private Socket socket;
+
+    public ServerHandler(Socket socket) {
+      this.socket = socket;
+    }
+    public void run() {
+      try (InputStream inputStream = this.socket.getInputStream()) {
+        try(OutputStream outputStream = this.socket.getOutputStream()) {
+          handle(inputStream, outputStream);
+        }
+      } catch (Exception e) {
+        try {
+          this.socket.close();
+        } catch (IOException ex) {
+        }
+        System.out.println("client disconnect");
+      }
+    }
+
+    private void handle(InputStream inputStream, OutputStream outputStream) throws IOException {
+      var reader = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8));
+      var writer = new BufferedWriter(new OutputStreamWriter(outputStream, StandardCharsets.UTF_8));
+      writer.write("hello!\n");
+      writer.flush();
+      while (true) {
+        String line = reader.readLine();
+        if (line.equals("bye")) {
+          writer.write("bye\n");
+          writer.flush();
+          break;
+        }
+        writer.write("ok: " + line + "\n");
+        writer.flush();
+      }
+    }
+  }
+
+  public static void main(String[] args) throws IOException {
+    ServerSocket serverSocket = new ServerSocket(6666);
+    System.out.println("server is started!");
+    while (true) {
+      Socket socket = serverSocket.accept();
+      System.out.println("connected from " + socket.getRemoteSocketAddress());
+      new Thread(new ServerHandler(socket)).start();
+    }
+    //TODO
+  }
+}
diff --git a/src/test/java/com/examplehub/basics/network/SocketServerTest.java b/src/test/java/com/examplehub/basics/network/SocketServerTest.java
@@ -0,0 +1,5 @@
+package com.examplehub.basics.network;
+
+class SocketServerTest {
+  //TODO
+}

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +package com.examplehub.basics.network;
++
 +class SocketServerTest {
 +  //TODO
 +}