com.arcadedb.server.ha.message.FileContentRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arcadedb-server Show documentation
The newest version!
/*
 * Copyright © 2021-present Arcade Data Ltd ([email protected])
 *
 * 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.
 *
 * SPDX-FileCopyrightText: 2021-present Arcade Data Ltd ([email protected])
 * SPDX-License-Identifier: Apache-2.0
 */
package com.arcadedb.server.ha.message;

import com.arcadedb.database.Binary;
import com.arcadedb.database.DatabaseInternal;
import com.arcadedb.engine.ComponentFile;
import com.arcadedb.engine.ImmutablePage;
import com.arcadedb.engine.PageId;
import com.arcadedb.engine.PaginatedComponentFile;
import com.arcadedb.log.LogManager;
import com.arcadedb.network.binary.NetworkProtocolException;
import com.arcadedb.server.ArcadeDBServer;
import com.arcadedb.server.ha.HAServer;

import java.io.*;
import java.util.concurrent.atomic.*;
import java.util.logging.*;

public class FileContentRequest extends HAAbstractCommand {
  private              String databaseName;
  private              int    fileId;
  private              int    fromPageInclusive; //  INCLUSIVE
  private              int    toPageInclusive; //  INCLUSIVE
  private static final int    CHUNK_MAX_PAGES = 10;

  public FileContentRequest() {
  }

  public FileContentRequest(final String dbName, final int fileId, final int pageFromInclusive, final int pageToInclusive) {
    this.databaseName = dbName;
    this.fileId = fileId;
    this.fromPageInclusive = pageFromInclusive;
    this.toPageInclusive = pageToInclusive;
  }

  @Override
  public HACommand execute(final HAServer server, final String remoteServerName, final long messageNumber) {
    final DatabaseInternal db = server.getServer().getDatabase(databaseName);
    final ComponentFile file = db.getFileManager().getFile(fileId);

    if (file instanceof PaginatedComponentFile) {
      final int pageSize = ((PaginatedComponentFile) file).getPageSize();

      try {
        final int totalPages = (int) (file.getSize() / pageSize);

        final Binary pagesContent = new Binary();

        final AtomicInteger pages = new AtomicInteger(0);

        if (toPageInclusive == -1)
          toPageInclusive = totalPages - 1;

//        db.getPageManager().suspendFlushAndExecute(() -> {
        for (int i = fromPageInclusive; i <= toPageInclusive && pages.get() < CHUNK_MAX_PAGES; ++i) {
          final PageId pageId = new PageId(db, fileId, i);
          final ImmutablePage page = db.getPageManager().getImmutablePage(pageId, pageSize, false, false);
          pagesContent.putByteArray(page.getContent().array(), pageSize);

          pages.incrementAndGet();
        }
//        });

        final boolean last = pages.get() > toPageInclusive;

        pagesContent.flip();

        return new FileContentResponse(databaseName, fileId, file.getFileName(), fromPageInclusive, pagesContent, pages.get(),
            last);

      } catch (final IOException e) {
        throw new NetworkProtocolException("Cannot load pages", e);
//      } catch (InterruptedException e) {
//        Thread.currentThread().interrupt();
//        throw new NetworkProtocolException("Cannot load pages", e);
      }
    }
    LogManager.instance().log(this, Level.SEVERE, "Cannot read not paginated file %s from the leader", file.getFileName());
    throw new NetworkProtocolException("Cannot read not paginated file " + file.getFileName() + " from the leader");
  }

  @Override
  public void toStream(final Binary stream) {
    stream.putString(databaseName);
    stream.putInt(fileId);
    stream.putInt(fromPageInclusive);
    stream.putInt(toPageInclusive);
  }

  @Override
  public void fromStream(final ArcadeDBServer server, final Binary stream) {
    databaseName = stream.getString();
    fileId = stream.getInt();
    fromPageInclusive = stream.getInt();
    toPageInclusive = stream.getInt();
  }

  @Override
  public String toString() {
    return "file(" + databaseName + " fileId=" + fileId + " fromPageInclusive=" + fromPageInclusive + " fromPageInclusive"
        + toPageInclusive + ")";
  }
}