testleak.cpp

fastdb-2.92的源码

//-< TESTLEAK.CPP >--------------------------------------------------*--------*
// FastDB                    Version 1.0         (c) 1999  GARRET    *     ?  *
// (Main Memory Database Management System)                          *   /\|  *
//                                                                   *  /  \  *
//                          Created:     26-Jul-2001  K.A. Knizhnik  * / [] \ *
//                          Last update: 26-Jul-2001  K.A. Knizhnik  * GARRET *
//-------------------------------------------------------------------*--------*
// Memory allocator test
//-------------------------------------------------------------------*--------*

#include "fastdb.h"
#include <stdio.h>

#define N_ITERATIONS  2000
#define N_KEYS        1000
#define VALUE_LENGTH  10000

class Test {
  public:
    char const* key;
    char const* value;

    TYPE_DESCRIPTOR((KEY(key, INDEXED), FIELD(value)));
};
   
REGISTER(Test);

int main(int argc, char* argv[])
{
    dbDatabase db;
    char buf[64];
    int nIterations = N_ITERATIONS;
    if (argc > 1) { 
        nIterations = atoi(argv[1]);
    }
    if (db.open("testleak")) {
        Test t;
        sprintf(buf, "KEY-%08d", 0);    
        char* p = new char[VALUE_LENGTH];       
        memset(p, ' ', VALUE_LENGTH);
        p[VALUE_LENGTH-1] = '\0';
        t.value = p;
        dbCursor<Test> cursor(dbCursorForUpdate);
        cursor.select();
        cursor.removeAll();
        dbQuery q;
        q = "key = ", buf;
        for (int i = 0; i < nIterations; i++) {             
            if (i > N_KEYS) { 
                sprintf(buf, "KEY-%08d", i - N_KEYS);
                int rc = cursor.select(q);
                assert(rc == 1);
                cursor.remove();
            }
            sprintf(buf, "KEY-%08d", i);
            t.key = buf;
            insert(t);
            db.commit();
            if (i % 100 == 0) { 
                printf("%d records\r", i);
                fflush(stdout);
            }
        }
        db.close();
        printf("\nDone\n");
    } else { 
        fprintf(stderr, "Failed to open database\n");
    }
    return 0;
}