org.opensearch.ml.common.dataframe.ColumnValueBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of opensearch-ml-common Show documentation

OpenSearch ML Common

There is a newer version: 2.17.1.0

/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.opensearch.ml.common.dataframe;

import java.util.Objects;

import lombok.experimental.UtilityClass;

@UtilityClass
public class ColumnValueBuilder {

    /**
     * build column value based on input object
     * @param object input object
     * @return ColumnValue
     */
    public ColumnValue build(Object object) {
        if (Objects.isNull(object)) {
            return new NullValue();
        }

        if (object instanceof Short) {
            return new ShortValue((Short) object);
        }

        if (object instanceof Integer) {
            return new IntValue((Integer) object);
        }

        if (object instanceof Long) {
            return new LongValue((Long) object);
        }

        if (object instanceof String) {
            return new StringValue((String) object);
        }

        if (object instanceof Double) {
            return new DoubleValue((Double) object);
        }

        if (object instanceof Boolean) {
            return new BooleanValue((Boolean) object);
        }

        if (object instanceof Float) {
            return new FloatValue((Float) object);
        }

        throw new IllegalArgumentException("unsupported type:" + object.getClass().getName());
    }
}