Canon/Canon.Server/Services/GridFsService.cs

42 lines
1.2 KiB
C#
Raw Permalink Normal View History

using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.GridFS;
namespace Canon.Server.Services;
public class GridFsService
{
private readonly IGridFSBucket _bucket;
public GridFsService(IMongoClient client, string databaseName)
{
IMongoDatabase database = client.GetDatabase(databaseName);
_bucket = new GridFSBucket(database);
}
public async Task<GridFSFileInfo?> FindAsync(string filename)
{
FilterDefinition<GridFSFileInfo> filters = Builders<GridFSFileInfo>.Filter.Eq(
file => file.Filename, filename);
using IAsyncCursor<GridFSFileInfo> cursor = await _bucket.FindAsync(filters);
return await cursor.FirstOrDefaultAsync();
}
public async Task<Stream> OpenReadStream(ObjectId id)
{
return await _bucket.OpenDownloadStreamAsync(id);
}
public async Task<string> UploadStream(Stream sourceStream, string contentType)
{
string filename = Guid.NewGuid().ToString();
await _bucket.UploadFromStreamAsync(filename, sourceStream,
new GridFSUploadOptions { Metadata = new BsonDocument { { "content-type", contentType } } });
return filename;
}
}