FiberTaskScheduler.cs SpicyPixel::Threading::Tasks::FiberTaskScheduler SpicyPixel SpicyPixel::Threading SpicyPixel::Threading::Tasks /* Author:AaronOneal,http://aarononeal.info Copyright(c)2012SpicyPixel,http://spicypixel.com Permissionisherebygranted,freeofcharge,toanypersonobtaining acopyofthissoftwareandassociateddocumentationfiles(the "Software"),todealintheSoftwarewithoutrestriction,including withoutlimitationtherightstouse,copy,modify,merge,publish, distribute,sublicense,and/orsellcopiesoftheSoftware,andto permitpersonstowhomtheSoftwareisfurnishedtodoso,subjectto thefollowingconditions: Theabovecopyrightnoticeandthispermissionnoticeshallbe includedinallcopiesorsubstantialportionsoftheSoftware. THESOFTWAREISPROVIDED"ASIS",WITHOUTWARRANTYOFANYKIND, EXPRESSORIMPLIED,INCLUDINGBUTNOTLIMITEDTOTHEWARRANTIESOF MERCHANTABILITY,FITNESSFORAPARTICULARPURPOSEAND NONINFRINGEMENT.INNOEVENTSHALLTHEAUTHORSORCOPYRIGHTHOLDERSBE LIABLEFORANYCLAIM,DAMAGESOROTHERLIABILITY,WHETHERINANACTION OFCONTRACT,TORTOROTHERWISE,ARISINGFROM,OUTOFORINCONNECTION WITHTHESOFTWAREORTHEUSEOROTHERDEALINGSINTHESOFTWARE. */ usingSystem; usingSystem.Collections; usingSystem.Collections.Generic; usingSystem.Threading; usingSystem.Threading.Tasks; namespaceSpicyPixel.Threading.Tasks { publicsealedclassFiberTaskScheduler:TaskScheduler,IDisposable { FiberSchedulerscheduler; publicCancellationTokenCancellationToken { get{returnscheduler.CancellationToken;} } publicFiberTaskScheduler():this(FiberScheduler.Current) { } publicFiberTaskScheduler(FiberSchedulerscheduler) { this.scheduler=scheduler==null?FiberScheduler.Current:scheduler; } publicFiberSchedulerFiberScheduler { get{returnscheduler;} } protectedinternaloverridevoidQueueTask(Tasktask) { //Startafibertorunthetask Fiber.Factory.StartNew(ExecuteTask(task),scheduler); } protectedoverrideboolTryExecuteTaskInline(Tasktask,booltaskWasPreviouslyQueued) { //1.Tasksusingfiberschedulershavethreadaffinityandmustrunonthethreadtheschedulerwascreatedon. //2.Fibertaskscannotruninlineherebecausetheymayyieldwhichisonlyhandledwhenqueued. // //Thatbeingsaid,thefiberschedulermaychoosetoinlineanywaywhen //queueingoccursontheschedulerthread,itjustdoesn'thappeninthismethod. if(scheduler.SchedulerThread!=Thread.CurrentThread||taskisYieldableTask) returnfalse; returnTryExecuteTask(task); } #regionimplementedabstractmembersofSystem.Threading.Tasks.TaskScheduler protectedinternaloverrideboolTryDequeue(Tasktask) { returnfalse; } protectedoverrideIEnumerable<Task>GetScheduledTasks() { returnnull; } #endregion privateIEnumeratorExecuteTask(Tasktask) { //ExecuteyieldableactionsusingtheMonoBehavior. // //Thecontinuationwillexecutethetaskitself //viaTryExecuteTask().Thattaskactionwill //rethrowanyexceptionscaughtduringtheyieldable //actioninvocationandTryExecuteTask()willset //thefinalstate. // //Thedownsidetothisapproachisthatyieldabletasks //aren'tconsideredrunninguntiltheyhavealready //completedbecauseTryExecuteTask()iswhatsetsstate. varyieldableTask=taskasYieldableTask; if(yieldableTask!=null) yieldreturnFiber.Factory.StartNew(ExecuteYieldableTask(yieldableTask),scheduler); //Runtheaction TryExecuteTask(task); } privateIEnumeratorExecuteYieldableTask(YieldableTasktask) { objectfiberResult=null; //TODO:Cleanup // //Thisusesinternalsetterstofakestartingthefiber //andthenusesinternalExecute().Itshouldbepossibleto //usepublicAPIsinsteadto: //1.Setafiberpropertytoassociatethetask //2.SetanexceptionhandlerontheFiberScheduler //3.Retrievethetaskinthehandlerfromthefiberandset //theexeption //4.Returnfromthehandlerwithoutrethrowing // //Thismethodcouldberemovedthenandtheinternalsetters //removedaswell. task.Fiber.Scheduler=scheduler; task.Fiber.Status=FiberStatus.Running; while(true) { try { //Throwherebecausetheschedulerisdisposedandwillnot //runanythingelse.Nofurtheraccessisvalid. CancellationToken.ThrowIfCancellationRequested(); fiberResult=task.Fiber.Execute(); if(fiberResultisStopInstruction){ //Checkforexceptions if(task.Fiber.IsFaulted){ task.FiberException=task.Fiber.Exception; } yieldbreak; } } catch(System.Threading.OperationCanceledExceptionex) { //Fiberexecutiondoesnotthrowsotheonlyexception //expectedisthecancellationabove. task.FiberException=ex; yieldbreak; } yieldreturnfiberResult; } } #regionIDisposableimplementation ~FiberTaskScheduler() { Dispose(false); } privateboolisDisposed=false; publicvoidDispose() { Dispose(true); GC.SuppressFinalize(this); } private/*protectedvirtual*/voidDispose(booldisposing) { //Donothingifalreadycalled if(isDisposed) return; if(disposing) { //Freeotherstate(managedobjects). scheduler.Dispose(); } //Freeyourownstate(unmanagedobjects). //Setlargefieldstonull. //Markdisposed isDisposed=true; } #endregion } }