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

Introduction

Here is the source code for org.mayocat.store.rdbms.dbi.argument.MapAsJsonArgumentFactory.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 java.util.Map;

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: bf3f957260b1b84f91379017bbaf43fad4d25041 $
 */
public class MapAsJsonArgumentFactory implements ArgumentFactory<Map> {
    @Override
    public boolean accepts(Class<?> expectedType, Object value, StatementContext ctx) {
        if (value == null) {
            return false;
        }
        return Map.class.isAssignableFrom(value.getClass());
    }

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

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

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