com.avanza.ymer.TestSpaceObjectFakeConverter.java Source code

Introduction

Here is the source code for com.avanza.ymer.TestSpaceObjectFakeConverter.java

Source

/*
 * Copyright 2015 Avanza Bank AB
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.avanza.ymer;

import org.springframework.data.mongodb.core.query.Query;

import com.avanza.ymer.DocumentConverter;
import com.mongodb.BasicDBObject;

public class TestSpaceObjectFakeConverter {

    static DocumentConverter create() {
        return DocumentConverter.create(new DocumentConverter.Provider() {
            @Override
            public BasicDBObject convertToDBObject(Object type) {
                if (type instanceof TestSpaceObject) {
                    TestSpaceObject testSpaceObject = (TestSpaceObject) type;
                    BasicDBObject dbObject = new BasicDBObject();
                    dbObject.put("_id", testSpaceObject.getId());
                    if (testSpaceObject.getMessage() != null) {
                        dbObject.put("message", testSpaceObject.getMessage());
                    }
                    return dbObject;
                } else if (type instanceof TestSpaceOtherObject) {
                    TestSpaceOtherObject testSpaceOtherObject = (TestSpaceOtherObject) type;
                    BasicDBObject dbObject = new BasicDBObject();
                    dbObject.put("_id", testSpaceOtherObject.getId());
                    if (testSpaceOtherObject.getMessage() != null) {
                        dbObject.put("message", testSpaceOtherObject.getMessage());
                    }
                    return dbObject;
                } else if (type instanceof TestReloadableSpaceObject) {
                    TestReloadableSpaceObject testSpaceObject = (TestReloadableSpaceObject) type;
                    BasicDBObject dbObject = new BasicDBObject();
                    dbObject.put("_id", testSpaceObject.getId());
                    dbObject.put("patched", testSpaceObject.isPatched());
                    dbObject.put("versionID", testSpaceObject.getVersionID());
                    if (testSpaceObject.getLatestRestoreVersion() != null) {
                        dbObject.put("latestRestoreVersion", testSpaceObject.getLatestRestoreVersion());
                    }
                    return dbObject;
                } else {
                    throw new RuntimeException("Unknown object type: " + type.getClass());
                }
            }

            @Override
            public <T> T convert(Class<T> toType, BasicDBObject document) {
                if (toType.equals(TestSpaceObject.class)) {
                    TestSpaceObject testSpaceObject = new TestSpaceObject();
                    testSpaceObject.setId(document.getString("_id"));
                    testSpaceObject.setMessage(document.getString("message"));
                    return toType.cast(testSpaceObject);
                } else if (toType.equals(TestReloadableSpaceObject.class)) {
                    TestReloadableSpaceObject testSpaceObject = new TestReloadableSpaceObject();
                    testSpaceObject.setId(document.getInt("_id"));
                    if (document.containsField("patched")) {
                        testSpaceObject.setPatched(document.getBoolean("patched"));
                    }
                    testSpaceObject.setVersionID(document.getInt("versionID"));
                    if (document.containsField("latestRestoreVersion")) {
                        testSpaceObject.setLatestRestoreVersion(document.getInt("latestRestoreVersion"));
                    }
                    return toType.cast(testSpaceObject);
                } else {
                    throw new RuntimeException("Unknown object type: " + toType);
                }
            }

            @Override
            public Object convert(Object type) {
                if (type instanceof Number) {
                    return type;
                }
                return type.toString();
            }

            @Override
            public Query toQuery(Object template) {
                throw new UnsupportedOperationException();
            }
        });

    }

}