package com.imyeyu.api.config;

import com.mongodb.client.MongoClient;
import com.mongodb.client.gridfs.GridFSBucket;
import com.mongodb.client.gridfs.GridFSBuckets;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @author 夜雨
 * @since 2024-02-23 10:55
 */
@Configuration
public class MongoConfig {

	@Value("${spring.data.mongodb.database}")
	private String db;

	@Bean
	public GridFSBucket gridFSBucket(MongoClient mongoClient) {
		return GridFSBuckets.create(mongoClient.getDatabase(db));
	}
}