public class BotClient extends Client {
public class BotSocketThread extends SocketThread {
}
@Override
protected SocketThread getSocketThread() {
return new BotSocketThread();
}
@Override
protected boolean shouldSendTextFromConsole() {
return false;
}
@Override
protected String getUserName() {
return String.format("date_bot_%02d", (int) (Math.random() * 100));
}
public static void main(String[] args) {
BotClient botClient = new BotClient();
botClient.run();
}
}package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
/**
* Created by kot on 29.10.17.
*/
public class Server {
private static Map<String, Connection> connectionMap = new java.util.concurrent.ConcurrentHashMap<>();
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
while (true) {
Message message = connection.receive();
if (message.getType() == MessageType.TEXT) {
String newMessage = userName + ": " + message.getData();
sendBroadcastMessage(new Message(MessageType.TEXT, newMessage));
} else ConsoleHelper.writeMessage("Something wrong with your message " + userName);
}
}
private void sendListOfUsers(Connection connection, String userName) throws IOException {
for (String name : connectionMap.keySet()) {
if (!name.equals(userName)) {
connection.send(new Message(MessageType.USER_ADDED, name));
}
}
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
String name = null;
while (true) {
connection.send(new Message(MessageType.NAME_REQUEST));
Message message = connection.receive();
if (message.getType() == MessageType.USER_NAME) {
name = message.getData();
if (!name.isEmpty() && !connectionMap.containsKey(name)) {
connectionMap.put(name, connection);
connection.send(new Message(MessageType.NAME_ACCEPTED));
return name;
}
}
}
}
public void run() {
System.out.println("Connect " + socket.getRemoteSocketAddress());
try {
Connection connection = new Connection(socket);
String userName = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED, userName));
sendListOfUsers(connection, userName);
serverMainLoop(connection, userName);
connectionMap.remove(userName);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, userName));
} catch (IOException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
}
}
}
public static void sendBroadcastMessage(Message message) {
for (Map.Entry<String, Connection> entry : connectionMap.entrySet()) {
try {
entry.getValue().send(message);
} catch (IOException e) {
System.out.println("не получилось отправить сообщение");
}
}
}
public static void main(String[] args) throws IOException {
ServerSocket serverSocket = new ServerSocket(ConsoleHelper.readInt());
System.out.println("сервер запущен.");
while (true) {
try {
Socket socket = serverSocket.accept();
Handler handler = new Handler(socket);
handler.start();
} catch (Exception e) {
serverSocket.close();
System.out.println(e.getMessage());
break;
}
}
}
}