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

时间:2018-10-26 05:10:20

标签: java mongodb generics inheritance design-patterns

更新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);

为了避免与Java的“ Collection”发生冲突,我将“ Collection”的名称交换为“ Person”。除了最后一行,一切似乎都不错,这给了我这个错误:insertOne (capture<? extends Utils.Person>) in MongoCollection cannot be applied to (Utils.Person)。 “ this”是指我的人类。我在这里做什么错了?

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

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类并依靠所有可以工作的东西。现在,当我在子类上调用save函数时,它将尝试保存父Collection,并且出现错误Can't find a codec for class Utils.MongoDb.Collection.(我的Collection类位于我自己的Utils.MongoDb命名空间中)。我只是想念什么吗?有没有更好的方法来解决这个问题?我能想到的唯一方法是将Collection构造函数中的代码复制并粘贴到每个类中,然后使用正确的类变量对其进行修改。我希望所有这些都清楚。预先感谢!

1 个答案:

答案 0 :(得分: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;
    }