Micronaut中调用MongoDB函数的正确方法咨询

不需要依赖第三方库，借助Micronaut自带的MongoDB客户端或Micronaut Data MongoDB扩展就能实现，以下是几种推荐方案：

方法一：用runCommand执行函数（兼容旧版MongoDB） #

如果你的MongoDB版本在4.4及以下，可通过注入的MongoDatabase实例调用runCommand执行自定义函数：

import jakarta.inject.Singleton;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

@Singleton
public class MongoFunctionService {
    private final MongoDatabase database;

    public MongoFunctionService(MongoDatabase database) {
        this.database = database;
    }

    public Document callCustomFunction(String functionName, Object... args) {
        // 构造调用命令，示例：调用myFunction(1, "test")
        Document command = new Document("eval", String.format("%s(%s)", 
            functionName, 
            String.join(", ", args)));
        return database.runCommand(command);
    }
}

注意：eval命令已被MongoDB标记为弃用，仅建议在旧版本环境中使用。

方法二：聚合管道+$function（官方推荐） #

MongoDB 4.4+推荐用$function操作符替代eval，Micronaut的Mongo客户端完全支持这种方式：

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.Aggregates;
import org.bson.Document;
import jakarta.inject.Singleton;
import java.util.Arrays;

@Singleton
public class MongoAggregateFunctionService {
    private final MongoCollection<Document> collection;

    public MongoAggregateFunctionService(MongoCollection<Document> collection) {
        this.collection = collection;
    }

    public void callFunctionWithAggregate() {
        // 定义自定义函数逻辑，示例为计算两数之和
        String functionCode = "function(a, b) { return a + b; }";
        
        // 构建聚合管道执行函数
        collection.aggregate(Arrays.asList(
            Aggregates.project(
                new Document("result", 
                    new Document("$function", 
                        new Document("body", functionCode)
                            .append("args", Arrays.asList("$num1", "$num2"))
                            .append("lang", "js")
                    )
                )
            )
        )).forEach(doc -> System.out.println(doc.get("result")));
    }
}

若调用MongoDB服务器上已定义的存储函数，可直接在$function中引用函数名：

// 调用服务器端已定义的myStoredFunction
collection.aggregate(Arrays.asList(
    Aggregates.match(
        new Document("$expr", 
            new Document("$eq", Arrays.asList("$function", Arrays.asList("myStoredFunction", "$field1", "$field2")))
        )
    )
));

方法三：Micronaut Data MongoDB扩展封装（可选） #

如果你使用Micronaut Data MongoDB，可通过自定义Repository方法封装函数调用：

import io.micronaut.data.mongodb.annotation.MongoRepository;
import io.micronaut.data.repository.CrudRepository;
import org.bson.Document;
import reactor.core.publisher.Flux;

@MongoRepository
public interface CustomFunctionRepository extends CrudRepository<Document, String> {
    @Aggregation(pipeline = {
        "{ '$project': { 'result': { '$function': { 'body': 'function(a) { return a * 2; }', 'args': ['$value'], 'lang': 'js' } } } }"
    })
    Flux<Document> calculateDoubleValue();
}

总结 #

• 优先选择聚合管道+$function的方式，兼容高版本MongoDB且符合官方规范
• 无需额外引入第三方库，Micronaut原生组件已覆盖需求

内容的提问来源于stack exchange，提问作者Tech Noob