public async Task <SMessageHistory> QueryById(Guid id)
{
var storeInfo = await StoreInfoHelper.GetHashStoreInfo(_storeInfoResolveService, _hashGroupRepository, _messageHistoryHashGroupName, id.ToString());
if (!storeInfo.TableNames.TryGetValue(HashEntityNames.SMessageHistory, out string tableName))
{
var fragment = new TextFragment()
{
Code = TextCodes.NotFoundKeyInHashNodeKeyInfo,
DefaultFormatting = "哈希组{0}中的哈希节点关键信息中找不到键值{1}",
ReplaceParameters = new List <object>()
{
_messageHistoryHashGroupName, HashEntityNames.SMessageHistory
}
};
throw new UtilityException((int)Errors.NotFoundKeyInHashNodeKeyInfo, fragment);
}
SMessageHistory smssageHistory = null;
await DBTransactionHelper.SqlTransactionWorkAsync(DBTypes.SqlServer, false, false, _messageQueueConnectionFactory.CreateReadForSMessageHistory(storeInfo.DBConnectionNames), async (conn, transaction) =>
{
SqlTransaction sqlTran = null;
if (transaction != null)
{
sqlTran = (SqlTransaction)transaction;
}
await using (SqlCommand command = new SqlCommand()
{
Connection = (SqlConnection)conn,
CommandType = CommandType.Text,
CommandText = string.Format(@"SELECT {0} FROM {1} WHERE id=@id", StoreHelper.GetSMessageHistorySelectFields(string.Empty), tableName),
Transaction = sqlTran,
})
{
var parameter = new SqlParameter("@id", SqlDbType.UniqueIdentifier)
{
Value = id
};
command.Parameters.Add(parameter);
command.Prepare();
SqlDataReader reader = null;
await using (reader = await command.ExecuteReaderAsync())
{
if (await reader.ReadAsync())
{
smssageHistory = new SMessageHistory();
StoreHelper.SetSMessageHistorySelectFields(smssageHistory, reader, string.Empty);
}
await reader.CloseAsync();
}
}
});
return(smssageHistory);
}
/// <summary>
/// 为消息历史记录从DbDataReader中赋值
/// </summary>
/// <param name="queue"></param>
/// <param name="reader"></param>
/// <param name="prefix"></param>
public static void SetSMessageHistorySelectFields(SMessageHistory history, DbDataReader reader, string prefix)
{
history.ID = (Guid)reader[string.Format("{0}id", prefix)];
if (reader[string.Format("{0}key", prefix)] != DBNull.Value)
{
history.Key = reader[string.Format("{0}key", prefix)].ToString();
}
if (reader[string.Format("{0}type", prefix)] != DBNull.Value)
{
history.Type = reader[string.Format("{0}type", prefix)].ToString();
}
if (reader[string.Format("{0}data", prefix)] != DBNull.Value)
{
history.Data = reader[string.Format("{0}data", prefix)].ToString();
}
if (reader[string.Format("{0}originalmessageid", prefix)] != DBNull.Value)
{
history.OriginalMessageID = (Guid)reader[string.Format("{0}originalmessageid", prefix)];
}
if (reader[string.Format("{0}delaymessageid", prefix)] != DBNull.Value)
{
history.DelayMessageID = (Guid)reader[string.Format("{0}delaymessageid", prefix)];
}
if (reader[string.Format("{0}status", prefix)] != DBNull.Value)
{
history.Status = Convert.ToInt16(reader[string.Format("{0}status", prefix)].ToString());
}
if (reader[string.Format("{0}createtime", prefix)] != DBNull.Value)
{
history.CreateTime = (DateTime)reader[string.Format("{0}createtime", prefix)];
}
if (reader[string.Format("{0}modifytime", prefix)] != DBNull.Value)
{
history.ModifyTime = (DateTime)reader[string.Format("{0}modifytime", prefix)];
}
}
public async Task Add(SMessageHistory history)
{
var storeInfo = await StoreInfoHelper.GetHashStoreInfo(_storeInfoResolveService, _hashGroupRepository, _messageHistoryHashGroupName, history.ID.ToString());
if (!storeInfo.TableNames.TryGetValue(HashEntityNames.SMessageHistory, out string tableName))
{
var fragment = new TextFragment()
{
Code = TextCodes.NotFoundKeyInHashNodeKeyInfo,
DefaultFormatting = "哈希组{0}中的哈希节点关键信息中找不到键值{1}",
ReplaceParameters = new List <object>()
{
_messageHistoryHashGroupName, HashEntityNames.SMessageHistory
}
};
throw new UtilityException((int)Errors.NotFoundKeyInHashNodeKeyInfo, fragment);
}
await DBTransactionHelper.SqlTransactionWorkAsync(DBTypes.SqlServer, false, false, _messageQueueConnectionFactory.CreateAllForSMessageHistory(storeInfo.DBConnectionNames), async (conn, transaction) =>
{
//新增
SqlTransaction sqlTran = null;
if (transaction != null)
{
sqlTran = (SqlTransaction)transaction;
}
await using (SqlCommand command = new SqlCommand()
{
Connection = (SqlConnection)conn,
CommandType = CommandType.Text,
Transaction = sqlTran,
})
{
SqlParameter parameter;
if (history.ID == Guid.Empty)
{
command.CommandText = string.Format(@"
INSERT INTO {0}
(
[id]
,[key]
,[type]
,[data]
,[originalmessageid]
,[delaymessageid]
,[status]
,[createtime]
,[modifytime]
)
VALUES
(default
, @key
, @type
, @data
, @originalmessageid
, @delaymessageid
, @status
, GETUTCDATE()
, GETUTCDATE());
select @newid =[id] from {0} where [sequence] = SCOPE_IDENTITY()", tableName);
parameter = new SqlParameter("@newid", SqlDbType.UniqueIdentifier)
{
Direction = ParameterDirection.Output
};
command.Parameters.Add(parameter);
}
else
{
command.CommandText = string.Format(@"
INSERT INTO {0}
(
[id]
,[key]
,[type]
,[data]
,[originalmessageid]
,[delaymessageid]
,[status]
,[createtime]
,[modifytime]
)
VALUES
(
@id
, @key
, @type
, @data
, @originalmessageid
, @delaymessageid
, @status
, GETUTCDATE()
, GETUTCDATE());", tableName);
parameter = new SqlParameter("@id", SqlDbType.UniqueIdentifier)
{
Value = history.ID
};
command.Parameters.Add(parameter);
}
parameter = new SqlParameter("@key", SqlDbType.VarChar, 150)
{
Value = history.Key
};
command.Parameters.Add(parameter);
parameter = new SqlParameter("@type", SqlDbType.VarChar, 150)
{
Value = history.Type
};
command.Parameters.Add(parameter);
parameter = new SqlParameter("@data", SqlDbType.VarChar, history.Data.Length)
{
Value = history.Data
};
command.Parameters.Add(parameter);
if (history.OriginalMessageID.HasValue)
{
parameter = new SqlParameter("@originalmessageid", SqlDbType.UniqueIdentifier)
{
Value = history.OriginalMessageID
};
}
else
{
parameter = new SqlParameter("@originalmessageid", SqlDbType.UniqueIdentifier)
{
Value = DBNull.Value
};
}
command.Parameters.Add(parameter);
if (history.DelayMessageID.HasValue)
{
parameter = new SqlParameter("@delaymessageid", SqlDbType.UniqueIdentifier)
{
Value = history.DelayMessageID
};
}
else
{
parameter = new SqlParameter("@delaymessageid", SqlDbType.UniqueIdentifier)
{
Value = DBNull.Value
};
}
command.Parameters.Add(parameter);
parameter = new SqlParameter("@status", SqlDbType.Int)
{
Value = history.Status
};
command.Parameters.Add(parameter);
await command.PrepareAsync();
await command.ExecuteNonQueryAsync();
//如果用户未赋值ID则创建成功后返回ID
if (history.ID == Guid.Empty)
{
history.ID = (Guid)command.Parameters["@newid"].Value;
}
;
}
});
}
