1 /* 2 * Copyright (c) 2011 Oracle and/or its affiliates. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * - Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 11 * - Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * - Neither the name of Oracle nor the names of its 16 * contributors may be used to endorse or promote products derived 17 * from this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 20 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 21 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 26 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 27 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 28 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 29 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 /* 33 * This source code is provided to illustrate the usage of a given feature 34 * or technique and has been deliberately simplified. Additional steps 35 * required for a production-quality application, such as security checks, 36 * input validation and proper error handling, might not be present in 37 * this sample code. 38 */ 39 40 41 import java.nio.ByteBuffer; 42 43 /** 44 * Writes all messages in our buffer to the other clients 45 * and appends new data read from the socket to our buffer 46 */ 47 class MessageReader implements DataReader { 48 private final ChatServer chatServer; 49 MessageReader(ChatServer chatServer)50 public MessageReader(ChatServer chatServer) { 51 this.chatServer = chatServer; 52 } 53 acceptsMessages()54 public boolean acceptsMessages() { 55 return true; 56 } 57 58 /** 59 * Write all full messages in our buffer to 60 * the other clients 61 * 62 * @param client the client to read messages from 63 */ 64 @Override beforeRead(Client client)65 public void beforeRead(Client client) { 66 // Check if we have any messages buffered and send them 67 String message = client.nextMessage(); 68 while (message != null) { 69 chatServer.writeMessageToClients(client, message); 70 message = client.nextMessage(); 71 } 72 } 73 74 /** 75 * Append the read buffer to the clients message buffer 76 * @param client the client to append messages to 77 * @param buffer the buffer we received from the socket 78 * @param bytes the number of bytes read into the buffer 79 */ 80 @Override onData(Client client, ByteBuffer buffer, int bytes)81 public void onData(Client client, ByteBuffer buffer, int bytes) { 82 buffer.flip(); 83 // Just append the message on the buffer 84 client.appendMessage(new String(buffer.array(), 0, bytes)); 85 } 86 } 87