java-http-server/Server.java at main · surya-moorthy/java-http-server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.net.ServerSocket;
import java.net.Socket;

public class Server {

    ServerSocket serverSocket;
    Socket server;

    DataInputStream in;
    DataOutputStream out;

    Server(int port) {
        try {
            serverSocket = new ServerSocket(port);
        } catch(Exception e) {
            System.out.println("Error while server init. : " + e.getMessage());
        }
    }

    public void accept() {

        try {

            if(server != null && server.isConnected()) {
                System.out.println("Server already connected to a client.");
                return;
            }

            server = serverSocket.accept();

            System.out.println("Server initialized.");

            in = new DataInputStream(server.getInputStream());
            out = new DataOutputStream(server.getOutputStream());

        } catch(Exception e) {
            System.out.println("Error while connecting : " + e.getMessage());
        }

    }

    public void send(String str) {
        try {
            out.writeUTF(str);
            out.flush();
        } catch(Exception e) {
            System.out.println("Error while sending : " + e.getMessage());
        }
    }

    public void receive() {
        try {

            while(true) {
                String msg = in.readUTF();
                System.out.println("Client : " + msg);
            }

        } catch(Exception e) {
            System.out.println("Error while receiving : " + e.getMessage());
        }
    }

    public void close() {

        if(server != null && !server.isConnected()) {
            System.out.println("Client is not connected.");
            return;
        }

        try {

            server.close();
            serverSocket.close();

        } catch(Exception e) {
            System.out.println("Error while closing the client." + e.getMessage());
        }

    }

}