All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • apoc.load.CSVResult Maven / Gradle / Ivy

    There is a newer version: 5.25.1
    Show newest version
    package apoc.load;

import apoc.load.util.Results;

import java.util.*;

import static java.util.Arrays.asList;
import static java.util.Collections.emptyList;
import static java.util.Collections.emptyMap;

public class CSVResult {
    public long lineNo;
    public List list;
    public List strings;
    public Map map;
    public Map stringMap;

    public CSVResult(String[] header, String[] list, long lineNo, boolean ignore, Map mapping, List nullValues, EnumSet results) {
        this.lineNo = lineNo;
        removeNullValues(list, nullValues);

        this.strings = results.contains(Results.strings) ?
                (List)createList(header, list, ignore, mapping, false) : emptyList();
        this.stringMap = results.contains(Results.stringMap) ?
                (Map)createMap(header, list, ignore, mapping,false) : emptyMap();
        this.map = results.contains(Results.map) ?
                createMap(header, list, ignore, mapping,true) : emptyMap();
        this.list = results.contains(Results.list) ?
                    createList(header, list, ignore, mapping, true) : emptyList();
    }

    public void removeNullValues(String[] list, List nullValues) {
        if (nullValues.isEmpty()) return;
        for (int i = 0; i < list.length; i++) {
            if (nullValues.contains(list[i]))list[i] = null;
        }
    }

    private List createList(String[] header, String[] list, boolean ignore, Map mappings, boolean convert) {
        if (!ignore && mappings.isEmpty()) return asList((Object[]) list);
        ArrayList result = new ArrayList<>(list.length);
        for (int i = 0; i < header.length; i++) {
            String name = header[i];
            if (name == null) continue;
            Mapping mapping = mappings.get(name);
            if (mapping != null) {
                if (mapping.ignore) continue;
                result.add(convert ? mapping.convert(list[i]) : list[i]);
            } else {
                result.add(list[i]);
            }
        }
        return result;
    }

    private Map createMap(String[] header, String[] list, boolean ignore, Map mappings, boolean convert) {
        if (header == null) return null;
        Map map = new LinkedHashMap<>(header.length, 1f);
        for (int i = 0; i < header.length; i++) {
            String name = header[i];
            if (ignore && name == null) continue;
            Mapping mapping = mappings.get(name);
            if (mapping == null) {
                map.put(name, list[i]);
            } else {
                if (mapping.ignore) continue;
                map.put(mapping.name, convert ? mapping.convert(list[i]) : list[i]);
            }
        }
        return map;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy