btestbulk.cpp

これは、btest.cpp に基づく一括作成および一括取得の例です。

#include <string.h>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <stdint.h>

#include "btrieveCpp.h"

static char *btrieveFileName = (char *)"squaresAndSquareRoots.btr";

#define MIN_X 0
#define MAX_X 255

#pragma pack(1)
typedef struct {
   uint8_t x;
   uint16_t xSquared;
   double xSquareRoot;
} record_t;
#pragma pack()

static Btrieve::StatusCode
createFile(BtrieveClient *btrieveClient)
{
   Btrieve::StatusCode status;
   BtrieveFileAttributes btrieveFileAttributes;

   // If SetFixedRecordLength() fails.
   if ((status = btrieveFileAttributes.SetFixedRecordLength(sizeof(record_t))) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFileAttributes::SetFixedRecordLength():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If FileCreate() fails.
   if ((status = btrieveClient->FileCreate(&btrieveFileAttributes, btrieveFileName, Btrieve::CREATE_MODE_OVERWRITE)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveClient::FileCreate():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

leave:
   return status;
}

static Btrieve::StatusCode
openFile(BtrieveClient *btrieveClient, BtrieveFile *btrieveFile)
{
   Btrieve::StatusCode status;

   // If FileOpen() fails.
   if ((status = btrieveClient->FileOpen(btrieveFile, btrieveFileName, NULL, Btrieve::OPEN_MODE_NORMAL)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveClient::FileOpen():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

leave:
   return status;
}

static Btrieve::StatusCode
loadFile(BtrieveFile *btrieveFile)
{
   Btrieve::StatusCode status = Btrieve::STATUS_CODE_NO_ERROR;
   BtrieveBulkCreatePayload btrieveBulkCreatePayload;
   BtrieveBulkCreateResult btrieveBulkCreateResult;
   long long cursorPosition;
   int i;
   int recordCount;
   record_t record;

   for (i = MIN_X; i <= MAX_X; i++)
   {
      record.x = (uint8_t)i;
      record.xSquared = (uint16_t)(i * i);
      record.xSquareRoot = sqrt((double)i);

      // If AddRecord() fails.
      if ((status = btrieveBulkCreatePayload.AddRecord((char *)&record, sizeof(record))) != Btrieve::STATUS_CODE_NO_ERROR)
      {
         printf("Error: BtrieveBulkCreatePayload::AddRecord():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
         goto leave;
      }
   }

   // If BulkCreate() fails.
   if ((status = btrieveFile->BulkCreate(&btrieveBulkCreatePayload, &btrieveBulkCreateResult)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFile::BulkCreate():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If GetRecordCount() fails.
   if ((recordCount = btrieveBulkCreateResult.GetRecordCount()) != (MAX_X - MIN_X + 1))
   {
      status = btrieveBulkCreateResult.GetLastStatusCode();
      printf("Error: BtrieveBulkCreateResult::GetRecordCount():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   for (i = 1; i <= recordCount; i++)
   {
      // If GetRecordCursorPosition() fails.
      if ((cursorPosition = btrieveBulkCreateResult.GetRecordCursorPosition(i)) < 0)
      {
         status = btrieveBulkCreateResult.GetLastStatusCode();
         printf("Error: BtrieveBulkCreateResult::GetRecordCursorPosition():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
         goto leave;
      }
   }

leave:
   return status;
}

static Btrieve::StatusCode
closeFile(BtrieveClient *btrieveClient, BtrieveFile *btrieveFile)
{
   Btrieve::StatusCode status;

   // If FileClose() fails.
   if ((status = btrieveClient->FileClose(btrieveFile)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveClient::FileClose():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

leave:
   return status;
}

static Btrieve::StatusCode
deleteFile(BtrieveClient *btrieveClient)
{
   Btrieve::StatusCode status;

   // If FileDelete() fails.
   if ((status = btrieveClient->FileDelete(btrieveFileName)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveClient::FileDelete():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

leave:
   return status;
}

static Btrieve::StatusCode
retrieveRecord(BtrieveFile *btrieveFile, uint8_t *key)
{
   Btrieve::StatusCode status = Btrieve::STATUS_CODE_NO_ERROR;
   BtrieveFilter btrieveFilter;
   BtrieveBulkRetrieveAttributes btrieveBulkRetrieveAttributes;
   BtrieveBulkRetrieveResult btrieveBulkRetrieveResult;
   int i;
   record_t record;
   long long cursorPosition;

   // If SetField() fails.
   if ((status = btrieveFilter.SetField(0, 1, Btrieve::DATA_TYPE_UNSIGNED_BINARY)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFilter::SetField():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If SetComparison() fails.
   if ((status = btrieveFilter.SetComparison(Btrieve::COMPARISON_EQUAL)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFilter::SetComparison():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If SetComparisonConstant() fails.
   if ((status = btrieveFilter.SetComparisonConstant((char *)key, 1)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFilter::SetComparisonConstant():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If AddField() fails.
   if ((status = btrieveBulkRetrieveAttributes.AddField(0, sizeof(record))) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveBulkRetrieveAttributes::AddField():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If AddFilter() fails.
   if ((status = btrieveBulkRetrieveAttributes.AddFilter(&btrieveFilter)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveBulkRetrieveAttributes::AddFilter():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If SetMaximumRecordCount() fails.
   if ((status = btrieveBulkRetrieveAttributes.SetMaximumRecordCount(1)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveBulkRetrieveAttributes::SetMaximumRecordCount():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If SetMaximumRejectCount() fails.
   if ((status = btrieveBulkRetrieveAttributes.SetMaximumRejectCount(MAX_X - MIN_X + 1)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveBulkRetrieveAttributes::SetMaximumRejectCount():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If SetSkipCurrentRecord() fails.
   if ((status = btrieveBulkRetrieveAttributes.SetSkipCurrentRecord(false)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveBulkRetrieveAttributes::SetSkipCurrentRecord():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If RecordRetrieveFirst() fails.
   if (btrieveFile->RecordRetrieveFirst(Btrieve::INDEX_NONE, (char *)&record, sizeof(record)) != sizeof(record))
   {
      status = btrieveFile->GetLastStatusCode();
      printf("Error: BtrieveFile::RecordRetrieveFirst():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If BulkRetrieveNext() fails.
   if ((status = btrieveFile->BulkRetrieveNext(&btrieveBulkRetrieveAttributes, &btrieveBulkRetrieveResult)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      printf("Error: BtrieveFile::BulkRetrieveNext():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If GetRecord() fails.
   if (btrieveBulkRetrieveResult.GetRecord(1, (char *)&record, sizeof(record)) != sizeof(record))
   {
      status = btrieveBulkRetrieveResult.GetLastStatusCode();
      printf("Error: BtrieveBulkRetrieveResult::GetRecord():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If GetRecordCount() fails.
   if ((i = btrieveBulkRetrieveResult.GetRecordCount()) != 1)
   {
      status = btrieveBulkRetrieveResult.GetLastStatusCode();
      printf("Error: BtrieveBulkRetrieveResult::GetRecordCount():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If GetRecordCursorPosition() fails.
   if ((cursorPosition = btrieveBulkRetrieveResult.GetRecordCursorPosition(1)) < 0)
   {
      status = btrieveBulkRetrieveResult.GetLastStatusCode();
      printf("Error: BtrieveBulkRetrieveResult::GetRecordCursorPosition():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   // If GetRecordLength() fails.
   if ((i = btrieveBulkRetrieveResult.GetRecordLength(1)) != sizeof(record))
   {
      status = btrieveBulkRetrieveResult.GetLastStatusCode();
      printf("Error: BtrieveBulkRetrieveResult::GetRecordLength():%d:%s.\n", status, Btrieve::StatusCodeToString(status));
      goto leave;
   }

   printf("record: (%u, %u, %g)\n", record.x, record.xSquared, record.xSquareRoot);

leave:
   return status;
}

int
main(int argc, char *argv[])
{
   static const char *usage = "Usage: %s uint8_value\n";

   BtrieveClient btrieveClient;
   Btrieve::StatusCode status = Btrieve::STATUS_CODE_UNKNOWN;
   BtrieveFile btrieveFile;
   uint8_t key;
   uint64_t integerValue;

   // If the incorrect number of arguments were given.
   if (argc != 2)
   {
      printf(usage, argv[0]);
      goto leave;
   }

   integerValue = atoi(argv[1]);

   // If integerValue is out of range.
   if ((integerValue < MIN_X) || (integerValue > MAX_X))
   {
      printf(usage, argv[0]);
      goto leave;
   }

   key = (uint8_t)integerValue;

   // If createFile() fails.
   if ((status = createFile(&btrieveClient)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

   // If openFile() fails.
   if ((status = openFile(&btrieveClient, &btrieveFile)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

   // If loadFile() fails.
   if ((status = loadFile(&btrieveFile)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

   // If retrieveRecord() fails.
   if ((status = retrieveRecord(&btrieveFile, &key)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

   // If closeFile() fails.
   if ((status = closeFile(&btrieveClient, &btrieveFile)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

   // If deleteFile() fails.
   if ((status = deleteFile(&btrieveClient)) != Btrieve::STATUS_CODE_NO_ERROR)
   {
      goto leave;
   }

leave:
   // If there wasn't a failure.
   if (status == Btrieve::STATUS_CODE_NO_ERROR)
      return 0;

   return 1;
}