components/cronet/android/java/src/org/chromium/net/impl/AndroidUploadDataProviderWrapper.java

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.net.impl;

import static org.chromium.net.impl.HttpEngineNativeProvider.EXT_API_LEVEL;
import static org.chromium.net.impl.HttpEngineNativeProvider.EXT_VERSION;

import androidx.annotation.RequiresExtension;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.Objects;

@RequiresExtension(extension = EXT_API_LEVEL, version = EXT_VERSION)
@SuppressWarnings("Override")
class AndroidUploadDataProviderWrapper extends android.net.http.UploadDataProvider {
    private final org.chromium.net.UploadDataProvider mBackend;

    public AndroidUploadDataProviderWrapper(org.chromium.net.UploadDataProvider backend) {
        Objects.requireNonNull(backend, "Invalid UploadDataProvider.");
        this.mBackend = backend;
    }

    @Override
    public long getLength() throws IOException {
        return mBackend.getLength();
    }

    @Override
    public void read(android.net.http.UploadDataSink uploadDataSink, ByteBuffer byteBuffer)
            throws IOException {
        AndroidUploadDataSinkWrapper wrapper = new AndroidUploadDataSinkWrapper(uploadDataSink);
        mBackend.read(wrapper, byteBuffer);
    }

    @Override
    public void rewind(android.net.http.UploadDataSink uploadDataSink) throws IOException {
        AndroidUploadDataSinkWrapper wrapper = new AndroidUploadDataSinkWrapper(uploadDataSink);
        mBackend.rewind(wrapper);
    }

    @Override
    public void close() throws IOException {
        mBackend.close();
    }
}