java - 使用泛型为 MongoDB POJO 创建包装器

Question

更新 2:我想我已经取得了一些进步。我的 IDE 对我的代码更满意，但我只有最后一行不高兴:

CodecRegistry pojoCodecRegistry = fromRegistries(MongoClient.getDefaultCodecRegistry(),
                fromProviders(PojoCodecProvider.builder().automatic(true).build()));
        MongoClientURI connectionString = new MongoClientURI("my-mongo-string");
        MongoClient mongoClient = new MongoClient(connectionString);
        MongoDatabase database = mongoClient.getDatabase("New");
        database = database.withCodecRegistry(pojoCodecRegistry);
        MongoCollection<? extends Person> collection = database.getCollection("people", this.getClass());
        collection = collection.withCodecRegistry(pojoCodecRegistry);
        collection.insertOne(this);

我已将我的“Collection”的名称换成“Person”，以暂时防止与 Java 的“Collection”发生冲突。除了最后一行，一切似乎都很好，这给了我这个错误:insertOne (capture<? extends Utils.Person>) in MongoCollection cannot be applied to (Utils.Person) . 'this' 指的是我的 person 类。我在这里做错了什么？

更新:下面的原始问题。我相信我现在已经确定泛型是我最好的选择。但是我在“生成”这两行时遇到了一些问题(假设我已经正确初始化了数据库):

MongoCollection<MyClass> collection = database.getCollection("SomeString", MyClass.class);

最好的方法是什么？ Java 似乎不喜欢在泛型上使用 .class。

原始问题:

所以我想利用 MongoDB 驱动程序存储 PJOS 的能力。 Example Here .但我打算将多个类存储到数据库中，我不想每次都重新键入所有连接代码，也不想复制粘贴然后对每个类的代码进行小幅修改。我想创建一个“Collection”类，然后对其进行扩展，但它并不像我预期的那样有效。这是我的代码:

public class Collection {
    private MongoCollection<Collection> collection;

    public Collection(){}

    public Collection(String databaseName){
        databaseName = databaseName.toLowerCase().replaceAll(" ", "");
        CodecRegistry pojoCodecRegistry = fromRegistries(com.mongodb.MongoClient.getDefaultCodecRegistry(),
                fromProviders(PojoCodecProvider.builder().automatic(true).build()));
        MongoClientURI connectionString = new MongoClientURI("my-mongo-connection-string");
        com.mongodb.MongoClient mongoClient = new com.mongodb.MongoClient(connectionString);
        MongoDatabase database = mongoClient.getDatabase(databaseName);
        database = database.withCodecRegistry(pojoCodecRegistry);
        collection = database.getCollection(this.getClass().getName(), Collection.class);
    }

    public Collection findOne(Document document){
        return collection.find(document).first();
    }

    public void save(){
        collection.insertOne(this);
    }

}

如您所见，我希望能够将数据库名称传递给继承自此类的任何类(或者任何正确的做法。)，并让我的代码连接到数据库并查找或正确保存相应子类的 POJO。因此，例如，如果我有一个带有电话号码的 Person 类，我可以扩展我的 Collections 类并依靠一切正常工作。现在，当我在子类上调用保存函数时，它会尝试保存父集合，但出现错误 Can't find a codec for class Utils.MongoDb.Collection. (我的 Collection 类在我自己的 Utils.MongoDb 命名空间中)。我只是错过了什么吗？有没有更好的方法来解决这个问题？我能想到的唯一方法是将我的 Collection 构造函数中的代码复制并粘贴到每个类中，并使用正确的类变量对其进行修改。我希望所有这些都清楚。提前致谢!

Answer 1

啊哈!我懂了!我必须做的是泛化并编写一个辅助方法，这样我就可以将“this”作为对象传入。我还必须对通用 T 进行强制转换。我知道这是一个未经检查的强制转换，但我不太确定如何在此处对其进行检查。无论哪种方式，它都有效!

 public <T> void saveObj(T obj){
        String dbName = "New";
        MongoDatabase database = getMongoDatabase(dbName);
        MongoCollection<T> collection = (MongoCollection<T>) database.getCollection(obj.getClass().getSimpleName(), obj.getClass());
        collection.insertOne(obj);
    }

    public MongoDatabase getMongoDatabase(String dbName) {
        CodecRegistry pojoCodecRegistry = fromRegistries(MongoClient.getDefaultCodecRegistry(),
                fromProviders(PojoCodecProvider.builder().automatic(true).build()));
        MongoClientURI connectionString = new MongoClientURI("my-mongo-string");
        MongoClient mongoClient = new MongoClient(connectionString);
        MongoDatabase database = mongoClient.getDatabase(dbName);
        database = database.withCodecRegistry(pojoCodecRegistry);
        return database;
    }