db.migration.util.DbUtil.java Source code

Introduction

Here is the source code for db.migration.util.DbUtil.java

Source

/*
    
epitope-service  T-cell epitope group matching service for HLA-DPB1 locus.
Copyright (c) 2014-2015 National Marrow Donor Program (NMDP)
    
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published
by the Free Software Foundation; either version 3 of the License, or (at
your option) any later version.
    
This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; with out even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
License for more details.
    
You should have received a copy of the GNU Lesser General Public License
along with this library;  if not, write to the Free Software Foundation,
Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA.
    
> http://www.gnu.org/licenses/lgpl.html
    
*/

package db.migration.util;

import java.net.URL;
import java.sql.Connection;
import java.util.Arrays;
import java.util.Iterator;

import org.skife.jdbi.v2.DBI;
import org.skife.jdbi.v2.Handle;
import org.skife.jdbi.v2.PreparedBatch;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.dataformat.csv.CsvMapper;
import com.fasterxml.jackson.dataformat.csv.CsvParser;

public class DbUtil {

    static Logger log = LoggerFactory.getLogger(DbUtil.class);

    public static void loadCsv(Connection conn, String sql, String csvResource) throws Exception {
        log.debug("csvResource: " + csvResource);
        URL csvUrl = DbUtil.class.getClassLoader().getResource(csvResource);
        log.debug("csvUrl: " + csvUrl);
        try (Handle handle = DBI.open(conn)) {
            PreparedBatch batch = handle.prepareBatch(sql);
            Iterator<Object[]> it = readCsv(csvUrl);
            while (it.hasNext()) {
                Object[] row = it.next();
                log.trace("row: " + Arrays.toString(row));
                batch.add(row);
            }
            batch.execute();
        }
    }

    public static Iterator<Object[]> readCsv(URL csvUrl) throws Exception {
        CsvMapper mapper = new CsvMapper();
        mapper.enable(CsvParser.Feature.WRAP_AS_ARRAY);
        return mapper.reader(String[].class).readValues(csvUrl);
    }

}