Make service pluggable so developers can plug-in custom executor-serv…

…ice (#1993)
lift · Jul 20, 2021 · 8e74e41 · 8e74e41
1 parent bb8fb65
commit 8e74e41
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/util/src/main/scala/net/liftweb/util/Schedule.scala b/core/util/src/main/scala/net/liftweb/util/Schedule.scala
@@ -72,16 +72,17 @@ sealed trait Schedule extends Loggable {
 
 
   /** The underlying <code>java.util.concurrent.ScheduledExecutor</code> */
-  private var service: ScheduledExecutorService = Executors.newSingleThreadScheduledExecutor(TF)
+  @volatile var buildService: () => ScheduledExecutorService = () => Executors.newSingleThreadScheduledExecutor(TF)
 
+  private var service: ScheduledExecutorService = buildService()
   private var pool = buildExecutor()
 
   /**
    * Re-create the underlying <code>SingleThreadScheduledExecutor</code>
    */
   def restart: Unit = synchronized
   { if ((service eq null) || service.isShutdown)
-    service = Executors.newSingleThreadScheduledExecutor(TF) 
+    service = buildService()
    if ((pool eq null) || pool.isShutdown)
      pool = buildExecutor()
  }