dbbab1c761
尽管里面有点粪,但还是生成了漂亮的树,就当给树施肥了 剩下一个小问题:未考虑节点的宽度(因为名称不一样长),但目前未发现对图生成的明显影响 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>
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;
|
|
}
|
|
}
|