Examples with LongExtractor org.neo4j.csv.reader.Extractors.LongExtractor used on opensource projects

Example 1 with LongExtractor

use of org.neo4j.csv.reader.Extractors.LongExtractor in project neo4j by neo4j.

the class CsvInputParser method next.

boolean next(InputEntityVisitor visitor) throws IOException {
    lineNumber++;
    int i = 0;
    Header.Entry entry = null;
    Header.Entry[] entries = header.entries();
    try {
        boolean doContinue = true;
        for (i = 0; i < entries.length && doContinue; i++) {
            entry = entries[i];
            if (!seeker.seek(mark, delimiter)) {
                if (i > 0) {
                    throw new UnexpectedEndOfInputException("Near " + mark);
                }
                // We're just at the end
                return false;
            }
            switch(entry.type()) {
                case ID:
                    if (seeker.tryExtract(mark, entry.extractor())) {
                        switch(idType) {
                            case STRING:
                            case INTEGER:
                                Object idValue = entry.extractor().value();
                                doContinue = visitor.id(idValue, entry.group());
                                if (entry.name() != null) {
                                    doContinue = visitor.property(entry.name(), idValue);
                                }
                                break;
                            case ACTUAL:
                                doContinue = visitor.id(((LongExtractor) entry.extractor()).longValue());
                                break;
                            default:
                                throw new IllegalArgumentException(idType.name());
                        }
                    }
                    break;
                case START_ID:
                    if (seeker.tryExtract(mark, entry.extractor())) {
                        switch(idType) {
                            case STRING:
                                doContinue = visitor.startId(entry.extractor().value(), entry.group());
                                break;
                            case INTEGER:
                                doContinue = visitor.startId(entry.extractor().value(), entry.group());
                                break;
                            case ACTUAL:
                                doContinue = visitor.startId(((LongExtractor) entry.extractor()).longValue());
                                break;
                            default:
                                throw new IllegalArgumentException(idType.name());
                        }
                    }
                    break;
                case END_ID:
                    if (seeker.tryExtract(mark, entry.extractor())) {
                        switch(idType) {
                            case STRING:
                                doContinue = visitor.endId(entry.extractor().value(), entry.group());
                                break;
                            case INTEGER:
                                doContinue = visitor.endId(entry.extractor().value(), entry.group());
                                break;
                            case ACTUAL:
                                doContinue = visitor.endId(((LongExtractor) entry.extractor()).longValue());
                                break;
                            default:
                                throw new IllegalArgumentException(idType.name());
                        }
                    }
                    break;
                case TYPE:
                    if (seeker.tryExtract(mark, entry.extractor())) {
                        doContinue = visitor.type((String) entry.extractor().value());
                    }
                    break;
                case PROPERTY:
                    if (seeker.tryExtract(mark, entry.extractor(), entry.optionalParameter())) {
                        // TODO since PropertyStore#encodeValue takes Object there's no point splitting up
                        // into different primitive types
                        Object value = entry.extractor().value();
                        if (!isEmptyArray(value)) {
                            doContinue = visitor.property(entry.name(), value);
                        }
                    }
                    break;
                case LABEL:
                    if (seeker.tryExtract(mark, entry.extractor())) {
                        Object labelsValue = entry.extractor().value();
                        if (labelsValue.getClass().isArray()) {
                            doContinue = visitor.labels((String[]) labelsValue);
                        } else {
                            doContinue = visitor.labels(new String[] { (String) labelsValue });
                        }
                    }
                    break;
                case IGNORE:
                    break;
                default:
                    throw new IllegalArgumentException(entry.type().toString());
            }
            if (mark.isEndOfLine()) {
                // We're at the end of the line, break and return an entity with what we have.
                break;
            }
        }
        while (!mark.isEndOfLine()) {
            seeker.seek(mark, delimiter);
            if (doContinue) {
                seeker.tryExtract(mark, stringExtractor, entry.optionalParameter());
                badCollector.collectExtraColumns(seeker.sourceDescription(), lineNumber, stringExtractor.value());
            }
        }
        visitor.endOfEntity();
        return true;
    } catch (final RuntimeException e) {
        String stringValue = null;
        try {
            Extractors extractors = new Extractors('?');
            if (seeker.tryExtract(mark, extractors.string(), entry.optionalParameter())) {
                stringValue = extractors.string().value();
            }
        } catch (Exception e1) {
        // OK
        }
        String message = format("ERROR in input" + "%n  data source: %s" + "%n  in field: %s" + "%n  for header: %s" + "%n  raw field value: %s" + "%n  original error: %s", seeker, entry + ":" + (i + 1), header, stringValue != null ? stringValue : "??", e.getMessage());
        if (e instanceof InputException) {
            throw Exceptions.withMessage(e, message);
        }
        throw new InputException(message, e);
    }
}