ThreadHandler/CFX-Fivem-ThreadHandler.cs at main · Sonali823/ThreadHandler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
namespace CFX-Fivem-ThreadHandler {
    public sealed class ThreadHandler {
        public static bool IsMainThread => Thread.CurrentThread.ManagedThreadId == 3;

        public static async Task ConditionalJumpToMainThread() {
            if (!IsMainThread)
                await BaseScript.Delay(0);
        }

        public static void ThrowIfNotMainThread() {
            if (!IsMainThread) throw new Exception("!!! NOT ON MAIN THREAD");
        }

        public static async Task CallOnMainThread(string name, Action functionToInvoke) {
            if (IsMainThread) {
                functionToInvoke();
                return;
            }

            await Task.Run(async () => {
                await BaseScript.Delay(0);
                ThrowIfNotMainThread();
                functionToInvoke();
                ThrowIfNotMainThread();
            })
            .ContinueWith(t => {
                if (t.IsFaulted)
                    throw t.Exception;
            }).ConfigureAwait(false);
        }

        public static async Task CallOnMainThread(string name, Func<Task> functionToInvoke) {
            if (IsMainThread) {
                await functionToInvoke().ConfigureAwait(false);
                return;
            }

            await Task.Run(async () => {
                await BaseScript.Delay(0);
                ThrowIfNotMainThread();
                await functionToInvoke().ConfigureAwait(false);
                ThrowIfNotMainThread();
            })
            .ContinueWith(t => {
                if (t.IsFaulted)
                    throw t.Exception;
            }).ConfigureAwait(false);
        }

        public static async Task<T> CallOnMainThread<T>(string name, Func<T> functionToInvoke) {
            if (IsMainThread) {
                return functionToInvoke();
            }

            return await Task.Run(async () => {
                await BaseScript.Delay(0);
                ThrowIfNotMainThread();
                var temp = functionToInvoke();
                ThrowIfNotMainThread();
                return temp;
            })
            .ContinueWith(t => {
                if (t.IsFaulted)
                    throw t.Exception;
                return t.Result;
            }).ConfigureAwait(false);
        }
    }
}