More ServerControl improvements (#9038)

* More ServerControl improvements

* Update server/src/mindustry/server/ServerControl.java

---------

Co-authored-by: Anuken <arnukren@gmail.com>

server/src/mindustry/server/ServerControl.java

@@ -343,7 +343,7 @@ public class ServerControl implements ApplicationListener{
 
         handler.register("stop", "Stop hosting the server.", arg -> {
             net.closeServer();
-            if(lastTask != null) lastTask.cancel();
+            cancelPlayTask();
             state.set(State.menu);
             info("Stopped server.");
         });
@@ -354,7 +354,7 @@ public class ServerControl implements ApplicationListener{
                 return;
             }
 
-            if(lastTask != null) lastTask.cancel();
+            cancelPlayTask();
 
             Gamemode preset = Gamemode.survival;
 
@@ -1063,6 +1063,13 @@ public class ServerControl implements ApplicationListener{
         maps.setNextMapOverride(map);
     }
 
+    /**
+     * Cancels the world load timer task, if it is scheduled. Can be useful for stopping a server or hosting a new game.
+     */
+    public void cancelPlayTask(){
+        if(lastTask != null) lastTask.cancel();
+    }
+
     /**
      * Resets the world state, starts a new game.
      * @param run What task to run to load a new world.
@@ -1078,11 +1085,11 @@ public class ServerControl implements ApplicationListener{
      */
     public void play(boolean wait, Runnable run){
         inGameOverWait = true;
-        if(lastTask != null) lastTask.cancel();
+        cancelPlayTask();
         
-        Runnable r = () -> {
+        Runnable reload = () -> {
+            try{
                 WorldReloader reloader = new WorldReloader();
-
                 reloader.begin();
 
                 run.run();
@@ -1092,24 +1099,16 @@ public class ServerControl implements ApplicationListener{
 
                 reloader.end();
                 inGameOverWait = false;
-        };
-
-        if(wait){
-            lastTask = new Task(){
-                @Override
-                public void run(){
-                    try{
-                        r.run();
             }catch(MapException e){
                 err("@: @", e.map.plainName(), e.getMessage());
                 net.closeServer();
             }
-                }
         };
 
-            Timer.schedule(lastTask, Config.roundExtraTime.num());
+        if(wait){
+            lastTask = Timer.schedule(reload, Config.roundExtraTime.num());
         }else{
-            r.run();
+            reload.run();
         }
     }