/**
   * Creates and returns the "backup" service request buffer for the Service Manager.
   *
   * @param service Service handle
   * @param options The isc_spb_bkp_* parameters options to be used
   * @return the "backup" service request buffer for the Service Manager.
   */
  protected ServiceRequestBuffer getBackupSRB(FbService service, int options) throws SQLException {
    ServiceRequestBuffer backupSPB = service.createServiceRequestBuffer();
    backupSPB.addArgument(isc_action_svc_backup);
    backupSPB.addArgument(isc_spb_dbname, getDatabase());
    addBackupsToBackupRequestBuffer(service, backupSPB);

    if (verboseBackup()) {
      backupSPB.addArgument(isc_spb_verbose);
    }

    backupSPB.addArgument(isc_spb_options, options);

    return backupSPB;
  }
Пример #2
0
 @Override
 public byte[] getServiceInfo(
     ServiceParameterBuffer serviceParameterBuffer,
     ServiceRequestBuffer serviceRequestBuffer,
     int maxBufferLength)
     throws SQLException {
   try {
     final byte[] serviceParameterBufferBytes =
         serviceParameterBuffer == null ? null : serviceParameterBuffer.toBytesWithType();
     final byte[] serviceRequestBufferBytes =
         serviceRequestBuffer == null ? null : serviceRequestBuffer.toBytes();
     final ByteBuffer responseBuffer = ByteBuffer.allocateDirect(maxBufferLength);
     synchronized (getSynchronizationObject()) {
       clientLibrary.isc_service_query(
           statusVector,
           handle,
           new IntByReference(0),
           (short) (serviceParameterBufferBytes != null ? serviceParameterBufferBytes.length : 0),
           serviceParameterBufferBytes,
           (short) (serviceRequestBufferBytes != null ? serviceRequestBufferBytes.length : 0),
           serviceRequestBufferBytes,
           (short) maxBufferLength,
           responseBuffer);
       processStatusVector();
     }
     byte[] responseArray = new byte[maxBufferLength];
     responseBuffer.get(responseArray);
     return responseArray;
   } catch (SQLException e) {
     exceptionListenerDispatcher.errorOccurred(e);
     throw e;
   }
 }
  /**
   * Creates and returns the "backup" service request buffer for the Service Manager.
   *
   * @param service Service handle
   * @param options The options to be used for the backup operation
   * @return the "backup" service request buffer for the Service Manager.
   */
  protected ServiceRequestBuffer getRestoreSRB(FbService service, int options) {
    ServiceRequestBuffer restoreSPB = service.createServiceRequestBuffer();
    restoreSPB.addArgument(isc_action_svc_restore);

    // restore files with sizes except the last one
    for (Iterator<PathSizeStruct> iter = restorePaths.iterator(); iter.hasNext(); ) {
      PathSizeStruct pathSize = iter.next();

      restoreSPB.addArgument(isc_spb_dbname, pathSize.getPath());

      if (iter.hasNext() && pathSize.getSize() != -1) {
        restoreSPB.addArgument(isc_spb_res_length, pathSize.getSize());
      }
    }

    addBackupsToRestoreRequestBuffer(service, restoreSPB);

    if (restoreBufferCount != -1) {
      restoreSPB.addArgument(isc_spb_res_buffers, restoreBufferCount);
    }

    if (restorePageSize != -1) {
      restoreSPB.addArgument(isc_spb_res_page_size, restorePageSize);
    }

    restoreSPB.addArgument(
        isc_spb_res_access_mode,
        (byte) (restoreReadOnly ? isc_spb_res_am_readonly : isc_spb_res_am_readwrite));

    if (verbose) {
      restoreSPB.addArgument(isc_spb_verbose);
    }

    if ((options & RESTORE_CREATE) != RESTORE_CREATE
        && (options & RESTORE_REPLACE) != RESTORE_REPLACE) {
      options |= restoreReplace ? RESTORE_REPLACE : RESTORE_CREATE;
    }

    restoreSPB.addArgument(isc_spb_options, options);

    return restoreSPB;
  }
Пример #4
0
  /**
   * Test for service action.
   *
   * <p>Replicates the behavior of {@link FBStatisticsManager#getHeaderPage()}.
   */
  @Test
  public void testStartServiceAction() throws Exception {
    FBManager fbManager = createFBManager();
    defaultDatabaseSetUp(fbManager);
    try (JnaService service = factory.serviceConnect(connectionInfo)) {
      service.attach();

      ServiceRequestBuffer actionSrb = service.createServiceRequestBuffer();
      actionSrb.addArgument(isc_action_svc_db_stats);
      actionSrb.addArgument(isc_spb_dbname, getDatabasePath());
      actionSrb.addArgument(isc_spb_options, isc_spb_sts_hdr_pages);

      service.startServiceAction(actionSrb);

      ServiceRequestBuffer infoSrb = service.createServiceRequestBuffer();
      infoSrb.addArgument(isc_info_svc_to_eof);
      int bufferSize = 1024;
      ByteArrayOutputStream bos = new ByteArrayOutputStream();
      boolean processing = true;
      while (processing) {
        byte[] buffer = service.getServiceInfo(null, infoSrb, bufferSize);

        switch (buffer[0]) {
          case isc_info_svc_to_eof:
            int dataLength = iscVaxInteger2(buffer, 1);
            if (dataLength == 0) {
              if (buffer[3] != isc_info_end) {
                throw new SQLException("Unexpected end of stream reached.");
              } else {
                processing = false;
                break;
              }
            }
            bos.write(buffer, 3, dataLength);
            break;
          case isc_info_truncated:
            bufferSize = bufferSize * 2;
            break;
          case isc_info_end:
            processing = false;
            break;
        }
      }
      String headerPage = service.getEncoding().decodeFromCharset(bos.toByteArray());
      assertThat(
          "Expected database header page content",
          headerPage,
          allOf(
              startsWith("\nDatabase"),
              containsString("Database header page information"),
              endsWith("*END*\n")));
    } finally {
      defaultDatabaseTearDown(fbManager);
    }
  }
Пример #5
0
 @Override
 public void startServiceAction(ServiceRequestBuffer serviceRequestBuffer) throws SQLException {
   try {
     final byte[] serviceRequestBufferBytes =
         serviceRequestBuffer == null ? null : serviceRequestBuffer.toBytes();
     synchronized (getSynchronizationObject()) {
       clientLibrary.isc_service_start(
           statusVector,
           handle,
           new IntByReference(0),
           (short) (serviceRequestBufferBytes != null ? serviceRequestBufferBytes.length : 0),
           serviceRequestBufferBytes);
       processStatusVector();
     }
   } catch (SQLException e) {
     exceptionListenerDispatcher.errorOccurred(e);
     throw e;
   }
 }