proc-manager.sml

来自「这是我们参加06年全国开源软件的竞赛作品」· SML 代码 · 共 62 行

(* proc-manager.sml
 *
 * COPYRIGHT (c) 1998 Bell Labs, Lucent Technologies.
 * COPYRIGHT (c) 1995 AT&T Bell Laboratories.
 * COPYRIGHT (c) 1989-1991 John H. Reppy
 *
 * Win32 process management (from its Unix counterpart)
 *
 *)

structure ProcManager : sig

    val addPid : Win32Process.pid -> Win32Process.exit_status Event.event

    val pollProcs : unit -> unit

    val anyWaiting : unit -> bool

  end = struct

    structure S = Scheduler
    structure WP = Win32Process
    structure R = Result

    datatype pid = PID of {
	wait : WP.exit_status R.result,
	pid : WP.pid
      }

    val waiting = ref ([] : pid list)

    fun addPid pid = let
	  val rv = Result.result()
	  in
	    waiting := PID{wait = rv, pid = pid} :: !waiting;
	    Result.getEvt rv
	  end

    fun pollProcs () = let
(** NOTE: it would be more efficient to poll for any zombie process,
 ** until there are no more.
 **)
	  fun pollPid pid = WP.waitForSingleObject pid
	  fun pollItem (item as PID{wait, pid}) = (
		case (pollPid pid)
		 of SOME(sts) => (
		      S.enqueueTmpThread (fn () => R.put(wait, sts));
		      false)
		  | NONE => true
		(* end case *))
		  handle ex => (
		    S.enqueueTmpThread (fn () => R.putExn (wait, ex));
		    false)
	  in
	    waiting := List.filter pollItem (! waiting)
	  end

    fun anyWaiting () = (case !waiting of [] => false | _ => true)

  end