initial commit

2017-06-19 17:30:18 -04:00
commit 6e265ea24b
763 changed files with 488824 additions and 0 deletions
@@ -0,0 +1,109 @@
+/*
+ * Copyright (C) 2012-2017 CypherCore <http://github.com/CypherCore>
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+using System;
+using System.Collections.Generic;
+using System.Diagnostics.Contracts;
+using System.Linq;
+using System.Threading.Tasks;
+
+namespace Framework.Database
+{
+    public class QueryCallback
+    {
+        public QueryCallback(Task<SQLResult> result)
+        {
+            _result = result;
+        }
+
+        public QueryCallback WithCallback(Action<SQLResult> callback)
+        {
+            return WithChainingCallback((queryCallback, result) => callback(result));
+        }
+
+        public QueryCallback WithCallback<T>(Action<T, SQLResult> callback, T obj)
+        {
+            return WithChainingCallback((queryCallback, result) => callback(obj, result));
+        }
+
+        public QueryCallback WithChainingCallback(Action<QueryCallback, SQLResult> callback)
+        {
+            _callbacks.Add(new QueryCallbackData(callback));
+            return this;
+        }
+
+        public void SetNextQuery(QueryCallback next)
+        {
+            _result = next._result;
+        }
+
+        public QueryCallbackStatus InvokeIfReady()
+        {
+            QueryCallbackData callback = _callbacks.First();
+            if (_result.IsCompleted)
+            {
+                Task<SQLResult> f = _result;
+                Action<QueryCallback, SQLResult> cb = callback._result;
+                _result = null;
+                callback.Clear();
+
+                cb(this, f.Result);
+
+                _callbacks.RemoveAt(0);
+                bool hasNext = _result != null;
+                if (_callbacks.Empty())
+                {
+                    Contract.Assert(!hasNext);
+                    return QueryCallbackStatus.Completed;
+                }
+
+                // abort chain
+                if (!hasNext)
+                    return QueryCallbackStatus.Completed;
+
+                return QueryCallbackStatus.NextStep;
+            }
+
+            return QueryCallbackStatus.NotReady;
+        }
+
+        Task<SQLResult> _result;
+        List<QueryCallbackData> _callbacks = new List<QueryCallbackData>();
+    }
+
+    struct QueryCallbackData
+    {
+        public QueryCallbackData(Action<QueryCallback, SQLResult> callback)
+        {
+            _result = callback;
+        }
+
+        public void Clear()
+        {
+            _result = null;
+        }
+
+        public Action<QueryCallback, SQLResult> _result;
+    }
+
+    public enum QueryCallbackStatus
+    {
+        NotReady,
+        NextStep,
+        Completed
+    }
+}