/
ChatServlet.java
76 lines (66 loc) · 2.33 KB
/
ChatServlet.java
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
package br.com.caelum.chat;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Queue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.atomic.AtomicInteger;
import javax.servlet.AsyncContext;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@WebServlet(urlPatterns = { "/subscribe" }, asyncSupported = true, loadOnStartup = 1)
public class ChatServlet extends HttpServlet {
private Queue<AsyncContext> clients = new ConcurrentLinkedQueue<AsyncContext>();
private BlockingQueue<String> messages = new LinkedBlockingQueue<String>();
private AtomicInteger contador = new AtomicInteger();
private AtomicInteger clientes = new AtomicInteger();
@Override
public void init() throws ServletException {
final ExecutorService executors = Executors.newCachedThreadPool();
Executors.newSingleThreadExecutor().execute(new Runnable() {
public void run() {
while (true) {
try {
final String message = messages.take();
for (final AsyncContext ctx : clients) {
executors.execute(new Runnable() {
public void run() {
try {
PrintWriter writer = ctx.getResponse()
.getWriter();
writer.println(message);
writer.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
});
}
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
});
}
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse arg1)
throws ServletException, IOException {
AsyncContext ctx = req.startAsync();
ctx.setTimeout(3000000);
clients.add(ctx);
System.out.println("novo cliente. id: " + clientes.incrementAndGet());
}
@Override
protected void doPost(HttpServletRequest req, HttpServletResponse arg1)
throws ServletException, IOException {
System.out.println("enviando mensagem para " + clientes + " clientes");
messages.add(String.format("mensagem número %d %n", contador.incrementAndGet()));
}
}