Mercurial Hosting > luan

annotate src/goodjava/webserver/ServerSentEvents.java @ 1740:c9c974817d0c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
add ServerSentEvents.sweep
author Franklin Schmidt <fschmidt@gmail.com>
date Thu, 03 Nov 2022 22:05:11 -0600
parents 9713f7fd50b3
children 623d9f62ab92
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
1 package goodjava.webserver;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
2
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
3 import java.io.Writer;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
4 import java.io.OutputStreamWriter;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
5 import java.io.BufferedWriter;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
6 import java.io.IOException;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
7 import java.net.Socket;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
8 import java.util.List;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
9 import java.util.ArrayList;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
10 import java.util.Map;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
11 import java.util.HashMap;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
12 import java.util.Collections;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
13 import java.util.Iterator;
1740
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
14 import java.util.Timer;
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
15 import java.util.TimerTask;
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
16 import goodjava.logging.Logger;
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
17 import goodjava.logging.LoggerFactory;
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
18
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
19
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
20 public class ServerSentEvents {
1740
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
21 private static final Logger logger = LoggerFactory.getLogger(ServerSentEvents.class);
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
22
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
23 private static class Con {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
24 final Socket socket;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
25 final Writer writer;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
26
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
27 Con(Socket socket) throws IOException {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
28 this.socket = socket;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
29 this.writer = new BufferedWriter( new OutputStreamWriter( socket.getOutputStream(), "UTF-8" ) );
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
30 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
31 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
32
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
33 private static class EventHandler {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
34 private final List<Con> cons = new ArrayList<Con>();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
35
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
36 synchronized void add(Con con) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
37 cons.add(con);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
38 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
39
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
40 synchronized void write( String url, String content ) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
41 Iterator<Con> iter = cons.iterator();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
42 while( iter.hasNext() ) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
43 Con con = iter.next();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
44 Writer writer = con.writer;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
45 try {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
46 writer.write(content);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
47 writer.write("\n\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
48 writer.flush();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
49 } catch(IOException e) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
50 iter.remove();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
51 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
52 }
1740
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
53 if( cons.isEmpty() ) {
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
54 map.remove(url);
1740
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
55 //logger.info("removed "+url);
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
56 }
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
57 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
58 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
59
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
60 private static final Map<String,EventHandler> map
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
61 = Collections.synchronizedMap(new HashMap<String,EventHandler>());
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
62
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
63 static void add(Socket socket,Request request) throws IOException {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
64 Con con = new Con(socket);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
65
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
66 Writer writer = con.writer;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
67 writer.write("HTTP/1.1 200 OK\r\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
68 writer.write("Access-Control-Allow-Origin: *\r\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
69 writer.write("Cache-Control: no-cache\r\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
70 writer.write("Content-Type: text/event-stream\r\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
71 writer.write("\r\n");
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
72 writer.flush();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
73
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
74 String url = request.url();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
75 EventHandler handler;
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
76 synchronized(map) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
77 handler = map.get(url);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
78 if( handler==null ) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
79 handler = new EventHandler();
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
80 map.put(url,handler);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
81 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
82 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
83 handler.add(con);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
84 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
85
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
86 public static void write( String url, String content ) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
87 EventHandler handler = map.get(url);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
88 if( handler != null )
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
89 handler.write(url,content);
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
90 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
91
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
92 public static String toData(String message) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
93 if( message.endsWith("\n") )
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
94 message = message.substring( 0, message.length() - 1 );
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
95 return "data: " + message.replace( "\n", "\ndata: " ) + "\n";
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
96 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
97
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
98 public static void writeMessage( String url, String message ) {
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
99 write( url, toData(message) );
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
100 }
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
101
1740
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
102 private static void sweep() {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
103 List<String> urls;
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
104 synchronized(map) {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
105 urls = new ArrayList<String>(map.keySet());
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
106 }
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
107 for( String url : urls ) {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
108 write( url, "event: ping\n" );
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
109 }
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
110 }
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
111
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
112 private static final Timer sweeper = new Timer("ServerSentEvents",true);
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
113 static {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
114 long period = 1000L*60*60; // hour
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
115 TimerTask task = new TimerTask() {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
116 public void run() {
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
117 sweep();
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
118 }
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
119 };
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
120 sweeper.schedule(task,period,period);
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
121 }
c9c974817d0c add ServerSentEvents.sweep
Franklin Schmidt <fschmidt@gmail.com>
parents: 1738
diff changeset
122
1738
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
123 private ServerSentEvents() {} // never
9713f7fd50b3 server-sent events
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
124 }