Canon/Canon.Server/Services/GridFsService.cs
Ichirinko dbbab1c761 feat-visualization (#33)
尽管里面有点粪,但还是生成了漂亮的树,就当给树施肥了
剩下一个小问题:未考虑节点的宽度(因为名称不一样长),但目前未发现对图生成的明显影响

Co-authored-by: jackfiled <xcrenchangjun@outlook.com>
Reviewed-on: PostGuard/Canon#33
Co-authored-by: Ichirinko <1621543655@qq.com>
Co-committed-by: Ichirinko <1621543655@qq.com>
2024-04-19 14:59:45 +08:00

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;
}
}