diff --git a/src/main/java/vip/fubuki/playersync/PlayerSync.java b/src/main/java/vip/fubuki/playersync/PlayerSync.java index 23e1e76..d9ca417 100644 --- a/src/main/java/vip/fubuki/playersync/PlayerSync.java +++ b/src/main/java/vip/fubuki/playersync/PlayerSync.java @@ -41,7 +41,7 @@ private void commonSetup(final FMLCommonSetupEvent event) { @SubscribeEvent public void onServerStarting(ServerStartingEvent event) throws SQLException { - JDBCsetUp.executeUpdate("CREATE DATABASE IF NOT EXISTS `playersync`",1); + JDBCsetUp.executeUpdate("CREATE DATABASE IF NOT EXISTS "+JdbcConfig.DATABASE_NAME.get(),1); JDBCsetUp.executeUpdate(""" CREATE TABLE IF NOT EXISTS `player_data` ( @@ -59,8 +59,6 @@ public void onServerStarting(ServerStartingEvent event) throws SQLException { `last_server` int DEFAULT NULL, PRIMARY KEY (`uuid`) );"""); - JDBCsetUp.executeUpdate("CREATE TABLE IF NOT EXISTS chat (player CHAR(36) NOT NULL,message TEXT," + - "timestamp BIGINT)"); JDBCsetUp.executeUpdate(""" CREATE TABLE IF NOT EXISTS server_info ( `id` INT NOT NULL, diff --git a/src/main/java/vip/fubuki/playersync/config/JdbcConfig.java b/src/main/java/vip/fubuki/playersync/config/JdbcConfig.java index 3408b14..d51ce1f 100644 --- a/src/main/java/vip/fubuki/playersync/config/JdbcConfig.java +++ b/src/main/java/vip/fubuki/playersync/config/JdbcConfig.java @@ -14,9 +14,13 @@ public class JdbcConfig { public static ForgeConfigSpec.IntValue PORT; public static ForgeConfigSpec.ConfigValue USERNAME; public static ForgeConfigSpec.ConfigValue PASSWORD; + public static ForgeConfigSpec.ConfigValue DATABASE_NAME; public static ForgeConfigSpec.ConfigValue> SYNC_WORLD; public static ForgeConfigSpec.BooleanValue USE_SSL; public static ForgeConfigSpec.BooleanValue SYNC_CHAT; + public static ForgeConfigSpec.BooleanValue IS_CHAT_SERVER; + public static ForgeConfigSpec.ConfigValue CHAT_SERVER_IP; + public static ForgeConfigSpec.IntValue CHAT_SERVER_PORT; public static ForgeConfigSpec.ConfigValue SERVER_ID; @@ -29,9 +33,13 @@ public class JdbcConfig { USE_SSL = COMMON_BUILDER.comment("whether use SSL").define("use_ssl", false); USERNAME = COMMON_BUILDER.comment("username").define("user_name", "root"); PASSWORD = COMMON_BUILDER.comment("password").define("password", "password"); + DATABASE_NAME = COMMON_BUILDER.comment("database name").define("db_name","playersync"); SERVER_ID = COMMON_BUILDER.comment("the server id should be unique").define("Server_id", new Random().nextInt(1,Integer.MAX_VALUE-1)); SYNC_WORLD = COMMON_BUILDER.comment("The worlds that will be synchronized.If running in server it is supposed to have only one").define("sync_world", new ArrayList<>()); SYNC_CHAT= COMMON_BUILDER.comment("Whether synchronize chat").define("sync_chat", true); + IS_CHAT_SERVER = COMMON_BUILDER.comment("Whether recieve messages from other servers as host").define("IsChatServer",false); + CHAT_SERVER_IP = COMMON_BUILDER.define("ChatServerIP","127.0.0.1"); + CHAT_SERVER_PORT = COMMON_BUILDER.defineInRange("ChatServerPort",7900,0,65535); COMMON_BUILDER.pop(); COMMON_CONFIG = COMMON_BUILDER.build(); } diff --git a/src/main/java/vip/fubuki/playersync/sync/ChatSync.java b/src/main/java/vip/fubuki/playersync/sync/ChatSync.java index 93f0f7e..52d8b20 100644 --- a/src/main/java/vip/fubuki/playersync/sync/ChatSync.java +++ b/src/main/java/vip/fubuki/playersync/sync/ChatSync.java @@ -2,54 +2,124 @@ import net.minecraft.network.chat.Component; import net.minecraft.server.players.PlayerList; +import net.minecraftforge.common.MinecraftForge; +import net.minecraftforge.event.entity.player.PlayerEvent; import net.minecraftforge.eventbus.api.SubscribeEvent; -import net.minecraftforge.fml.common.Mod; -import vip.fubuki.playersync.util.JDBCsetUp; +import vip.fubuki.playersync.config.JdbcConfig; -import java.sql.Connection; -import java.sql.PreparedStatement; -import java.sql.ResultSet; -import java.sql.SQLException; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.net.ServerSocket; +import java.net.Socket; +import java.util.Objects; +import java.util.Scanner; +import java.util.Set; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; -@Mod.EventBusSubscriber public class ChatSync { - static int tick = 0; - static long current = System.currentTimeMillis(); - - public static void register(){} - @SubscribeEvent - public static void onPlayerChat(net.minecraftforge.event.ServerChatEvent event) throws SQLException { - String sql = "INSERT INTO chat (player, message, timestamp) VALUES (?, ?, ?)"; - try (Connection connection = JDBCsetUp.getConnection(); - PreparedStatement preparedStatement = connection.prepareStatement(sql)) { - preparedStatement.setString(1, event.getUsername()); - preparedStatement.setString(2, event.getRawText()); - preparedStatement.setLong(3, current); - preparedStatement.executeUpdate(); + static PlayerList playerList; + + static ServerSocket serverSocket; + static Socket clientSocket; + static Set SocketList; + static ExecutorService executorService = Executors.newCachedThreadPool(); + + public static void register(){ + if(JdbcConfig.IS_CHAT_SERVER.get()) + new Thread(ChatSync::ServerSocket).start(); + ClientSocket(); + MinecraftForge.EVENT_BUS.register(ChatSync.class); + } + + + private static void ServerSocket() { + try { + serverSocket = new ServerSocket(JdbcConfig.CHAT_SERVER_PORT.get()); + while (true) { + Socket newSocket = serverSocket.accept(); + SocketList.add(newSocket); + executorService.submit(() -> handleClient(newSocket)); + } + } catch (IOException e) { + e.printStackTrace(); + } finally { + try { + serverSocket.close(); + } catch (IOException e) { + e.printStackTrace(); + } } } - @SubscribeEvent - public static void Tick(net.minecraftforge.event.TickEvent.ServerTickEvent event) throws SQLException { - tick++; - if(tick == 20) { - ReadMessage(event.getServer().getPlayerList()); + private static void handleClient(Socket socket) { + try (InputStream inputStream = socket.getInputStream()) { + byte[] buffer = new byte[1024]; + int bytesRead; + while ((bytesRead = inputStream.read(buffer)) != -1) { + String message = new String(buffer, 0, bytesRead); + broadcastMessage(socket, message); + } + } catch (IOException e) { + e.printStackTrace(); + } finally { + SocketList.remove(socket); + try { + socket.close(); + } catch (IOException e) { + e.printStackTrace(); + } } } - public static void ReadMessage(PlayerList playerList) throws SQLException { - JDBCsetUp.QueryResult queryResult=JDBCsetUp.executeQuery("SELECT * FROM chat WHERE timestamp > " + current); - ResultSet resultSet= queryResult.getResultSet(); - current = System.currentTimeMillis(); - tick = 0; - while(resultSet.next()) { - String player = resultSet.getString("player"); - String message = resultSet.getString("message"); - Component textComponents = Component.literal(player+": "+message); - playerList.broadcastSystemMessage(textComponents, true); + private static void broadcastMessage(Socket sender, String message) { + for (Socket socket : SocketList) { + if (!socket.equals(sender)) { + try { + OutputStream outputStream = socket.getOutputStream(); + outputStream.write(message.getBytes()); + } catch (IOException e) { + e.printStackTrace(); + } + } } - resultSet.close(); - queryResult.getConnection().close(); + } + + private static void ClientSocket() { + try { + clientSocket = new Socket(JdbcConfig.CHAT_SERVER_IP.get(), JdbcConfig.CHAT_SERVER_PORT.get()); + Scanner scanner = new Scanner(clientSocket.getInputStream()); + while (scanner.hasNextLine()) { + String line = scanner.nextLine(); + Component textComponents = Component.nullToEmpty(line); + playerList.broadcastSystemMessage(textComponents,true); + } + } catch (IOException e) { + e.printStackTrace(); + reconnectClient(); + } + } + + private static void reconnectClient() { + //TODO + } + + @SubscribeEvent + public static void onPlayerChat(net.minecraftforge.event.ServerChatEvent event) throws IOException { + String message= event.getUsername()+":"+event.getMessage(); + OutputStream outputStream = clientSocket.getOutputStream(); + outputStream.write(message.getBytes()); + } + + @SubscribeEvent + public static void onPlayerJoin(PlayerEvent.PlayerLoggedInEvent event){ + playerList= Objects.requireNonNull(event.getEntity().getServer()).getPlayerList(); + } + + @SubscribeEvent + public static void onPlayerLeave(PlayerEvent.PlayerLoggedOutEvent event){ + playerList= Objects.requireNonNull(event.getEntity().getServer()).getPlayerList(); } } diff --git a/src/main/java/vip/fubuki/playersync/sync/VanillaSync.java b/src/main/java/vip/fubuki/playersync/sync/VanillaSync.java index 3709e30..0ec61b1 100644 --- a/src/main/java/vip/fubuki/playersync/sync/VanillaSync.java +++ b/src/main/java/vip/fubuki/playersync/sync/VanillaSync.java @@ -43,7 +43,7 @@ public static void register(){} public static void doPlayerJoin(PlayerEvent.PlayerLoggedInEvent event) throws SQLException, CommandSyntaxException, IOException { String player_uuid = event.getEntity().getUUID().toString(); JDBCsetUp.QueryResult queryResult=JDBCsetUp.executeQuery("SELECT online, last_server FROM player_data WHERE uuid='"+player_uuid+"'"); - ResultSet resultSet=queryResult.getResultSet(); + ResultSet resultSet=queryResult.resultSet(); ServerPlayer serverPlayer = (ServerPlayer) event.getEntity(); if(!resultSet.next()){ Store(event.getEntity(),true,Dist.CLIENT.isDedicatedServer()); @@ -52,11 +52,11 @@ public static void doPlayerJoin(PlayerEvent.PlayerLoggedInEvent event) throws SQ boolean online = resultSet.getBoolean("online"); int lastServer = resultSet.getInt("last_server"); queryResult=JDBCsetUp.executeQuery("SELECT * FROM player_data WHERE uuid='"+player_uuid+"'"); - resultSet= queryResult.getResultSet(); + resultSet= queryResult.resultSet(); if(online && lastServer != JdbcConfig.SERVER_ID.get()) { queryResult=JDBCsetUp.executeQuery("SELECT last_update,enable FROM server_info WHERE id='"+lastServer+"'"); - ResultSet getServerInfo = queryResult.getResultSet(); + ResultSet getServerInfo = queryResult.resultSet(); if(getServerInfo.next()){ long last_update = getServerInfo.getLong("last_update"); boolean enable = getServerInfo.getBoolean("enable"); diff --git a/src/main/java/vip/fubuki/playersync/util/JDBCsetUp.java b/src/main/java/vip/fubuki/playersync/util/JDBCsetUp.java index b6b6300..b66fa9b 100644 --- a/src/main/java/vip/fubuki/playersync/util/JDBCsetUp.java +++ b/src/main/java/vip/fubuki/playersync/util/JDBCsetUp.java @@ -13,8 +13,9 @@ public static Connection getConnection() throws SQLException { } public static QueryResult executeQuery(String sql) throws SQLException{ - Connection connection = getConnection(); - PreparedStatement useStatement = connection.prepareStatement("USE `playersync`"); + Connection connection = getConnection(); + PreparedStatement useStatement = connection.prepareStatement("USE ?"); + useStatement.setString(1, JdbcConfig.DATABASE_NAME.get()); useStatement.executeUpdate(); PreparedStatement queryStatement = connection.prepareStatement(sql); @@ -22,42 +23,42 @@ public static QueryResult executeQuery(String sql) throws SQLException{ return new QueryResult(connection,resultSet); } - public static int executeUpdate(String sql) throws SQLException{ + public static void executeUpdate(String sql) throws SQLException{ try (Connection connection = getConnection()) { - PreparedStatement useStatement = connection.prepareStatement("USE `playersync`"); + PreparedStatement useStatement = connection.prepareStatement("USE ?"); + useStatement.setString(1, JdbcConfig.DATABASE_NAME.get()); useStatement.executeUpdate(); try (PreparedStatement updateStatement = connection.prepareStatement(sql)) { - return updateStatement.executeUpdate(); + updateStatement.executeUpdate(); } } } - public static int executeUpdate(String sql,int i) throws SQLException{ + public static void Update(String sql, String... argument) throws SQLException{ + Connection connection = getConnection(); + + PreparedStatement useStatement = connection.prepareStatement("USE ?"); + useStatement.setString(1, JdbcConfig.DATABASE_NAME.get()); + useStatement.executeUpdate(); + + PreparedStatement updateStatement = connection.prepareStatement(sql); + for (int i = 1; i <= argument.length; i++) { + updateStatement.setString(i,argument[i]); + } + updateStatement.executeUpdate(); + } + + public static void executeUpdate(String sql, int i) throws SQLException{ try (Connection connection = getConnection()) { try (PreparedStatement updateStatement = connection.prepareStatement(sql)) { - return updateStatement.executeUpdate(); + updateStatement.executeUpdate(); } } } - public static class QueryResult{ - private final Connection connection; - private final ResultSet resultSet; - - public QueryResult(Connection connection, ResultSet resultSet) { - this.connection = connection; - this.resultSet = resultSet; - } - - public Connection getConnection() { - return connection; - } - - public ResultSet getResultSet() { - return resultSet; - } + public record QueryResult(Connection connection, ResultSet resultSet) { } }