org.mayocat.store.rdbms.dbi.argument.JsonArgumentAsJsonArgumentFactory.java Source code

Introduction

Here is the source code for org.mayocat.store.rdbms.dbi.argument.JsonArgumentAsJsonArgumentFactory.java

Source

/*
 * Copyright (c) 2012, Mayocat <hello@mayocat.org>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.mayocat.store.rdbms.dbi.argument;

import java.sql.PreparedStatement;
import java.sql.SQLException;

import org.postgresql.util.PGobject;
import org.skife.jdbi.v2.StatementContext;
import org.skife.jdbi.v2.tweak.Argument;
import org.skife.jdbi.v2.tweak.ArgumentFactory;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * @version $Id: 1efeac46b32dfdf7b3d1efa3dec5041efee44e62 $
 */
public class JsonArgumentAsJsonArgumentFactory implements ArgumentFactory<JsonArgument> {
    @Override
    public boolean accepts(Class<?> expectedType, Object value, StatementContext ctx) {
        if (value == null) {
            return false;
        }
        return JsonArgument.class.isAssignableFrom(value.getClass());
    }

    @Override
    public Argument build(Class<?> expectedType, final JsonArgument value, StatementContext ctx) {
        try {
            final PGobject jsonObject = new PGobject();
            final ObjectMapper mapper = new ObjectMapper();

            jsonObject.setType("json");
            jsonObject.setValue(mapper.writeValueAsString(value.getWrapped()));

            return new Argument() {
                @Override
                public void apply(int position, PreparedStatement statement, StatementContext ctx)
                        throws SQLException {
                    statement.setObject(position, jsonObject);
                }
            };
        } catch (JsonProcessingException e) {
            throw new IllegalStateException("Failed to map JSON argument", e);
        } catch (SQLException e) {
            throw new IllegalStateException("Failed to map JSON argument", e);
        }
    }
}