io.resys.thena.docdb.file.spi.BlobFileBuilderImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thena-docdb-api Show documentation
There is a newer version: 1.010
package io.resys.thena.docdb.file.spi;

import java.util.ArrayList;
import java.util.Arrays;

/*-
 * #%L
 * thena-docdb-pgsql
 * %%
 * Copyright (C) 2021 Copyright 2021 ReSys OÜ
 * %%
 * 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.
 * #L%
 */

import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

import io.resys.thena.docdb.api.models.Objects.Blob;
import io.resys.thena.docdb.api.models.Objects.Tree;
import io.resys.thena.docdb.file.FileBuilder.BlobFileBuilder;
import io.resys.thena.docdb.file.tables.BlobTable.BlobTableRow;
import io.resys.thena.docdb.file.tables.ImmutableBlobTableRow;
import io.resys.thena.docdb.file.tables.ImmutableFileStatement;
import io.resys.thena.docdb.file.tables.ImmutableFileTuple;
import io.resys.thena.docdb.file.tables.ImmutableFileTupleList;
import io.resys.thena.docdb.file.tables.Table;
import io.resys.thena.docdb.file.tables.Table.FileStatement;
import io.resys.thena.docdb.file.tables.Table.FileTuple;
import io.resys.thena.docdb.file.tables.Table.FileTupleList;
import io.resys.thena.docdb.spi.ClientCollections;
import io.vertx.mutiny.sqlclient.Tuple;
import lombok.RequiredArgsConstructor;

@RequiredArgsConstructor
public class BlobFileBuilderImpl implements BlobFileBuilder {
  
  private final ClientCollections ctx;
  
  @Override
  public FileStatement create() {
    return ImmutableFileStatement.builder()
        .value("create BLOB table if it does not exist")
        .command((root) -> {
          root.getRepoTable(ctx).getBlobs().create();
          return Arrays.asList(new Table.Row() {});
        })
        .build();
  }
  @Override
  public FileStatement findAll() {
    return ImmutableFileStatement.builder()
        .value("Select all from BLOB table")
        .command((root) -> root.getRepoTable(ctx).getBlobs().getRows())
        .build();
  }
  @Override
  public FileTuple getById(String blobId) {
    return ImmutableFileTuple.builder()
        .value("Select by id from BLOB table")
        .command((root) -> root.getRepoTable(ctx).getBlobs().getRows()
            .stream().filter(r -> r.getId().equals(blobId)).collect(Collectors.toList()))
        .props(Tuple.of(blobId))
        .build();
  }
  @Override
  public FileTuple findByIds(Collection blobId) {
    List tuple = new ArrayList<>();
    for(final var id : blobId) {
      tuple.add(id);
    }
    
    return ImmutableFileTuple.builder()
        .value("Select by multiple id-s from BLOB table")
        .command((root) -> root.getRepoTable(ctx).getBlobs().getRows()
            .stream().filter(r -> blobId.contains(r.getId())).collect(Collectors.toList()))
        .props(Tuple.from(tuple))
        .build();
  }
  @Override
  public FileTuple findByTree(Tree tree) {
    return ImmutableFileTuple.builder()
        .value("Select by tree from BLOB and TREE_ITEM table")
        .command((root) -> {
          final var ids = root.getRepoTable(ctx).getTreeItems().getRows().stream()
            .filter(item -> item.getTree().equals(tree.getId()))
            .map(item -> item.getBlob())
            .collect(Collectors.toList());
          
          return root.getRepoTable(ctx).getBlobs().getRows()
              .stream().filter(r -> ids.contains(r.getId())).collect(Collectors.toList());
          
        })
        .props(Tuple.of(tree.getId()))
        .build();
  }
  @Override
  public FileTuple insertOne(Blob blob) {
    return ImmutableFileTuple.builder()
        .value("Insert new row into BLOB table")
        .command((root) -> {
          
          final var exists = root.getRepoTable(ctx).getBlobs().getRows()
            .stream().filter(r -> r.getId().equals(blob.getId()))
            .findFirst();
          if(!exists.isEmpty()) {
            return Arrays.asList(exists.get());
          }
          
          final var newRow = ImmutableBlobTableRow.builder().id(blob.getId()).value(blob.getValue()).build();
          root.getRepoTable(ctx).getBlobs().insert(newRow);
          
          return Arrays.asList(newRow);
        })
        .props(Tuple.of(blob.getId(), blob.getValue()))
        .build();
  }
  @Override
  public FileTupleList insertAll(Collection blobs) {
    return ImmutableFileTupleList.builder()
        .value("Insert new rows into BLOB table")
        .command((root) -> {
          final var byId = root.getRepoTable(ctx).getBlobs()
              .getRows().stream()
              .collect(Collectors.toMap(e -> e.getId(), e -> e));
          
          final var inserts = new ArrayList();
          final var results = new ArrayList();
          
          blobs.stream().forEach(blob -> {
            if(byId.containsKey(blob.getId())) {
              final var newRow = ImmutableBlobTableRow.builder().id(blob.getId()).value(blob.getValue()).build();
              inserts.add(newRow);
              results.add(newRow);
            } else {
              results.add(byId.get(blob.getId()));              
            }  
          });
          root.getRepoTable(ctx).getBlobs().insertAll(inserts);
          return results;
        })
        .build();
  }
}