packagecom.chat.demo;/** Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
*http://www.apache.org/licenses/LICENSE-2.0*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.*/
importjava.io.IOException;importjava.util.Set;importjava.util.concurrent.CopyOnWriteArraySet;importjava.util.concurrent.atomic.AtomicInteger;importjavax.websocket.OnClose;importjavax.websocket.OnError;importjavax.websocket.OnMessage;importjavax.websocket.OnOpen;importjavax.websocket.Session;importjavax.websocket.server.ServerEndpoint;//import org.apache.juli.logging.Log;//import org.apache.juli.logging.LogFactory;//import util.HTMLFilter;
@ServerEndpoint(value= "/websocket/chat")public classChatAnnotation {//private static final Log log = LogFactory.getLog(ChatAnnotation.class);
private static final String GUEST_PREFIX = "Guest";private static final AtomicInteger connectionIds = new AtomicInteger(0);private static final Set connections =
new CopyOnWriteArraySet();private finalString nickname;privateSession session;publicChatAnnotation() {
nickname= GUEST_PREFIX +connectionIds.getAndIncrement();
}
@OnOpenpublic voidstart(Session session) {this.session =session;
connections.add(this);
String message= String.format("* %s %s", nickname, "has joined.");
broadcast(message);
}
@OnClosepublic voidend() {
connections.remove(this);
String message= String.format("* %s %s",
nickname,"has disconnected.");
broadcast(message);
}
@OnMessagepublic voidincoming(String message) {//Never trust the client
String filteredMessage = String.format("%s: %s",
nickname, message.toString());
broadcast(filteredMessage);
}
@OnErrorpublic void onError(Throwable t) throwsThrowable {
System.out.println("Chat Error: " +t.toString());
}private static voidbroadcast(String msg) {for(ChatAnnotation client : connections) {try{synchronized(client) {
client.session.getBasicRemote().sendText(msg);
}
}catch(IOException e) {//log.debug("Chat Error: Failed to send message to client", e);
connections.remove(client);try{
client.session.close();
}catch(IOException e1) {//Ignore
}
String message= String.format("* %s %s",
client.nickname,"has been disconnected.");
broadcast(message);
}
}
}
}