io.github.honhimw.ms.internal.DocumentsImpl Maven / Gradle / Ivy

Go to download
/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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 io.github.honhimw.ms.internal;

import io.github.honhimw.ms.api.Documents;
import io.github.honhimw.ms.api.reactive.ReactiveDocuments;
import io.github.honhimw.ms.json.TypeRef;
import io.github.honhimw.ms.model.*;
import io.github.honhimw.ms.support.ReactorUtils;
import jakarta.annotation.Nullable;

import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.function.Consumer;

/**
 * @author hon_him
 * @since 2024-01-03
 */

class DocumentsImpl implements Documents {

    private final ReactiveDocuments _documents;

    protected DocumentsImpl(ReactiveDocuments documents) {
        this._documents = documents;
    }

    @Override
    public Page> list(@Nullable Integer offset, @Nullable Integer limit) {
        return ReactorUtils.blockNonNull(_documents.list(offset, limit));
    }

    @Override
    public Page> list(Consumer page) {
        return ReactorUtils.blockNonNull(_documents.list(page));
    }

    @Override
    public Page> list(GetDocumentRequest page) {
        return ReactorUtils.blockNonNull(_documents.list(page));
    }

    @Override
    public  Page list(@Nullable Integer offset, @Nullable Integer limit, TypeRef typeRef) {
        return ReactorUtils.blockNonNull(_documents.list(offset, limit, typeRef));
    }

    @Override
    public  Page list(@Nullable Integer offset, @Nullable Integer limit, Class type) {
        return ReactorUtils.blockNonNull(_documents.list(offset, limit, type));
    }

    @Override
    public  Page list(Consumer page, TypeRef typeRef) {
        return ReactorUtils.blockNonNull(_documents.list(page, typeRef));
    }

    @Override
    public  Page list(Consumer page, Class type) {
        return ReactorUtils.blockNonNull(_documents.list(page, type));
    }

    @Override
    public TaskInfo save(@Nullable String json) {
        return ReactorUtils.blockNonNull(_documents.save(json));
    }

    @Override
    public TaskInfo save(Object one) {
        return ReactorUtils.blockNonNull(_documents.save(one));
    }

    @Override
    public TaskInfo save(Collection collection) {
        return ReactorUtils.blockNonNull(_documents.save(collection));
    }

    @Override
    public TaskInfo saveVectorized(VectorizedDocument one) {
        return ReactorUtils.blockNonNull(_documents.saveVectorized(one));
    }

    @Override
    public TaskInfo saveVectorized(Collection collection) {
        return ReactorUtils.blockNonNull(_documents.saveVectorized(collection));
    }

    @Override
    public TaskInfo update(String json) {
        return ReactorUtils.blockNonNull(_documents.update(json));
    }

    @Override
    public TaskInfo update(Object one) {
        return ReactorUtils.blockNonNull(_documents.update(one));
    }

    @Override
    public TaskInfo update(Collection collection) {
        return ReactorUtils.blockNonNull(_documents.update(collection));
    }

    @Override
    public TaskInfo deleteAll() {
        return ReactorUtils.blockNonNull(_documents.deleteAll());
    }

    @Override
    public Page> batchGet(BatchGetDocumentsRequest fetch) {
        return ReactorUtils.blockNonNull(_documents.batchGet(fetch));
    }

    @Override
    public Page> batchGet(Consumer builder) {
        return ReactorUtils.blockNonNull(_documents.batchGet(builder));
    }

    @Override
    public  Page batchGet(BatchGetDocumentsRequest fetch, TypeRef typeRef) {
        return ReactorUtils.blockNonNull(_documents.batchGet(fetch, typeRef));
    }

    @Override
    public  Page batchGet(Consumer builder, TypeRef typeRef) {
        return ReactorUtils.blockNonNull(_documents.batchGet(builder, typeRef));
    }

    @Override
    public  Page batchGet(BatchGetDocumentsRequest fetch, Class type) {
        return ReactorUtils.blockNonNull(_documents.batchGet(fetch, type));
    }

    @Override
    public  Page batchGet(Consumer builder, Class type) {
        return ReactorUtils.blockNonNull(_documents.batchGet(builder, type));
    }

    @Override
    public TaskInfo batchDelete(List ids) {
        return ReactorUtils.blockNonNull(_documents.batchDelete(ids));
    }

    @Override
    public TaskInfo delete(FilterableAttributesRequest filter) {
        return ReactorUtils.blockNonNull(_documents.delete(filter));
    }

    @Override
    public Optional> get(String id, @Nullable String... fields) {
        return _documents.get(id, fields).blockOptional();
    }

    @Override
    public  Optional get(String id, TypeRef typeRef, @Nullable String... fields) {
        return _documents.get(id, typeRef, fields).blockOptional();
    }

    @Override
    public  Optional get(String id, Class type, @Nullable String... fields) {
        return _documents.get(id, type, fields).blockOptional();
    }

    @Override
    public TaskInfo delete(String id) {
        return ReactorUtils.blockNonNull(_documents.delete(id));
    }
}