/// <summary>
/// Obtains the lock.
/// </summary>
/// <returns><c>true</c> if the lock has been obtained, <c>false</c> otherwise.</returns>
public override bool Obtain()
{
ICommandBuilder builder = _sqlStorageProviderUtility.GetCommandBuilder2();
DbConnection connection = builder.GetConnection(_connString);
DbTransaction transaction = _sqlStorageProviderUtility.BeginTransaction(connection);
QueryBuilder queryBuilder = new QueryBuilder(builder);
string query = queryBuilder.DeleteFrom("SearchIndexLock");
query = queryBuilder.Where(query, "Wiki", WhereOperator.Equals, "Wiki");
query = queryBuilder.AndWhere(query, "Name", WhereOperator.Equals, "Name");
List <Parameter> parameters = new List <Parameter>(2);
parameters.Add(new Parameter(ParameterType.String, "Wiki", _wiki));
parameters.Add(new Parameter(ParameterType.String, "Name", _name));
DbCommand command = builder.GetCommand(transaction, query, parameters);
int rows = _sqlStorageProviderUtility.ExecuteNonQuery(command, false);
if (rows != 0)
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
return(false); // Deletion command failed (0 is the only accepted value)
}
query = queryBuilder.InsertInto("SearchIndexLock",
new string[] { "wiki", "Name", "Value" }, new string[] { "Wiki", "Name", "Value" });
parameters = new List <Parameter>(3);
parameters.Add(new Parameter(ParameterType.String, "Wiki", _wiki));
parameters.Add(new Parameter(ParameterType.String, "Name", _name));
parameters.Add(new Parameter(ParameterType.String, "Value", "locked"));
command = builder.GetCommand(transaction, query, parameters);
rows = _sqlStorageProviderUtility.ExecuteNonQuery(command, false);
if (rows == 1)
{
_sqlStorageProviderUtility.CommitTransaction(transaction);
}
else
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
}
return(rows == 1);
}
/// <summary>
/// Deletes a file.
/// </summary>
/// <param name="name">The name of the file to be deleted.</param>
public override void DeleteFile(string name)
{
ICommandBuilder builder = _sqlStorageProviderUtility.GetCommandBuilder2();
DbConnection connection = builder.GetConnection(_connString);
DbTransaction transaction = _sqlStorageProviderUtility.BeginTransaction(connection);
if (!FileExists(transaction, name))
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
return;
}
QueryBuilder queryBuilder = new QueryBuilder(builder);
string query = queryBuilder.DeleteFrom("SearchIndex");
query = queryBuilder.Where(query, "Wiki", WhereOperator.Equals, "Wiki");
query = queryBuilder.AndWhere(query, "Name", WhereOperator.Equals, "Name");
List <Parameter> parameters = new List <Parameter>(2);
parameters.Add(new Parameter(ParameterType.String, "Wiki", _wiki));
parameters.Add(new Parameter(ParameterType.String, "Name", name));
DbCommand command = builder.GetCommand(transaction, query, parameters);
int rows = _sqlStorageProviderUtility.ExecuteNonQuery(command, false);
if (rows != 1)
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
}
else
{
_sqlStorageProviderUtility.CommitTransaction(transaction);
if (_cacheDirectory.FileExists(name + ".blob"))
{
_cacheDirectory.DeleteFile(name + ".blob");
}
if (_cacheDirectory.FileExists(name))
{
_cacheDirectory.DeleteFile(name);
}
}
}
/// <summary>
/// Closes the sql index output.
/// </summary>
public override void Close()
{
string fileName = _fileName;
// make sure it's all written out
_indexOutput.Flush();
long originalLength = _indexOutput.Length();
_indexOutput.Close();
Stream fileStream = new StreamInput(CacheDirectory.OpenInput(fileName));
try {
// push the file stream up to the db.
ICommandBuilder builder = _sqlStorageProviderUtility.GetCommandBuilder2();
DbConnection connection = builder.GetConnection(_connString);
DbTransaction transaction = _sqlStorageProviderUtility.BeginTransaction(connection);
QueryBuilder queryBuilder = new QueryBuilder(builder);
//bool fileExists = FileExists(transaction, _wiki, _fileName);
// To achieve decent performance, an UPDATE query is issued if the file exists,
// otherwise an INSERT query is issued
string query;
List <Parameter> parameters;
byte[] fileData = null;
int size = Tools.ReadStream(fileStream, ref fileData, MaxFileSize);
if (size < 0)
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
throw new ArgumentException("Source Stream contains too much data", "sourceStream");
}
//if(fileExists) {
// query = queryBuilder.Update("SearchIndex", new string[] { "Size", "LastModified", "Data" }, new string[] { "Size", "LastModified", "Data" });
// query = queryBuilder.Where(query, "Wiki", WhereOperator.Equals, "Wiki");
// query = queryBuilder.AndWhere(query, "Name", WhereOperator.Equals, "Name");
// parameters = new List<Parameter>(5);
// parameters.Add(new Parameter(ParameterType.String, "Wiki", _wiki));
// parameters.Add(new Parameter(ParameterType.Int64, "Size", (long)originalLength));
// parameters.Add(new Parameter(ParameterType.DateTime, "LastModified", DateTime.Now.ToUniversalTime()));
// parameters.Add(new Parameter(ParameterType.ByteArray, "Data", fileData));
// parameters.Add(new Parameter(ParameterType.String, "Name", _fileName));
//}
//else {
query = queryBuilder.InsertInto("SearchIndex", new string[] { "Wiki", "Name", "Size", "LastModified", "Data" },
new string[] { "Wiki", "Name", "Size", "LastModified", "Data" });
parameters = new List <Parameter>(5);
parameters.Add(new Parameter(ParameterType.String, "Wiki", _wiki));
parameters.Add(new Parameter(ParameterType.String, "Name", _fileName));
parameters.Add(new Parameter(ParameterType.Int64, "Size", (long)originalLength));
parameters.Add(new Parameter(ParameterType.DateTime, "LastModified", DateTime.Now.ToUniversalTime()));
parameters.Add(new Parameter(ParameterType.ByteArray, "Data", fileData));
//}
DbCommand command = builder.GetCommand(transaction, query, parameters);
int rows = _sqlStorageProviderUtility.ExecuteNonQuery(command, false);
if (rows == 1)
{
_sqlStorageProviderUtility.CommitTransaction(transaction);
}
else
{
_sqlStorageProviderUtility.RollbackTransaction(transaction);
}
}
finally {
fileStream.Dispose();
}
// clean up
_indexOutput = null;
GC.SuppressFinalize(this);
}