Add utility to ensure that a kernel with arbitrary workers exist.

Author: aadcg

src/kernel/core.lisp

@@ -260,7 +260,7 @@ A kernel will not be garbage collected until `end-kernel' is called."
     (fill-workers workers kernel)
     kernel))
 
-(defun check-kernel ()
+(defun check-kernel (&key worker-count)
   "Ensures the value of `*kernel*' is a kernel instance. Provides the
 MAKE-KERNEL and STORE-VALUE restarts. Returns `*kernel*'."
   (or *kernel*
@@ -275,6 +275,16 @@ MAKE-KERNEL and STORE-VALUE restarts. Returns `*kernel*'."
           (check-type value kernel)
           (setf *kernel* value)))))
 
+(defun ensure-kernel (worker-count &rest other-keys)
+  "Return `*kernel*', a kernel instance featuring WORKER-COUNT."
+  ;; Notice that when *kernel* exists with 2 workers, then calling:
+  ;; (ensure-kernel N :name "a-different-name-from-*kernel*")
+  ;; has no effect since the number of workers match.
+  (if (and *kernel*
+           (eq (length (workers *kernel*)) worker-count))
+      *kernel*
+      (setf *kernel* (apply #'make-kernel worker-count other-keys))))
+
 (defmacro define-worker-info-reader (name slot &optional (result slot))
   `(defun ,name ()
      ,(format nil "Return the ~a passed to `make-kernel'."

src/kernel/package.lisp

@@ -42,6 +42,7 @@
         #+lparallel.with-stealing-scheduler #:lparallel.spin-queue)
   (:export #:make-kernel
            #:check-kernel
+           #:ensure-kernel
            #:end-kernel
            #:kernel-worker-count
            #:kernel-worker-index