Rahul/socket/read/pr4 (#6575)

* Making examples simple

* Changing variable names

* Modificatons in naming

* Review changes
This commit is contained in:
rahusriv 2019-03-23 05:17:23 +05:30 committed by maibin
parent 4151e55d3b
commit cc73a3051b
2 changed files with 35 additions and 45 deletions

View File

@ -1,35 +1,27 @@
package com.baeldung.socket.read; package com.baeldung.socket.read;
import java.net.*; import java.net.*;
import java.nio.charset.StandardCharsets;
import java.io.*; import java.io.*;
public class Client { public class Client {
//Initialize socket, input and output stream
private Socket socket = null;
private DataInputStream in = null;
private DataOutputStream out = null;
public void runClient(String ip, int port) { public void runClient(String ip, int port) {
try { try {
socket = new Socket(ip, port); Socket socket = new Socket(ip, port);
System.out.println("Connected to server ..."); System.out.println("Connected to server ...");
in = new DataInputStream(System.in); DataInputStream in = new DataInputStream(System.in);
out = new DataOutputStream(socket.getOutputStream()); DataOutputStream out = new DataOutputStream(socket.getOutputStream());
} catch(Exception e) {
e.printStackTrace();
}
char type = 's'; // s for string char type = 's'; // s for string
int length = 29; int length = 29;
String data = "This is a string of length 29"; String data = "This is a string of length 29";
byte[] dataInBytes = data.getBytes(); byte[] dataInBytes = data.getBytes(StandardCharsets.UTF_8);
//Sending data in TLV format //Sending data in TLV format
try {
out.writeChar(type); out.writeChar(type);
out.writeInt(length); out.writeInt(length);
out.write(dataInBytes); out.write(dataInBytes);
} catch (IOException e) { } catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace(); e.printStackTrace();
} }
} }

View File

@ -1,48 +1,46 @@
package com.baeldung.socket.read; package com.baeldung.socket.read;
import java.net.*; import java.net.*;
import java.nio.charset.StandardCharsets;
import java.io.*; import java.io.*;
public class Server { public class Server {
//Socket and input stream
private Socket socket = null;
private ServerSocket server = null;
private DataInputStream in = null;
public void runServer(int port) { public void runServer(int port) {
//Start the server and wait for connection //Start the server and wait for connection
try { try {
server = new ServerSocket(port); ServerSocket server = new ServerSocket(port);
System.out.println("Server Started. Waiting for connection ..."); System.out.println("Server Started. Waiting for connection ...");
socket = server.accept(); Socket socket = server.accept();
System.out.println("Got connection from client."); System.out.println("Got connection from client.");
//Get input stream from socket variable and convert the same to DataInputStream //Get input stream from socket variable and convert the same to DataInputStream
in = new DataInputStream(new BufferedInputStream(socket.getInputStream())); DataInputStream in = new DataInputStream(new BufferedInputStream(socket.getInputStream()));
//Read type and length of data //Read type and length of data
char dataType = in.readChar(); char dataType = in.readChar();
int length = in.readInt(); int length = in.readInt();
System.out.println("Type : "+dataType); System.out.println("Type : "+dataType);
System.out.println("Lenght :"+length); System.out.println("Lenght :"+length);
if(dataType == 's') {
//Read String data in bytes //Read String data in bytes
byte[] messageByte = new byte[length]; byte[] messageByte = new byte[length];
boolean end = false; boolean end = false;
String dataString = ""; StringBuilder dataString = new StringBuilder(length);
int totalBytesRead = 0; int totalBytesRead = 0;
//We need to run while loop, to read all data in that stream //We need to run while loop, to read all data in that stream
while(!end) { while(!end) {
int currentBytesRead = in.read(messageByte); int currentBytesRead = in.read(messageByte);
totalBytesRead = currentBytesRead + totalBytesRead; totalBytesRead = currentBytesRead + totalBytesRead;
if(totalBytesRead <= length) { if(totalBytesRead <= length) {
dataString += new String(messageByte,0,currentBytesRead); dataString.append(new String(messageByte,0,currentBytesRead,StandardCharsets.UTF_8));
} else { } else {
dataString += new String(messageByte,0,length - totalBytesRead + currentBytesRead); dataString.append(new String(messageByte,0,length - totalBytesRead + currentBytesRead,StandardCharsets.UTF_8));
} }
if(dataString.length()>=length) { if(dataString.length()>=length) {
end = true; end = true;
} }
} }
System.out.println("Read "+length+" bytes of message from client. Message = "+dataString);; System.out.println("Read "+length+" bytes of message from client. Message = "+dataString);
}
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }