org.jnosql.diana.cassandra.column.UDTBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cassandra-driver Show documentation

The Eclipse JNoSQL communication layer, Diana, to Cassandra

There is a newer version: 0.0.9

/*
 *  Copyright (c) 2017 Otávio Santana and others
 *   All rights reserved. This program and the accompanying materials
 *   are made available under the terms of the Eclipse Public License v1.0
 *   and Apache License v2.0 which accompanies this distribution.
 *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
 *
 *   You may elect to redistribute this code under either of these licenses.
 *
 *   Contributors:
 *
 *   Otavio Santana
 */
package org.jnosql.diana.cassandra.column;


import org.jnosql.diana.api.column.Column;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.stream.StreamSupport;

/**
 * The UDT builder
 */
class UDTBuilder implements UDTNameBuilder, UDTElementBuilder, UDTFinisherBuilder {

    private String name;

    private String typeName;

    private List columns = new ArrayList<>();

    private Iterable> udts = new ArrayList<>();


    UDTBuilder(String userType) {
        this.typeName = userType;
    }

    @Override
    public UDTBuilder withName(String name) throws NullPointerException {
        this.name = Objects.requireNonNull(name, "name is required");
        return this;
    }


    @Override
    public UDTBuilder addUDT(Iterable udt) throws NullPointerException {
        Objects.requireNonNull(udt, "udt is required");
        StreamSupport.stream(udt.spliterator(), false).forEach(this.columns::add);
        return this;
    }

    @Override
    public UDTBuilder addUDTs(Iterable> udts) throws NullPointerException {
        Objects.requireNonNull(udts, "udts is required");
        this.udts = udts;
        return this;
    }

    @Override
    public UDT build() throws IllegalStateException {
        if (Objects.isNull(name)) {
            throw new IllegalStateException("name is required");
        }
        if (Objects.isNull(typeName)) {
            throw new IllegalStateException("typeName is required");
        }

        if (udts.spliterator().getExactSizeIfKnown() == 0) {
            return new DefaultUDT(name, typeName, columns);
        } else {
            return new IterableUDT(name, typeName, udts);
        }
    }


}