static BlobStorageBase()
{
Providers = TypeResolver.GetTypesByInterface(typeof(IBlobProvider))
.Select(t =>
{
try
{
var instance = (IBlobProvider)Activator.CreateInstance(t);
SnTrace.System.Write("BlobProvider found: {0}", t.FullName);
return(instance);
}
catch (MissingMethodException)
{
// no default constructor: provider must be instantiated manually
SnTrace.System.Write("BlobProvider found, but must be configured manually: {0}", t.FullName);
}
catch (Exception ex)
{
SnLog.WriteException(ex, $"Error during instantiating {t.FullName}.");
}
return(null);
})
.Where(instance => instance != null)
.ToDictionary(x => x.GetType().FullName, x => x);
BuiltInProvider = new BuiltInBlobProvider();
}
static BlobStorageBase()
{
Providers = TypeResolver.GetTypesByInterface(typeof(IBlobProvider))
.Select(t =>
{
SnTrace.System.Write("BlobProvider found: {0}", t.FullName);
return((IBlobProvider)Activator.CreateInstance(t));
})
.ToDictionary(x => x.GetType().FullName, x => x);
BuiltInProvider = new BuiltInBlobProvider();
}
/// <summary>
/// Loads a segment from the binary data beginning at the specified position.
/// </summary>
/// <param name="fileId">File record identifier.</param>
/// <param name="position">Starting position of the segment.</param>
/// <param name="count">Number of bytes to load.</param>
/// <returns>Byte array containing the requested number of bytes (or less if there is not enough in the binary data).</returns>
protected internal static byte[] LoadBinaryFragment(int fileId, long position, int count)
{
var ctx = GetBlobStorageContext(fileId);
var provider = ctx.Provider;
if (provider == BuiltInProvider)
{
return(BuiltInBlobProvider.ReadRandom(ctx, position, count));
}
var realCount = Convert.ToInt32(Math.Min(ctx.Length - position, count));
var bytes = new byte[realCount];
using (var stream = provider.GetStreamForRead(ctx))
{
stream.Seek(position, SeekOrigin.Begin);
stream.Read(bytes, 0, realCount);
}
return(bytes);
}
/// <summary>
/// Loads a segment from the binary data beginning at the specified position.
/// </summary>
/// <param name="fileId">File record identifier.</param>
/// <param name="position">Starting position of the segment.</param>
/// <param name="count">Number of bytes to load.</param>
/// <param name="cancellationToken">The token to monitor for cancellation requests.</param>
/// <returns>A Task that represents the asynchronous operation containing
/// a byte array containing the requested number of bytes (or less if there is not enough in the binary data).</returns>
protected internal static async Task <byte[]> LoadBinaryFragmentAsync(int fileId, long position, int count,
CancellationToken cancellationToken)
{
var ctx = await GetBlobStorageContextAsync(fileId, cancellationToken).ConfigureAwait(false);
var provider = ctx.Provider;
if (provider == BuiltInProvider)
{
return(BuiltInBlobProvider.ReadRandom(ctx, position, count));
}
var realCount = Convert.ToInt32(Math.Min(ctx.Length - position, count));
var bytes = new byte[realCount];
using (var stream = provider.GetStreamForRead(ctx))
{
stream.Seek(position, SeekOrigin.Begin);
stream.Read(bytes, 0, realCount);
}
return(bytes);
}
/// <summary>
/// Updates an existing binary property value in the database and the blob storage.
/// </summary>
/// <param name="blobProvider">Blob storage provider.</param>
/// <param name="value">Binary data to update.</param>
public void UpdateBinaryProperty(IBlobProvider blobProvider, BinaryDataValue value)
{
var streamLength = value.Stream?.Length ?? 0;
var isExternal = false;
if (!IsBuiltInOrSqlFileStreamProvider(blobProvider))
{
var ctx = new BlobStorageContext(blobProvider, value.BlobProviderData)
{
VersionId = 0,
PropertyTypeId = 0,
FileId = value.FileId,
Length = streamLength,
};
blobProvider.Allocate(ctx);
isExternal = true;
value.BlobProviderName = ctx.Provider.GetType().FullName;
value.BlobProviderData = BlobStorageContext.SerializeBlobProviderData(ctx.BlobProviderData);
}
else
{
value.BlobProviderName = null;
value.BlobProviderData = null;
}
var isRepositoryStream = value.Stream is RepositoryStream || value.Stream is SenseNetSqlFileStream;
var hasStream = isRepositoryStream || value.Stream is MemoryStream;
if (!isExternal && !hasStream)
{
// do not do any database operation if the stream is not modified
return;
}
SqlFileStreamData fileStreamData = null;
SqlProcedure cmd = null;
try
{
string sql;
CommandType commandType;
if (IsBuiltInOrSqlFileStreamProvider(blobProvider))
{
commandType = CommandType.StoredProcedure;
sql = "proc_BinaryProperty_Update";
}
else
{
commandType = CommandType.Text;
sql = UpdateBinarypropertyNewFilerowFilestreamScript;
}
cmd = new SqlProcedure {
CommandText = sql, CommandType = commandType
};
cmd.Parameters.Add("@BinaryPropertyId", SqlDbType.Int).Value = value.Id;
cmd.Parameters.Add("@ContentType", SqlDbType.NVarChar, 450).Value = value.ContentType;
cmd.Parameters.Add("@FileNameWithoutExtension", SqlDbType.NVarChar, 450).Value = value.FileName.FileNameWithoutExtension == null ? DBNull.Value : (object)value.FileName.FileNameWithoutExtension;
cmd.Parameters.Add("@Extension", SqlDbType.NVarChar, 50).Value = ValidateExtension(value.FileName.Extension);
cmd.Parameters.Add("@Size", SqlDbType.BigInt).Value = value.Size;
cmd.Parameters.Add("@Checksum", SqlDbType.VarChar, 200).Value = value.Checksum != null ? (object)value.Checksum : DBNull.Value;
cmd.Parameters.Add("@BlobProvider", SqlDbType.NVarChar, 450).Value = value.BlobProviderName != null ? (object)value.BlobProviderName : DBNull.Value;
cmd.Parameters.Add("@BlobProviderData", SqlDbType.NVarChar, int.MaxValue).Value = value.BlobProviderData != null ? (object)value.BlobProviderData : DBNull.Value;
int fileId;
string path;
byte[] transactionContext;
// Update row and retrieve file path and
// transaction context for the Filestream column
using (var reader = cmd.ExecuteReader())
{
reader.Read();
fileId = reader.GetInt32(0);
path = reader.GetSafeString(1);
transactionContext = reader.IsDBNull(2) ? null : reader.GetSqlBytes(2).Buffer;
}
if (!string.IsNullOrEmpty(path))
{
fileStreamData = new SqlFileStreamData {
Path = path, TransactionContext = transactionContext
}
}
;
if (fileId > 0 && fileId != value.FileId)
{
value.FileId = fileId;
}
}
finally
{
cmd?.Dispose();
}
if (blobProvider == BlobStorageBase.BuiltInProvider)
{
var ctx = new BlobStorageContext(blobProvider, value.BlobProviderData)
{
VersionId = 0,
PropertyTypeId = 0,
FileId = value.FileId,
Length = streamLength,
BlobProviderData = new SqlFileStreamBlobProviderData {
FileStreamData = fileStreamData
}
};
BuiltInBlobProvider.UpdateStream(ctx, value.Stream);
}
else if (blobProvider is SqlFileStreamBlobProvider)
{
var ctx = new BlobStorageContext(blobProvider, value.BlobProviderData)
{
VersionId = 0,
PropertyTypeId = 0,
FileId = value.FileId,
Length = streamLength,
BlobProviderData = new SqlFileStreamBlobProviderData {
FileStreamData = fileStreamData
}
};
SqlFileStreamBlobProvider.UpdateStream(ctx, value.Stream);
}
else
{
var ctx = new BlobStorageContext(blobProvider, value.BlobProviderData)
{
VersionId = 0,
PropertyTypeId = 0,
FileId = value.FileId,
Length = streamLength,
};
using (var stream = blobProvider.GetStreamForWrite(ctx))
value.Stream?.CopyTo(stream);
}
}
/// <summary>
/// Inserts a new binary property value into the metadata database and the blob storage,
/// removing the previous one if the content is not new.
/// </summary>
/// <param name="blobProvider">Blob storage provider.</param>
/// <param name="value">Binary data to insert.</param>
/// <param name="versionId">Content version id.</param>
/// <param name="propertyTypeId">Binary property type id.</param>
/// <param name="isNewNode">Whether this value belongs to a new or an existing node.</param>
public void InsertBinaryProperty(IBlobProvider blobProvider, BinaryDataValue value, int versionId, int propertyTypeId, bool isNewNode)
{
var streamLength = value.Stream?.Length ?? 0;
var useFileStream = SqlFileStreamBlobProvider.UseFileStream(blobProvider, streamLength);
var ctx = new BlobStorageContext(blobProvider)
{
VersionId = versionId, PropertyTypeId = propertyTypeId, FileId = 0, Length = streamLength
};
// In case of an external provider allocate the place for bytes and
// write the stream beforehand and get the generated provider data.
// Note that the external provider does not need an existing record
// in the Files table to work, it just stores the bytes.
if (!IsBuiltInOrSqlFileStreamProvider(blobProvider))
{
blobProvider.Allocate(ctx);
using (var stream = blobProvider.GetStreamForWrite(ctx))
value.Stream?.CopyTo(stream);
value.BlobProviderName = ctx.Provider.GetType().FullName;
value.BlobProviderData = BlobStorageContext.SerializeBlobProviderData(ctx.BlobProviderData);
}
SqlProcedure cmd = null;
SqlFileStreamData fileStreamData = null;
try
{
cmd = useFileStream
? new SqlProcedure {
CommandText = isNewNode ? InsertBinaryPropertyFilestreamScript : DeleteAndInsertBinaryPropertyFilestream, CommandType = CommandType.Text
}
: new SqlProcedure {
CommandText = isNewNode ? InsertBinaryPropertyScript : DeleteAndInsertBinaryProperty, CommandType = CommandType.Text
};
cmd.Parameters.Add("@VersionId", SqlDbType.Int).Value = versionId != 0 ? (object)versionId : DBNull.Value;
cmd.Parameters.Add("@PropertyTypeId", SqlDbType.Int).Value = propertyTypeId != 0 ? (object)propertyTypeId : DBNull.Value;
cmd.Parameters.Add("@ContentType", SqlDbType.NVarChar, 450).Value = value.ContentType;
cmd.Parameters.Add("@FileNameWithoutExtension", SqlDbType.NVarChar, 450).Value = value.FileName.FileNameWithoutExtension == null ? DBNull.Value : (object)value.FileName.FileNameWithoutExtension;
cmd.Parameters.Add("@Extension", SqlDbType.NVarChar, 50).Value = ValidateExtension(value.FileName.Extension);
cmd.Parameters.Add("@Size", SqlDbType.BigInt).Value = Math.Max(0, value.Size);
cmd.Parameters.Add("@BlobProvider", SqlDbType.NVarChar, 450).Value = value.BlobProviderName != null ? (object)value.BlobProviderName : DBNull.Value;
cmd.Parameters.Add("@BlobProviderData", SqlDbType.NVarChar, int.MaxValue).Value = value.BlobProviderData != null ? (object)value.BlobProviderData : DBNull.Value;
cmd.Parameters.Add("@Checksum", SqlDbType.VarChar, 200).Value = value.Checksum != null ? (object)value.Checksum : DBNull.Value;
// insert binary and file rows and retrieve file path and transaction context for the Filestream column
using (var reader = cmd.ExecuteReader())
{
reader.Read();
value.Id = Convert.ToInt32(reader[0]);
value.FileId = Convert.ToInt32(reader[1]);
value.Timestamp = Utility.Convert.BytesToLong((byte[])reader.GetValue(2));
if (useFileStream)
{
fileStreamData = new SqlFileStreamData
{
Path = reader.GetString(3),
TransactionContext = reader.GetSqlBytes(4).Buffer
};
}
}
}
finally
{
cmd.Dispose();
}
// The BuiltIn blob provider saves the stream after the record
// was saved into the Files table, because simple varbinary
// and sql filestream columns must exist before we can write a
// stream into the record.
// ReSharper disable once InvertIf
if (blobProvider == BlobStorageBase.BuiltInProvider && value.Stream != null)
{
ctx.FileId = value.FileId;
ctx.BlobProviderData = new SqlFileStreamBlobProviderData {
FileStreamData = fileStreamData
};
BuiltInBlobProvider.AddStream(ctx, value.Stream);
}
else if (blobProvider is SqlFileStreamBlobProvider && value.Stream != null)
{
ctx.FileId = value.FileId;
ctx.BlobProviderData = new SqlFileStreamBlobProviderData {
FileStreamData = fileStreamData
};
SqlFileStreamBlobProvider.AddStream(ctx, value.Stream);
}
}
