Advertisement
Guest User

Untitled

a guest
Mar 16th, 2017
74
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Haxe 3.05 KB | None | 0 0
  1. package;
  2. import neko.vm.Mutex;
  3. import neko.vm.Thread;
  4.  
  5. class PoolThread {
  6.     private var _name:String;
  7.     private var _pool:ThreadPool;
  8.    
  9.     private var _thread:Thread;
  10.    
  11.     public function new(name:String, pool:ThreadPool) {
  12.         _name = name;
  13.         _pool = pool;
  14.        
  15.         _thread = Thread.create(run);
  16.         _thread.sendMessage(Thread.current());
  17.         _thread.sendMessage(_pool);
  18.         Thread.readMessage(true);
  19.     }
  20.    
  21.     public function assign(task:ThreadTask) {
  22.         _thread.sendMessage(task);
  23.     }
  24.    
  25.     @:access(ThreadPool)
  26.     private function run() {
  27.         trace('${_name} running\n');
  28.         var callingThread:Thread = Thread.readMessage(true);
  29.         var pool:ThreadPool = Thread.readMessage(true);
  30.         callingThread.sendMessage(null);
  31.        
  32.         while (true) {
  33.             trace('${_name} waiting\n');
  34.             var message = Thread.readMessage(true);
  35.            
  36.             if (Std.is(message, ThreadTask)) {
  37.                 var task:ThreadTask = cast(message, ThreadTask);
  38.                 trace('${_name} assigned\n');
  39.                
  40.                 var result = task.task(task.arg);
  41.                
  42.                 trace('${_name} complete - ${result}\n');
  43.                 pool.threadTaskComplete(this);
  44.             }
  45.         }
  46.        
  47.         trace('${_name} ended\n');
  48.     }
  49. }
  50.  
  51. class ThreadTask {
  52.     public var task:Dynamic->Dynamic;
  53.     public var arg:Dynamic;
  54.    
  55.     public function new(task:Dynamic->Dynamic, arg:Dynamic) {
  56.         this.task = task;
  57.         this.arg = arg;
  58.     }
  59. }
  60.  
  61. class ThreadPool {
  62.     private var _threads:Array<PoolThread>;
  63.     private var _threadPool:Array<PoolThread>;
  64.     private var _lock:Mutex;
  65.    
  66.     private var _tasks:Array<ThreadTask>;
  67.    
  68.     public function new(size:Int, namePrefix:String = "thread-") {
  69.         _threads = [];
  70.         _threadPool = [];
  71.         _lock = new Mutex();
  72.        
  73.         _tasks = [];
  74.        
  75.         for (i in 0...size) {
  76.             var thread:PoolThread = new PoolThread('${namePrefix}${i}', this);
  77.             _threads.push(thread);
  78.             _threadPool.push(thread);
  79.         }
  80.     }
  81.    
  82.     public function addTask(task:Dynamic->Dynamic, arg:Dynamic = null) {
  83.         _lock.acquire();
  84.         _tasks.push(new ThreadTask(task, arg));
  85.         _lock.release();
  86.        
  87.         checkTasks();
  88.     }
  89.    
  90.     private function threadTaskComplete(thread:PoolThread) {
  91.         _lock.acquire();
  92.         _threadPool.push(thread);
  93.         _lock.release();
  94.         checkTasks();
  95.     }
  96.    
  97.     private function checkTasks() {
  98.         var thread:PoolThread = _threadPool.pop();
  99.         if (thread == null) {
  100.             return;
  101.         }
  102.        
  103.         if (_tasks.length == 0) {
  104.             trace("DONE!");
  105.             //Thread.current().sendMessage("all complete");
  106.             return;
  107.         }
  108.        
  109.         _lock.acquire();
  110.         var task = _tasks[0];
  111.         _tasks.remove(task);
  112.         _lock.release();
  113.        
  114.         thread.assign(task);
  115.     }
  116. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement