ai.djl.ndarray.BytesSupplierImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Deep Java Library - api
The newest version!
/*
 * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance
 * with the License. A copy of the License is located at
 *
 * http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
 * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions
 * and limitations under the License.
 */
package ai.djl.ndarray;

import ai.djl.util.JsonUtils;

import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;

class BytesSupplierImpl implements BytesSupplier {

    private byte[] buf;
    private String value;
    private Object obj;

    BytesSupplierImpl(byte[] buf) {
        this.buf = buf;
    }

    BytesSupplierImpl(String value) {
        this.value = value;
    }

    BytesSupplierImpl(Object obj) {
        this.obj = obj;
    }

    /** {@inheritDoc} */
    @Override
    public byte[] getAsBytes() {
        if (buf == null) {
            if (value == null) {
                value = JsonUtils.toJson(obj) + '\n';
            }
            buf = value.getBytes(StandardCharsets.UTF_8);
        }
        return buf;
    }

    /** {@inheritDoc} */
    @Override
    public String getAsString() {
        if (value == null) {
            if (obj != null) {
                value = JsonUtils.toJson(obj) + '\n';
            } else {
                value = new String(buf, StandardCharsets.UTF_8);
            }
        }
        return value;
    }

    /** {@inheritDoc} */
    @Override
    public Object getAsObject() {
        if (obj != null) {
            return obj;
        } else if (value != null) {
            return value;
        }
        return buf;
    }

    /** {@inheritDoc} */
    @Override
    public ByteBuffer toByteBuffer() {
        return ByteBuffer.wrap(getAsBytes());
    }
}