task-tls: Tweaks and fixes, task_id helper

Author: jpsamaroo

src/task-tls.jl

@@ -1,6 +1,6 @@
 # In-Thunk Helpers
 
-struct DTaskTLS
+mutable struct DTaskTLS
     processor::Processor
     sch_uid::UInt
     sch_handle::Any # FIXME: SchedulerHandle
@@ -10,6 +10,8 @@ end
 
 const DTASK_TLS = TaskLocalValue{Union{DTaskTLS,Nothing}}(()->nothing)
 
+Base.copy(tls::DTaskTLS) = DTaskTLS(tls.processor, tls.sch_uid, tls.sch_handle, tls.task_spec, tls.cancel_token)
+
 """
     get_tls() -> DTaskTLS
 
@@ -32,22 +34,29 @@ end
 Returns `true` if currently executing in a [`DTask`](@ref), else `false`.
 """
 in_task() = DTASK_TLS[] !== nothing
-@deprecate in_thunk() in_task()
+@deprecate(in_thunk(), in_task())
+
+"""
+    task_id() -> Int
+
+Returns the ID of the current [`DTask`](@ref).
+"""
+task_id() = get_tls().sch_handle.thunk_id.id
 
 """
     task_processor() -> Processor
 
 Get the current processor executing the current [`DTask`](@ref).
 """
 task_processor() = get_tls().processor
-@deprecate thunk_processor() task_processor()
+@deprecate(thunk_processor(), task_processor())
 
 """
     task_cancelled() -> Bool
 
 Returns `true` if the current [`DTask`](@ref) has been cancelled, else `false`.
 """
-task_cancelled() = get_tls().cancel_token.cancelled[]
+task_cancelled() = is_cancelled(get_tls().cancel_token)
 
 """
     task_may_cancel!()
@@ -59,3 +68,10 @@ function task_may_cancel!()
         throw(InterruptException())
     end
 end
+
+"""
+    task_cancel!()
+
+Cancels the current [`DTask`](@ref).
+"""
+task_cancel!() = cancel!(get_tls().cancel_token)