/* * (c) Copyright Ascensio System SIA 2010-2015 * * This program is a free software product. You can redistribute it and/or * modify it under the terms of the GNU Affero General Public License (AGPL) * version 3 as published by the Free Software Foundation. In accordance with * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect * that Ascensio System SIA expressly excludes the warranty of non-infringement * of any third-party rights. * * This program is distributed WITHOUT ANY WARRANTY; without even the implied * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html * * You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia, * EU, LV-1021. * * The interactive user interfaces in modified source and object code versions * of the Program must display Appropriate Legal Notices, as required under * Section 5 of the GNU AGPL version 3. * * Pursuant to Section 7(b) of the License you must retain the original Product * logo when distributing the program. Pursuant to Section 7(e) we decline to * grant you any rights under trademark law for use of our trademarks. * * All the Product's GUI elements, including illustrations and icon sets, as * well as technical writing content are licensed under the terms of the * Creative Commons Attribution-ShareAlike 4.0 International. See the License * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode * */ using System; using System.Collections.Generic; using System.Text; using System.Data; using System.Configuration; using ASC.Common.Data; using ASC.Common.Data.Sql; using ASC.Common.Data.Sql.Expressions; using log4net; namespace FileConverterUtils2 { class TaskResultDataBase : ITaskResultInterface { private static readonly ILog _log = LogManager.GetLogger(typeof(TaskResultDataBase)); private static int mc_nRandomMaxNumber = 10000; private static System.Random m_oRandom = new System.Random(); private string m_sConnectionString = ConfigurationManager.AppSettings["utils.taskresult.db.connectionstring"]; private delegate IDataReader ExecuteReader(); private delegate int ExecuteNonQuery(); private delegate ErrorTypes DelegateAdd(string sKey, TaskResultData oTast); private delegate ErrorTypes DelegateUpdate(string sKey, TaskResultDataToUpdate oTast); private delegate ErrorTypes DelegateUpdateIf(string sKey, TaskResultDataToUpdate oMask, TaskResultDataToUpdate oTast, out bool bUpdate); private delegate ErrorTypes DelegateGet(string sKey, out TaskResultData oTaskResultData); private delegate ErrorTypes DelegateGets(string[] aKeys, out TaskResultData[] aTaskResultDatas); private delegate ErrorTypes DelegateGetEditing(out TaskResultData[] oTast); private delegate ErrorTypes DelegateRemove(string sKey); private class TransportClass : TransportClassAsyncOperation { public ExecuteReader m_delegateReader = null; public ExecuteNonQuery m_delegateNonQuery = null; public IDbConnection m_oSqlCon = null; public IDbCommand m_oCommand = null; public TaskResultData m_oTast = null; public ErrorTypes m_eError = ErrorTypes.NoError; public bool m_bCreate = false; public TransportClass(AsyncCallback fCallback, object oParam) : base(fCallback, oParam) { } public override void Close() { if (null != m_oCommand) { m_oCommand.Dispose(); m_oCommand = null; } if (null != m_oSqlCon) { m_oSqlCon.Close(); m_oSqlCon.Dispose(); m_oSqlCon = null; } } public override void Dispose() { m_eError = ErrorTypes.TaskResult; try { if (null != m_oCommand) { m_oCommand.Dispose(); m_oCommand = null; } if (null != m_oSqlCon) { m_oSqlCon.Dispose(); m_oSqlCon = null; } } catch { } } } private TransportClass m_oAddRandomKey = null; private TransportClass m_oGetOrCreate = null; private DelegateAdd m_oAdd = null; private DelegateUpdate m_oUpdate = null; private DelegateUpdateIf m_oUpdateIf = null; private DelegateGet m_oGet = null; private DelegateGets m_oGets = null; private DelegateGetEditing m_oGetEditing = null; private DelegateRemove m_oRemove = null; public ErrorTypes Add(string sKey, TaskResultData oTast) { ErrorTypes eError = ErrorTypes.NoError; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetINSERTString(oTast); oSelCommand.ExecuteNonQuery(); } } } catch { eError = ErrorTypes.TaskResult; } return eError; } public void AddBegin(string sKey, TaskResultData oTast, AsyncCallback fCallback, object oParam) { DelegateAdd oDelegateAdd = new DelegateAdd(Add); m_oAdd = oDelegateAdd; oDelegateAdd.BeginInvoke(sKey, oTast, fCallback, oParam); } public ErrorTypes AddEnd(IAsyncResult ar) { ErrorTypes eError = ErrorTypes.NoError; try { eError = m_oAdd.EndInvoke(ar); } catch { eError = ErrorTypes.TaskResult; } return eError; } public ErrorTypes AddRandomKey(string sKey, TaskResultData oTastToAdd, out TaskResultData oTastAdded) { ErrorTypes eError = ErrorTypes.TaskResult; oTastAdded = oTastToAdd.Clone(); try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); while (true) { string sNewKey = sKey + "_" + m_oRandom.Next(mc_nRandomMaxNumber); oTastAdded.sKey = sNewKey; using (IDbCommand oAddCommand = sqlCon.CreateCommand()) { oAddCommand.CommandText = GetINSERTString(oTastAdded); bool bExist = false; try { oAddCommand.ExecuteNonQuery(); } catch { bExist = true; } if (false == bExist) { eError = ErrorTypes.NoError; break; } } } } } catch { eError = ErrorTypes.TaskResult; } if (ErrorTypes.NoError != eError) { oTastAdded = null; } return eError; } public void AddRandomKeyBegin(string sKey, TaskResultData oTastToAdd, AsyncCallback fCallback, object oParam) { m_oAddRandomKey = new TransportClass(fCallback, oParam); m_oAddRandomKey.m_oTast = oTastToAdd.Clone(); try { m_oAddRandomKey.m_oSqlCon = GetDbConnection(); m_oAddRandomKey.m_oSqlCon.Open(); string sNewKey = sKey + "_" + m_oRandom.Next(mc_nRandomMaxNumber); m_oAddRandomKey.m_oTast.sKey = sNewKey; IDbCommand oSelCommand = m_oAddRandomKey.m_oSqlCon.CreateCommand(); m_oAddRandomKey.m_oCommand = oSelCommand; oSelCommand.CommandText = GetINSERTString(m_oAddRandomKey.m_oTast); m_oAddRandomKey.m_delegateNonQuery = new ExecuteNonQuery(oSelCommand.ExecuteNonQuery); m_oAddRandomKey.m_delegateNonQuery.BeginInvoke(AddRandomKeyCallback, null); } catch { m_oAddRandomKey.DisposeAndCallback(); } } public ErrorTypes AddRandomKeyEnd(IAsyncResult ar, out TaskResultData oTast) { oTast = null; if (ErrorTypes.NoError == m_oAddRandomKey.m_eError) { oTast = m_oAddRandomKey.m_oTast; } return m_oAddRandomKey.m_eError; } public ErrorTypes Update(string sKey, TaskResultDataToUpdate oTast) { ErrorTypes eErrorTypes = ErrorTypes.NoError; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetUPDATEString(sKey, oTast); oSelCommand.ExecuteNonQuery(); } } } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public void UpdateBegin(string sKey, TaskResultDataToUpdate oTast, AsyncCallback fCallback, object oParam) { m_oUpdate = Update; m_oUpdate.BeginInvoke(sKey, oTast, fCallback, oParam); } public ErrorTypes UpdateEnd(IAsyncResult ar) { ErrorTypes eErrorTypes = ErrorTypes.NoError; try { eErrorTypes = m_oUpdate.EndInvoke(ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public ErrorTypes UpdateIf(string sKey, TaskResultDataToUpdate oMask, TaskResultDataToUpdate oTast, out bool bUpdate) { ErrorTypes eErrorTypes = ErrorTypes.NoError; bUpdate = false; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetUPDATEIFString(sKey, oMask, oTast); if (oSelCommand.ExecuteNonQuery() > 0) bUpdate = true; } } } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public void UpdateIfBegin(string sKey, TaskResultDataToUpdate oMask, TaskResultDataToUpdate oTast, AsyncCallback fCallback, object oParam) { m_oUpdateIf = UpdateIf; bool bUpdate; m_oUpdateIf.BeginInvoke(sKey, oMask, oTast, out bUpdate, fCallback, oParam); } public ErrorTypes UpdateIfEnd(IAsyncResult ar, out bool bUpdate) { ErrorTypes eErrorTypes = ErrorTypes.NoError; bUpdate = false; try { eErrorTypes = m_oUpdateIf.EndInvoke(out bUpdate, ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public ErrorTypes Get(string sKey, out TaskResultData oTaskResultData) { ErrorTypes eErrorTypes = ErrorTypes.TaskResult; oTaskResultData = null; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetSELECTString(sKey); using (IDataReader oReader = oSelCommand.ExecuteReader()) { if (true == oReader.Read()) { oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); eErrorTypes = ErrorTypes.NoError; } } } } } catch { } return eErrorTypes; } public void GetBegin(string sKey, AsyncCallback fCallback, object oParam) { TaskResultData oTast; m_oGet = Get; m_oGet.BeginInvoke(sKey, out oTast, fCallback, oParam); } public ErrorTypes GetEnd(IAsyncResult ar, out TaskResultData oTast) { oTast = null; ErrorTypes eErrorTypes = ErrorTypes.NoError; try { eErrorTypes = m_oGet.EndInvoke(out oTast, ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public ErrorTypes Get(string[] aKeys, out TaskResultData[] sTaskResults) { ErrorTypes eErrorTypes = ErrorTypes.NoError; sTaskResults = new TaskResultData[aKeys.Length]; for (int i = 0, length = sTaskResults.Length; i < length; ++i) sTaskResults[i] = null; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetSELECTString(aKeys); using (IDataReader oReader = oSelCommand.ExecuteReader()) { Dictionary mapKeyToIndex = new Dictionary(); for (int i = 0, length = aKeys.Length; i < length; ++i) mapKeyToIndex[aKeys[i]] = i; while (oReader.Read()) { TaskResultData oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); int nIndex; if (mapKeyToIndex.TryGetValue(oTaskResultData.sKey, out nIndex)) sTaskResults[nIndex] = oTaskResultData; } } } } } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public void GetBegin(string[] aKeys, AsyncCallback fCallback, object oParam) { TaskResultData[] oTast; m_oGets = Get; m_oGets.BeginInvoke(aKeys, out oTast, fCallback, oParam); } public ErrorTypes GetEnd(IAsyncResult ar, out TaskResultData[] sTaskResults) { ErrorTypes eErrorTypes = ErrorTypes.NoError; sTaskResults = null; try { eErrorTypes = m_oGets.EndInvoke(out sTaskResults, ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public ErrorTypes GetEditing(out TaskResultData[] oTast) { ErrorTypes eErrorTypes = ErrorTypes.NoError; oTast = new TaskResultData[0]; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetSELECTEditingString(); using (IDataReader oReader = oSelCommand.ExecuteReader()) { List aData = new List(); while (oReader.Read()) { TaskResultData oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); aData.Add(oTaskResultData); } oTast = aData.ToArray(); } } } } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public void GetEditingBegin(AsyncCallback fCallback, object oParam) { TaskResultData[] oTast; m_oGetEditing = GetEditing; m_oGetEditing.BeginInvoke(out oTast, fCallback, oParam); } public ErrorTypes GetEditingEnd(IAsyncResult ar, out TaskResultData[] oTast) { ErrorTypes eErrorTypes = ErrorTypes.NoError; oTast = new TaskResultData[0]; try { eErrorTypes = m_oGetEditing.EndInvoke(out oTast, ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } public ErrorTypes GetExpired(int nMaxCount, out List aTasts) { aTasts = null; ErrorTypes eErrorTypes = ErrorTypes.NoError; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetSELECTExpiredTasksString(nMaxCount); using (IDataReader oReader = oSelCommand.ExecuteReader()) { aTasts = new List(); while (oReader.Read()) { TaskResultData oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); aTasts.Add(oTaskResultData); eErrorTypes = ErrorTypes.NoError; } } } } } catch { } return eErrorTypes; } public ErrorTypes GetOrCreate(string sKey, TaskResultData oDefaultTast, out TaskResultData oTaskResultData, out bool bCreate) { ErrorTypes eErrorTypes = ErrorTypes.TaskResult; oTaskResultData = null; bCreate = false; try { using (IDbConnection sqlCon = GetDbConnection()) { sqlCon.Open(); if (Constants.mc_sPostgreProvider == Utils.GetDbConnectionProviderName(m_sConnectionString)) { using (IDbCommand oAddCommand = sqlCon.CreateCommand()) { oAddCommand.CommandText = GetINSERTString(oDefaultTast); bool bExist = false; try { oAddCommand.ExecuteNonQuery(); } catch { bExist = true; } if (bExist) { using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { TaskResultDataToUpdate oToUpdate = new TaskResultDataToUpdate(); oToUpdate.oLastOpenDate = DateTime.UtcNow; oSelCommand.CommandText = GetUPDATEString(sKey, oToUpdate) + GetSELECTString(sKey); using (IDataReader oReader = oSelCommand.ExecuteReader()) { if (true == oReader.Read()) { oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); bCreate = false; eErrorTypes = ErrorTypes.NoError; } } } } else { oTaskResultData = oDefaultTast.Clone(); bCreate = true; eErrorTypes = ErrorTypes.NoError; } } } else { using (IDbCommand oAddCommand = sqlCon.CreateCommand()) { oAddCommand.CommandText = GetUPSERTString(oDefaultTast); bool bExist = false; try { int nNumRowsAffected = oAddCommand.ExecuteNonQuery(); oTaskResultData = oDefaultTast.Clone(); eErrorTypes = ErrorTypes.NoError; bCreate = (nNumRowsAffected == 1); bExist = (nNumRowsAffected > 1); } catch { bExist = true; } if (bExist) { using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetSELECTString(sKey); using (IDataReader oReader = oSelCommand.ExecuteReader()) { if (true == oReader.Read()) { oTaskResultData = new TaskResultData(); TaskResultFromReader(oTaskResultData, oReader); bCreate = false; eErrorTypes = ErrorTypes.NoError; } } } } } } } } catch { } return eErrorTypes; } public void GetOrCreateBegin(string sKey, TaskResultData oDataToAdd, AsyncCallback fCallback, object oParam) { m_oGetOrCreate = new TransportClass(fCallback, oParam); m_oGetOrCreate.m_oTast = oDataToAdd; m_oGetOrCreate.m_bCreate = true; try { m_oGetOrCreate.m_oSqlCon = GetDbConnection(); m_oGetOrCreate.m_oSqlCon.Open(); if (Constants.mc_sPostgreProvider == Utils.GetDbConnectionProviderName(m_sConnectionString)) { IDbCommand oInsCommand = m_oGetOrCreate.m_oSqlCon.CreateCommand(); oInsCommand.CommandText = GetINSERTString(m_oGetOrCreate.m_oTast); m_oGetOrCreate.m_oCommand = oInsCommand; m_oGetOrCreate.m_delegateNonQuery = new ExecuteNonQuery(oInsCommand.ExecuteNonQuery); m_oGetOrCreate.m_delegateNonQuery.BeginInvoke(GetOrCreateCallback, null); } else { IDbCommand oSelCommand = m_oGetOrCreate.m_oSqlCon.CreateCommand(); oSelCommand.CommandText = GetUPSERTString(m_oGetOrCreate.m_oTast); m_oGetOrCreate.m_oCommand = oSelCommand; m_oGetOrCreate.m_delegateNonQuery = new ExecuteNonQuery(oSelCommand.ExecuteNonQuery); m_oGetOrCreate.m_delegateNonQuery.BeginInvoke(GetOrCreateCallback2, null); } } catch { m_oGetOrCreate.DisposeAndCallback(); } } public ErrorTypes GetOrCreateEnd(IAsyncResult ar, out TaskResultData oDataAdded, out bool bCreate) { bCreate = m_oGetOrCreate.m_bCreate; oDataAdded = null; if (ErrorTypes.NoError == m_oGetOrCreate.m_eError) { try { if (null != m_oGetOrCreate.m_delegateReader) { using (IDataReader oReader = m_oGetOrCreate.m_delegateReader.EndInvoke(ar)) { if (true == oReader.Read()) { m_oGetOrCreate.m_oTast = new TaskResultData(); TaskResultFromReader(m_oGetOrCreate.m_oTast, oReader); } else m_oGetOrCreate.m_eError = ErrorTypes.TaskResult; } } m_oGetOrCreate.Close(); if (null != m_oGetOrCreate.m_oTast) oDataAdded = m_oGetOrCreate.m_oTast.Clone(); } catch(Exception e) { _log.Error("Exception catch in GetOrCreateEnd:", e); m_oGetOrCreate.Dispose(); } } return m_oGetOrCreate.m_eError; } public ErrorTypes Remove(string sKey) { ErrorTypes eErrorTypes = ErrorTypes.TaskResult; using (IDbConnection sqlCon = GetDbConnection()) { try { sqlCon.Open(); using (IDbCommand oSelCommand = sqlCon.CreateCommand()) { oSelCommand.CommandText = GetDELETEString(sKey); oSelCommand.ExecuteNonQuery(); eErrorTypes = ErrorTypes.NoError; } } catch { } } return eErrorTypes; } public void RemoveBegin(string sKey, AsyncCallback fCallback, object oParam) { m_oRemove = Remove; m_oRemove.BeginInvoke(sKey, fCallback, oParam); } public ErrorTypes RemoveEnd(IAsyncResult ar) { ErrorTypes eErrorTypes = ErrorTypes.NoError; try { eErrorTypes = m_oRemove.EndInvoke(ar); } catch { eErrorTypes = ErrorTypes.TaskResult; } return eErrorTypes; } private void GetOrCreateCallback(IAsyncResult ar) { try { bool bExist = false; try { m_oGetOrCreate.m_delegateNonQuery.EndInvoke(ar); } catch { bExist = true; } if(bExist) { m_oGetOrCreate.m_bCreate = false; if (null != m_oGetOrCreate.m_oCommand) { m_oGetOrCreate.m_oCommand.Dispose(); m_oGetOrCreate.m_oCommand = null; } IDbCommand oSelCommand = m_oGetOrCreate.m_oSqlCon.CreateCommand(); TaskResultDataToUpdate oToUpdate = new TaskResultDataToUpdate(); oToUpdate.oLastOpenDate = DateTime.UtcNow; oSelCommand.CommandText = GetUPDATEString(m_oGetOrCreate.m_oTast.sKey, oToUpdate) + GetSELECTString(m_oGetOrCreate.m_oTast.sKey); m_oGetOrCreate.m_oCommand = oSelCommand; m_oGetOrCreate.m_delegateReader = new ExecuteReader(oSelCommand.ExecuteReader); m_oGetOrCreate.m_delegateReader.BeginInvoke(m_oGetOrCreate.m_fCallback, m_oGetOrCreate.m_oParam); } else { m_oGetOrCreate.m_bCreate = true; m_oGetOrCreate.m_delegateReader = null; m_oGetOrCreate.FireCallback(); } } catch { m_oGetOrCreate.DisposeAndCallback(); } } private void GetOrCreateCallback2(IAsyncResult ar) { try { bool bExist = false; try { int nNumRowsAffected = m_oGetOrCreate.m_delegateNonQuery.EndInvoke(ar); m_oGetOrCreate.m_bCreate = (nNumRowsAffected == 1); bExist = (nNumRowsAffected > 1); if (null != m_oGetOrCreate.m_oCommand) { m_oGetOrCreate.m_oCommand.Dispose(); m_oGetOrCreate.m_oCommand = null; } } catch (Exception e) { _log.Error("Exception;", e); m_oGetOrCreate.m_bCreate = false; bExist = true; } if (bExist) { IDbCommand oSelCommand = m_oGetOrCreate.m_oSqlCon.CreateCommand(); oSelCommand.CommandText = GetSELECTString(m_oGetOrCreate.m_oTast.sKey); m_oGetOrCreate.m_oCommand = oSelCommand; m_oGetOrCreate.m_delegateReader = new ExecuteReader(oSelCommand.ExecuteReader); m_oGetOrCreate.m_delegateReader.BeginInvoke(m_oGetOrCreate.m_fCallback, m_oGetOrCreate.m_oParam); } else { m_oGetOrCreate.m_delegateReader = null; m_oGetOrCreate.FireCallback(); } } catch { m_oGetOrCreate.DisposeAndCallback(); } } private void AddRandomKeyCallback(IAsyncResult ar) { try { bool bExist = false; try { m_oAddRandomKey.m_delegateNonQuery.EndInvoke(ar); if (null != m_oAddRandomKey.m_oCommand) { m_oAddRandomKey.m_oCommand.Dispose(); m_oAddRandomKey.m_oCommand = null; } } catch { bExist = true; } if(bExist) { string sNewKey = m_oAddRandomKey.m_oTast.sKey + "_" + m_oRandom.Next(mc_nRandomMaxNumber); m_oAddRandomKey.m_oTast.sKey = sNewKey; IDbCommand oSelCommand = m_oAddRandomKey.m_oSqlCon.CreateCommand(); m_oAddRandomKey.m_oCommand = oSelCommand; oSelCommand.CommandText = GetINSERTString(m_oAddRandomKey.m_oTast); m_oAddRandomKey.m_delegateNonQuery = new ExecuteNonQuery(oSelCommand.ExecuteNonQuery); m_oAddRandomKey.m_delegateNonQuery.BeginInvoke(AddRandomKeyCallback, null); } else { m_oAddRandomKey.Close(); m_oAddRandomKey.FireCallback(); } } catch { m_oAddRandomKey.DisposeAndCallback(); } } private IDbConnection GetDbConnection() { var cs = ConfigurationManager.ConnectionStrings[m_sConnectionString]; var dbProvider = System.Data.Common.DbProviderFactories.GetFactory(cs.ProviderName); var connection = dbProvider.CreateConnection(); connection.ConnectionString = cs.ConnectionString; return connection; } private DbManager GetDbManager(string connectionStringName) { var cs = ConfigurationManager.ConnectionStrings[connectionStringName]; if (!DbRegistry.IsDatabaseRegistered(connectionStringName)) { DbRegistry.RegisterDatabase(connectionStringName, cs); } return new DbManager(connectionStringName); } private void TaskResultFromReader(TaskResultData oTaskResultData, IDataReader oReader) { oTaskResultData.sKey = Convert.ToString(oReader["tr_key"]); oTaskResultData.sFormat = Convert.ToString(oReader["tr_format"]); oTaskResultData.eStatus = (FileStatus)Convert.ToByte(oReader["tr_status"]); oTaskResultData.nStatusInfo = Convert.ToInt32(oReader["tr_status_info"]); oTaskResultData.oLastOpenDate = Convert.ToDateTime(oReader["tr_last_open_date"]); oTaskResultData.sTitle = Convert.ToString(oReader["tr_title"]); } private string GetUPSERTString(TaskResultData oTast) { return string.Format("INSERT INTO tast_result ( tr_key, tr_format, tr_status, tr_status_info, tr_last_open_date, tr_title ) VALUES ('{0}', '{1}', '{2}' , '{3}', '{4}', '{5}') ON DUPLICATE KEY UPDATE tr_last_open_date = '{6}';", Utils.MySqlEscape(oTast.sKey, m_sConnectionString), Utils.MySqlEscape(oTast.sFormat, m_sConnectionString), oTast.eStatus.ToString("d"), oTast.nStatusInfo.ToString(), DateTime.UtcNow.ToString(Constants.mc_sDateTimeFormat), Utils.MySqlEscape(oTast.sTitle, m_sConnectionString), DateTime.UtcNow.ToString(Constants.mc_sDateTimeFormat)); } private string GetINSERTString(TaskResultData oTast) { return string.Format("INSERT INTO tast_result ( tr_key, tr_format, tr_status, tr_status_info, tr_last_open_date, tr_title ) VALUES ('{0}', '{1}', '{2}' , '{3}', '{4}', '{5}');", Utils.MySqlEscape(oTast.sKey, m_sConnectionString), Utils.MySqlEscape(oTast.sFormat, m_sConnectionString), oTast.eStatus.ToString("d"), oTast.nStatusInfo.ToString(), DateTime.UtcNow.ToString(Constants.mc_sDateTimeFormat), Utils.MySqlEscape(oTast.sTitle, m_sConnectionString)); } private string GetUPDATEString(string sKey, TaskResultDataToUpdate oTast) { List aValues = new List(); TaskResultDataToUpdateToArray(aValues, oTast, true); return string.Format("UPDATE tast_result SET {0} WHERE tr_key='{1}';", string.Join(",", aValues.ToArray()), Utils.MySqlEscape(sKey, m_sConnectionString)); } private string GetUPDATEIFString(string sKey, TaskResultDataToUpdate oMask, TaskResultDataToUpdate oTast) { List aValues = new List(); TaskResultDataToUpdateToArray(aValues, oTast, true); List aWheres = new List(); aWheres.Add("tr_key='" + Utils.MySqlEscape(sKey, m_sConnectionString) + "'"); TaskResultDataToUpdateToArray(aWheres, oMask, false); return string.Format("UPDATE tast_result SET {0} WHERE {1};", string.Join(",", aValues.ToArray()), string.Join(" AND ", aWheres.ToArray())); } private string GetSELECTString(string sKey) { return string.Format("SELECT * FROM tast_result WHERE tr_key='{0}';", Utils.MySqlEscape(sKey, m_sConnectionString)); } private string GetSELECTString(string[] aKeys) { StringBuilder sb = new StringBuilder(); for (int i = 0, length = aKeys.Length; i < length; ++i) { if(0 == i) sb.AppendFormat("'{0}'", Utils.MySqlEscape(aKeys[i], m_sConnectionString)); else sb.AppendFormat(",'{0}'", Utils.MySqlEscape(aKeys[i], m_sConnectionString)); } return string.Format("SELECT * FROM tast_result WHERE tr_key IN({0});", sb.ToString()); } private string GetSELECTEditingString() { return string.Format("SELECT * FROM tast_result WHERE tr_editing='{0}' OR tr_status='{1}';", FileStatus.Ok, FileStatus.UpdateVersion); } private string GetDELETEString(string sKey) { return string.Format("DELETE FROM tast_result WHERE tr_key='{0}';", Utils.MySqlEscape(sKey, m_sConnectionString)); } private string GetSELECTExpiredTasksString(int nMaxCount) { DateTime oDateTimeExpired = DateTime.UtcNow - TimeSpan.Parse(ConfigurationManager.AppSettings["utils.taskresult.ttl"] ?? "7.00:00:00"); string strDateTimeExpired = oDateTimeExpired.ToString(Constants.mc_sDateTimeFormat); return string.Format("SELECT * FROM tast_result WHERE tr_last_open_date <= '{0}' AND tr_key NOT IN (SELECT dp_key FROM doc_pucker) LIMIT {1};", strDateTimeExpired, nMaxCount); } private void TaskResultDataToUpdateToArray(List aValues, TaskResultDataToUpdate oTask, bool bUpdateTime) { if (false == string.IsNullOrEmpty(oTask.sFormat)) aValues.Add("tr_format='" + Utils.MySqlEscape(oTask.sFormat, m_sConnectionString) + "'"); if (oTask.eStatus.HasValue) aValues.Add("tr_status='" + oTask.eStatus.Value.ToString("d") + "'"); if (oTask.nStatusInfo.HasValue) aValues.Add("tr_status_info='" + oTask.nStatusInfo.Value.ToString() + "'"); if(bUpdateTime) aValues.Add("tr_last_open_date='" + DateTime.UtcNow.ToString(Constants.mc_sDateTimeFormat) + "'"); if (false == string.IsNullOrEmpty(oTask.sTitle)) aValues.Add("tr_title='" + Utils.MySqlEscape(oTask.sTitle, m_sConnectionString) + "'"); } } }