42 lines
1.2 KiB
C#
42 lines
1.2 KiB
C#
|
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;
|
|||
|
}
|
|||
|
}
|