testleak.cpp

FastDb是高效的内存数据库系统

//-< 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>

USE_FASTDB_NAMESPACE

#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);

class Record { 
  public:
    db_int8 count;

    TYPE_DESCRIPTOR((FIELD(count)));
};

REGISTER(Record);

int main(int argc, char* argv[])
{
    int i;
    dbDatabase db;
    int nIterations = N_ITERATIONS;
    if (argc > 1) { 
        nIterations = atoi(argv[1]);
    }
    if (db.open("testleak")) {
        Record rec;
        rec.count = 0;
        db.insert(rec);
        dbCursor<Record> uc(dbCursorForUpdate);  
        for (i = 0; i < nIterations; i++) {             
            uc.select();
            assert(uc->count == i);
            uc->count += 1;
            uc.update();
            db.commit();
            if (i % 100 == 0) { 
                printf("%d records\r", i);
                fflush(stdout);
            }
        }
        printf("\nUpdate test passed\n");
        Test t;
        char buf[64];
        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 (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);
            }
        }
        printf("\nRemove test passed\n");
        db.close();
        printf("\nDone\n");
    } else { 
        fprintf(stderr, "Failed to open database\n");
    }
    return 0;
}