edu.illinois.cs.cogcomp.wikifier.utils.io.CSVAdapter.java Source code

Introduction

Here is the source code for edu.illinois.cs.cogcomp.wikifier.utils.io.CSVAdapter.java

Source

/**
 * This software is released under the University of Illinois/Research and Academic Use License. See
 * the LICENSE file in the root folder for details. Copyright (c) 2016
 *
 * Developed by: The Cognitive Computation Group University of Illinois at Urbana-Champaign
 * http://cogcomp.cs.illinois.edu/
 */
package edu.illinois.cs.cogcomp.wikifier.utils.io;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import javax.xml.bind.annotation.adapters.XmlAdapter;

import org.apache.commons.lang3.StringUtils;

/**
 * Parses comma separated strings to list of strings
 * @author cheng88
 *
 */
public class CSVAdapter extends XmlAdapter<String, List<String>> {

    @Override
    public List<String> unmarshal(String v) throws Exception {
        return new ArrayList<String>(Arrays.asList(StringUtils.split(v, ',')));
    }

    @Override
    public String marshal(List<String> v) throws Exception {
        return StringUtils.join(v, ',');
    }

    public static class CSVSetAdapter extends XmlAdapter<String, Set<String>> {

        @Override
        public Set<String> unmarshal(String v) throws Exception {
            return new HashSet<String>(Arrays.asList(StringUtils.split(v, ',')));
        }

        @Override
        public String marshal(Set<String> v) throws Exception {
            return StringUtils.join(v, ',');
        }

    }

}