From 98860ffd02ef2ffcddbecbc1cd1de9bc15758815 Mon Sep 17 00:00:00 2001
From: md_5 <md_5@live.com.au>
Date: Wed, 22 May 2013 17:07:46 +1000
Subject: [PATCH] SLightly more atomic locations.yml saving to guard against
 ctrl+c'ing users

---
 .../java/net/md_5/bungee/YamlReconnectHandler.java    | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/proxy/src/main/java/net/md_5/bungee/YamlReconnectHandler.java b/proxy/src/main/java/net/md_5/bungee/YamlReconnectHandler.java
index 47c6bf14..b445f418 100644
--- a/proxy/src/main/java/net/md_5/bungee/YamlReconnectHandler.java
+++ b/proxy/src/main/java/net/md_5/bungee/YamlReconnectHandler.java
@@ -20,7 +20,8 @@ public class YamlReconnectHandler implements ReconnectHandler
 {
 
     private final Yaml yaml = new Yaml();
-    private final File file = new File( "locations.yml" );
+    private final File readFile = new File( "locations.yml" );
+    private final File writeFile = new File( "locations.yml~" );
     /*========================================================================*/
     private Map<String, String> data;
 
@@ -29,8 +30,8 @@ public class YamlReconnectHandler implements ReconnectHandler
     {
         try
         {
-            file.createNewFile();
-            try ( FileReader rd = new FileReader( file ) )
+            readFile.createNewFile();
+            try ( FileReader rd = new FileReader( readFile ) )
             {
                 data = yaml.loadAs( rd, Map.class );
             }
@@ -85,9 +86,11 @@ public class YamlReconnectHandler implements ReconnectHandler
     @Override
     public synchronized void save()
     {
-        try ( FileWriter wr = new FileWriter( file ) )
+        try ( FileWriter wr = new FileWriter( writeFile ) )
         {
             yaml.dump( data, wr );
+            readFile.delete();
+            writeFile.renameTo( readFile );
         } catch ( IOException ex )
         {
             ProxyServer.getInstance().getLogger().log( Level.WARNING, "Could not save reconnect locations", ex );