testleak.cpp

来自「FastDb是高效的内存数据库系统」· C++ 代码 · 共 82 行

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