// ************************************************************
void
testCreateTableAndInsert (void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2}
};
int numInserts = 9, i;
Record *r;
RID *rids;
Schema *schema;
testName = "test creating a new table and inserting tuples";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
//printf("\n Testing create table");
TEST_CHECK(createTable("test_table_r.txt",schema));
//printf("\n Testing open Table");
TEST_CHECK(openTable(table, "test_table_r.txt"));
printf("\n Opened ");
// insert rows into table
for(i = 0; i < numInserts; i++)
{
printf("\n Inserting");
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_r.txt"));
printf("\n Opened successsfully");
// randomly retrieve records from the table and compare to inserted ones
for(i = 0; i < 1000; i++)
{
int pos = rand() % numInserts;
RID rid = rids[pos];
printf("\n getting records");
TEST_CHECK(getRecord(table, rid, r));
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, inserts[pos]), r, schema, "compare records");
}
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r.txt"));
TEST_CHECK(shutdownRecordManager());
free(rids);
free(table);
TEST_DONE();
}
RC testTombstone(void)
{
RM_TableData *table = (RM_TableData *)malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{ 1, "aaaa", 3 },
{ 2, "bbbb", 2 },
{ 3, "cccc", 1 },
{ 4, "dddd", 3 },
{ 5, "eeee", 5 },
{ 6, "ffff", 1 },
{ 7, "gggg", 3 },
{ 8, "hhhh", 3 },
{ 9, "iiii", 2 },
{ 10, "jjjj", 5 },
};
int numInserts = 10, numUpdates = 3, numDeletes = 5, numFinal = 5, i;
Record *r;
RID *rids;
Schema *schema;
testName = "test creating a new table and insert,update,delete tuples";
schema = testSchema();
rids = (RID *)malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r", schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for (i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table, r));
rids[i] = r->id;
}
TEST_CHECK(deleteRecord(table, rids[9]));
RID id;
id.page = rids[9].page;
id.slot = rids[10].slot;
int isTombstone = checkIfTombstoneEncountered(table, rids[9]);
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
free(table);
if (isTombstone == 1) {
return RC_OK;
}
else {
return RC_NOT_A_TOMBSTONE;
}
}
void testUserInterface()
{
int attrNum=0;
int keys[]={};
printf("Enter the number of attributes: \n");
scanf("%d",&attrNum);
int attrTypes[attrNum];
int attrSize[attrNum];
DataType dt[attrNum];
char *attrName[attrNum];
int numRec;
int j;
for(j=0 ; j< attrNum ; j++){
printf("Enter the attribute name for %d attribute :",j+1);
attrName[j]=malloc(100);
scanf("%s",attrName[j]);
printf("Enter the type of attribute for %d attribute \n int : 0 \n string : 1 \n Float : 2 \n Boolean : 3 \n",j+1);
scanf("%d",&attrTypes[j]);
if(attrTypes[j]==0)
dt[j] = DT_INT;
else if(attrTypes[j]==1)
dt[j] = DT_STRING;
else if(attrTypes[j]==2)
dt[j] = DT_FLOAT;
else if(attrTypes[j]==3)
dt[j] = DT_BOOL;
if(attrTypes[j]!= 1){
attrSize[j] = 0;
}else{
printf("Enter the size of the string attribute \n");
scanf("%d",&attrSize[j]);
}
}
Schema *schema;
schema = createSchema(attrNum, attrName, dt, attrSize, 0 ,keys);
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2}
};
int numInserts = 9, i;
Record *r;
RID *rids;
testName = "testing user interface";
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r",schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_r"));
// randomly retrieve records from the table and compare to inserted ones with (int,string,int) format.
if(attrNum <= 3)
{
for(i = 0; i < 1000; i++)
{
int pos = rand() % numInserts;
RID rid = rids[pos];
TEST_CHECK(getRecord(table, rid, r));
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, inserts[pos]), r, schema, "compare records");
}
}
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
free(rids);
free(table);
TEST_DONE();
}
void testScansTwo (void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
bool foundScan[] = {
FALSE,
FALSE,
FALSE,
FALSE,
FALSE,
FALSE,
FALSE,
FALSE,
FALSE,
FALSE
};
int numInserts = 10, i;
Record *r;
RID *rids;
Schema *schema;
RM_ScanHandle *sc = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
Expr *sel, *left, *right, *first, *se;
int rc;
testName = "test creating a new table and inserting tuples";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r",schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_r"));
// Select 1 record with INT in condition a=2.
MAKE_CONS(left, stringToValue("i2"));
MAKE_ATTRREF(right, 0);
MAKE_BINOP_EXPR(sel, left, right, OP_COMP_EQUAL);
createRecord(&r, schema);
TEST_CHECK(startScan(table, sc, sel));
while((rc = next(sc, r)) == RC_OK)
{
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, inserts[1]), r, schema, "compare records");
}
if (rc != RC_NO_TUPLES)
TEST_CHECK(rc);
TEST_CHECK(closeScan(sc));
// Select 1 record with STRING in condition b='ffff'.
MAKE_CONS(left, stringToValue("sffff"));
MAKE_ATTRREF(right, 1);
MAKE_BINOP_EXPR(sel, left, right, OP_COMP_EQUAL);
createRecord(&r, schema);
TEST_CHECK(startScan(table, sc, sel));
while((rc = next(sc, r)) == RC_OK)
{
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, inserts[5]), r, schema, "compare records");
serializeRecord(r, schema);
}
if (rc != RC_NO_TUPLES)
TEST_CHECK(rc);
TEST_CHECK(closeScan(sc));
// Select all records, with condition being false
MAKE_CONS(left, stringToValue("i4"));
MAKE_ATTRREF(right, 2);
MAKE_BINOP_EXPR(first, right, left, OP_COMP_SMALLER);
MAKE_UNOP_EXPR(se, first, OP_BOOL_NOT);
TEST_CHECK(startScan(table, sc, se));
while((rc = next(sc, r)) == RC_OK)
{
serializeRecord(r, schema);
for(i = 0; i < numInserts; i++)
{
if (memcmp(fromTestRecord(schema, inserts[i])->data,r->data,getRecordSize(schema)) == 0)
foundScan[i] = TRUE;
}
}
if (rc != RC_NO_TUPLES)
TEST_CHECK(rc);
TEST_CHECK(closeScan(sc));
ASSERT_TRUE(!foundScan[0], "not greater than four");
ASSERT_TRUE(foundScan[4], "greater than four");
ASSERT_TRUE(foundScan[9], "greater than four");
// clean up
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
freeRecord(r);
free(table);
free(sc);
freeExpr(sel);
TEST_DONE();
}
void testScans (void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
TestRecord scanOneResult[] = {
{3, "cccc", 1},
{6, "ffff", 1},
};
bool foundScan[] = {
FALSE,
FALSE
};
int numInserts = 10, scanSizeOne = 2, i;
Record *r;
RID *rids;
Schema *schema;
RM_ScanHandle *sc = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
Expr *sel, *left, *right;
int rc;
testName = "test creating a new table and inserting tuples";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r",schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_r"));
// run some scans
MAKE_CONS(left, stringToValue("i1"));
MAKE_ATTRREF(right, 2);
MAKE_BINOP_EXPR(sel, left, right, OP_COMP_EQUAL);
TEST_CHECK(startScan(table, sc, sel));
while((rc = next(sc, r)) == RC_OK)
{
for(i = 0; i < scanSizeOne; i++)
{
if (memcmp(fromTestRecord(schema, scanOneResult[i])->data,r->data,getRecordSize(schema)) == 0)
foundScan[i] = TRUE;
}
}
if (rc != RC_NO_TUPLES)
TEST_CHECK(rc);
TEST_CHECK(closeScan(sc));
for(i = 0; i < scanSizeOne; i++)
ASSERT_TRUE(foundScan[i], "check for scan result");
// clean up
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
free(table);
free(sc);
freeExpr(sel);
TEST_DONE();
}
void
testInsertManyRecords(void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
TestRecord realInserts[10000];
TestRecord updates[] = {
{3333, "iiii", 6}
};
int numInserts = 10000, i, numcheck=50;
int randomRec = 3333;
Record *r;
RID *rids;
Schema *schema;
testName = "test creating a new table and inserting 10000 records then updating record from rids[3333]";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_t",schema));
TEST_CHECK(openTable(table, "test_table_t"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
realInserts[i] = inserts[i%10];
realInserts[i].a = i;
r = fromTestRecord(schema, realInserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_t"));
// retrieve records from the table and compare to expected final stage
for(i = 0; i < numcheck; i++)
{
RID rid = rids[i];
TEST_CHECK(getRecord(table, rid, r));
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, realInserts[i]), r, schema, "compare records");
}
r = fromTestRecord(schema, updates[0]);
r->id = rids[randomRec];
TEST_CHECK(updateRecord(table,r));
TEST_CHECK(getRecord(table, rids[randomRec], r));
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, updates[0]), r, schema, "compare records");
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_t"));
TEST_CHECK(shutdownRecordManager());
freeRecord(r);
free(table);
TEST_DONE();
}
void
testUpdateTable (void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
TestRecord updates[] = {
{1, "iiii", 6},
{2, "iiii", 6},
{3, "iiii", 6}
};
int deletes[] = {
9,
6,
7,
8,
5
};
TestRecord finalR[] = {
{1, "iiii", 6},
{2, "iiii", 6},
{3, "iiii", 6},
{4, "dddd", 3},
{5, "eeee", 5},
};
int numInserts = 10, numUpdates = 3, numDeletes = 5, numFinal = 5, i;
Record *r;
RID *rids;
Schema *schema;
testName = "test creating a new table and insert,update,delete tuples";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r",schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
// delete rows from table
for(i = 0; i < numDeletes; i++)
{
TEST_CHECK(deleteRecord(table,rids[deletes[i]]));
}
// update rows into table
for(i = 0; i < numUpdates; i++)
{
r = fromTestRecord(schema, updates[i]);
r->id = rids[i];
TEST_CHECK(updateRecord(table,r));
}
TEST_CHECK(closeTable(table));
TEST_CHECK(openTable(table, "test_table_r"));
// retrieve records from the table and compare to expected final stage
for(i = 0; i < numFinal; i++)
{
RID rid = rids[i];
TEST_CHECK(getRecord(table, rid, r));
ASSERT_EQUALS_RECORDS(fromTestRecord(schema, finalR[i]), r, schema, "compare records");
}
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
free(table);
TEST_DONE();
}
void
testMultipleScans(void)
{
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
int numInserts = 10, i, scanOne=0, scanTwo=0;
Record *r;
RID *rids;
Schema *schema;
testName = "test running muliple scans ";
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
RM_ScanHandle *sc1 = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
RM_ScanHandle *sc2 = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
Expr *se1, *left, *right;
int rc,rc2;
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_r",schema));
TEST_CHECK(openTable(table, "test_table_r"));
// insert rows into table
for(i = 0; i < numInserts; i++)
{
r = fromTestRecord(schema, inserts[i]);
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
// Mix 2 scans with c=3 as condition
MAKE_CONS(left, stringToValue("i3"));
MAKE_ATTRREF(right, 2);
MAKE_BINOP_EXPR(se1, left, right, OP_COMP_EQUAL);
createRecord(&r, schema);
TEST_CHECK(startScan(table, sc1, se1));
TEST_CHECK(startScan(table, sc2, se1));
if ((rc2 = next(sc2, r)) == RC_OK)
scanTwo++;
i = 0;
while((rc = next(sc1, r)) == RC_OK)
{
scanOne++;
i++;
if (i % 3 == 0)
if ((rc2 = next(sc2, r)) == RC_OK)
scanTwo++;
}
while((rc2 = next(sc2, r)) == RC_OK)
scanTwo++;
ASSERT_TRUE(scanOne == scanTwo, "scans returned same number of tuples");
if (rc != RC_NO_TUPLES)
TEST_CHECK(rc);
TEST_CHECK(closeScan(sc1));
TEST_CHECK(closeScan(sc2));
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_r"));
TEST_CHECK(shutdownRecordManager());
free(rids);
free(table);
TEST_DONE();
}
/*
* AUTHOR: Nikhil
* DESC: Parse the sql stmts and execute the respective methods
*/
int main(int argc, char *argv[]){
char readline[ARRAY_LENGTH];
char temparray[ARRAY_LENGTH];
char *temp;
char *temp1;
char *tableName;
char *length;
int charLength;
int numattr=0,k,count=0,numTuples=0;
Schema *schema;
RID *rids;
char columns[ARRAY_LENGTH][ARRAY_LENGTH];
int i=0,j=0;
if(fork() == 0) {
execv("/usr/bin/clear", argv);
exit(1);
}
else wait(NULL);
printf("\n SQL parser\n");
while(1){
printf("\n >>");
fflush(stdin);
gets(readline);
memcpy(temparray,readline,sizeof(readline));
if(strstr(readline,"create table") || strstr(readline,"CREATE TABLE")){
parser(readline,argv);
tableName=strtok(argv[2],"(");
temp=strstr(temparray,"(");
temp++;
temp1=strtok(temp,",");
while(temp1!=NULL){
strcpy(columns[i],temp1);
temp1=strtok(NULL,",");
i++;
}
j=i;
numattr=j;
strcpy(columns[i-1],strtok(columns[i-1],")"));
char* colname[numattr];
char* datatype[numattr];
DataType dt[numattr];
int sizes[numattr];
int keyattr[numattr];
for(i=0;i<j;i++){
parser(columns[i],argv);
colname[i]=argv[0];
datatype[i]=argv[1];
if(strstr(Upper(datatype[i]),"VARCHAR")) {
dt[i]=DT_STRING;
length=strtok(datatype[i],"(");
length=strtok(NULL,"(");
charLength=strtok(length,")");
if(length==NULL){
printf("\nWrong Format");
break;
}
sizes[i]=atoi(charLength);
}
if(strstr(Upper(datatype[i]),"INT")){
dt[i]=DT_INT;
}
else if(strstr(Upper(datatype[i]),"FLOAT")){
dt[i]=DT_FLOAT;
}
else if(strstr(Upper(datatype[i]),"BOOL")){
dt[i]=DT_BOOL; }
else{
sizes[i]=0;
}
if(strstr(Upper(datatype[i]),"PRIMARYKEY")){
keyattr[i]=1;
}
else{
keyattr[i]=0;
}
}
for(k=0;k<sizeof(keyattr)/sizeof(keyattr[0]);k++){
if(keyattr[i]==1){
count++;
}
}
if(count>0){
printf("wrong format...cannot contain more than one primary key");
break;
}
schema=testSchema(numattr,colname,dt,sizes,keyattr);
initRecordManager(NULL);
createTable(Upper(tableName),schema);
printf("Table Created Successfully");
shutdownRecordManager();
}
if(strstr(Upper(readline),"INSERT INTO")){
parser(readline,argv);
tableName=strtok(argv[2],"(");
temp=strtok(temparray,"(");
temp=strtok(NULL,"(");
strcpy(columns[0],temp);
strcpy(columns[0],strtok(columns[0],")"));
char* values[i];
i=0;
values[0]=strtok(columns[0],",");
while(values[i]!=NULL){
i++;
values[i]=strtok(NULL,",");
}
numTuples=i;
rids = (RID *) malloc(sizeof(RID) * numTuples);
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
initRecordManager(NULL);
int fd = open(Upper(tableName), O_WRONLY);
if(fd<0){
printf("Database Doesn't Exist");
break;
}
openTable(table, tableName);
schema=table->schema;
Record *result;
Value *value;
createRecord(&result, schema);
for(i = 0; i < numTuples; i++)
{
if(schema->dataTypes[i]==DT_INT || schema->dataTypes[i]==DT_FLOAT || schema->dataTypes[i]==DT_BOOL){
MAKE_VALUE(value, schema->dataTypes[i], atoi(values[i]));
}
if(schema->dataTypes[i]==DT_STRING){
MAKE_STRING_VALUE(value, values[i]);
}
setAttr(result, schema, i, value);
freeVal(value);
}
insertRecord(table,result);
rids[i] = result->id;
printf("Inserted Successfully");
closeTable(table);
shutdownRecordManager();
}
if(strstr(Upper(readline),"UPDATE TABLE")){
parser(readline,argv);
tableName=strtok(argv[2],"(");
temp1=strtok(temparray,"(");
temp=strtok(NULL,"(");
strcpy(columns[0],temp);
strcpy(columns[0],strtok(columns[0],")"));
temp1=strtok(temp1,"=");
temp1=strtok(NULL,"=");
char* values[i];
i=0;
values[0]=strtok(columns[0],",");
while(values[i]!=NULL){
i++;
values[i]=strtok(NULL,",");
}
numTuples=i;
rids = (RID *) malloc(sizeof(RID) * numTuples);
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
initRecordManager(NULL);
int fd = open(Upper(tableName), O_WRONLY);
if(fd<0){
printf("Database Doesn't Exist");
break;
}
openTable(table, Upper(tableName));
schema=table->schema;
Record *result;
Value *value;
createRecord(&result, schema);
for(i = 0; i < numTuples; i++)
{
if(schema->dataTypes[i]==DT_INT || schema->dataTypes[i]==DT_FLOAT || schema->dataTypes[i]==DT_BOOL){
MAKE_VALUE(value, schema->dataTypes[i], atoi(values[i]));
}
if(schema->dataTypes[i]==DT_STRING){
MAKE_STRING_VALUE(value, values[i]);
}
setAttr(result, schema, i, value);
freeVal(value);
}
result->id=rids[i] ;
updateRecord(table,result);
printf("Updated Successfully");
closeTable(table);
shutdownRecordManager();
}
if(strstr(Upper(readline),"DELETE FROM")){
parser(readline,argv);
tableName=strtok(argv[2],"(");
temp=strtok(temparray,"(");
temp=strtok(NULL,"(");
strcpy(columns[0],temp);
strcpy(columns[0],strtok(columns[0],")"));
int r_id=atoi(columns[0]);
numTuples=i;
rids = (RID *) malloc(sizeof(RID) * numTuples);
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
initRecordManager(NULL);
int fd = open(Upper(tableName), O_WRONLY);
if(fd<0){
printf("Database Doesn't Exist");
break;
}
openTable(table, Upper(tableName));
schema=table->schema;
deleteRecord(table,rids[r_id]);
printf("Deleted Successfully");
closeTable(table);
shutdownRecordManager();
}
}
}
void
testCheckingPrimaryKeyWithSingleAttribute(void)
{
//just like test_assign3_1.c we insert records in the same way at first.
RM_TableData *table = (RM_TableData *) malloc(sizeof(RM_TableData));
TestRecord inserts[] = {
{1, "aaaa", 3},
{2, "bbbb", 2},
{3, "cccc", 1},
{4, "dddd", 3},
{5, "eeee", 5},
{6, "ffff", 1},
{7, "gggg", 3},
{8, "hhhh", 3},
{9, "iiii", 2},
{10, "jjjj", 5},
};
TestRecord realInserts[10000];
int numInserts = 100, i;
Record *r;
RID *rids;
Schema *schema;
testName = "test checking primary key with single attribute";
//this schema is exactly the same as test_assign3_1.c, this primary key only have 1 attibute.
schema = testSchema();
rids = (RID *) malloc(sizeof(RID) * numInserts);
TEST_CHECK(initRecordManager(NULL));
TEST_CHECK(createTable("test_table_t",schema));
TEST_CHECK(openTable(table, "test_table_t"));
for(i = 0; i <numInserts; i++)
{
realInserts[i] = inserts[i%10];
realInserts[i].a = i;
r = fromTestRecord(schema, realInserts[i]);
printf("inserting record {%d,\"%s\",%d}",realInserts[i].a,realInserts[i].b,realInserts[i].c);
//Extra Credit: Check primary key constraints.
//check whether there exists a tuple that have the same key attribute value with this new record.
if (CheckPrimaryKey(table, r)==RC_OK) {
printf(" Success!\n");
}
else {
printf(" Failed! A tuple with same attribute value already exists!\n");
}
TEST_CHECK(insertRecord(table,r));
rids[i] = r->id;
}
//here is the code to check distinct,we insert some records that are already exist in the file, if detected, the checking function will return a RC value differen from RC_OK.
TestRecord newInsert;
newInsert=inserts[0];
newInsert.a=0;
r=fromTestRecord(schema, newInsert);
printf("inserting record {%d,\"%s\",%d}",newInsert.a,newInsert.b,newInsert.c);
//Extra Credit: Check primary key constraints.
//check whether there exists a tuple that have the same key attribute value with this new record.
if (CheckPrimaryKey(table, r)==RC_OK) {
printf(" Success!\n");
}
else {
printf(" Failed! A tuple with same attribute value already exists!\n");
}
insertRecord(table,r);
TEST_CHECK(closeTable(table));
TEST_CHECK(deleteTable("test_table_t"));
TEST_CHECK(shutdownRecordManager());
freeRecord(r);
free(table);
TEST_DONE();
}
int
main (void)
{
initRecordManager(NULL);
int in;
int cl = 0;
while(cl == 0){
printf("What to do:\n1.Add new table\n2.insert\n3.update\n4.delete\n5.scan\n6.print table\n7.close\n");
in=0;
scanf("%d",&in);
RM_TableData *table;
Record *result,*r;
Value *value,*v;
int i,j;
switch(in)
{
case 0:
printf("invalid input\n");
break;
case 1:
printf("\nAdding new table:\n");
printf("Table name: ");
char tbn[50];
scanf("%s",tbn);
char* tbnp = (char *) malloc(sizeof(char)*sizeof(tbn));
memcpy(tbnp,tbn,sizeof(tbn));
printf("number of attributes: ");
int nat = 0;
scanf("%d",&nat);
printf("number of keys: ");
int nk = 0;
scanf("%d",&nk);
Schema *sch;
char **names = (char **) malloc(sizeof(char*) * nat);
DataType* dt = (DataType *) malloc(sizeof(DataType) * nat);
int* sizes = (int *) malloc(sizeof(int) * nat);
int* keys = (int *) malloc(sizeof(int) * nk);
int kptr = 0;
for(i=0;i<nat;i++){
printf("Attribute %d name: ", i+1);
char buf[100];
scanf("%s", buf);
char* tmp = (char *) malloc(sizeof(char)*sizeof(buf));
memcpy(tmp,buf,sizeof(buf));
names[i] = tmp;
printf("Attribute %d datatype: [I/S/F/B]: ", i+1);
char dtp;
scanf(" %c", &dtp);
//printf("\ndtp = %c", dtp);
switch(dtp)
{
case 'I':
dt[i] = DT_INT;
sizes[i] = 0;
break;
case 'S':
dt[i] = DT_STRING;
printf("Attribute %d length: ", i+1);
scanf("%d",&sizes[i]);
printf("length is: %d\n", sizes[i]);
break;
case 'F':
dt[i] = DT_FLOAT;
sizes[i] = 0;
break;
case 'B':
dt[i] = DT_BOOL;
sizes[i] = 0;
break;
}
printf("Is Attribute %d a key?: [Y/N] ", i+1);
char isk;
scanf(" %c", &isk);
if(isk == 'Y'){
keys[kptr] = i;
kptr++;
}
//printf("schema add: %s,%d,%c\n",names[i],sizes[i],isk);
}
sch = createSchema(nat,names,dt,sizes,nk,keys);
createTable(tbnp,sch);
break;
case 2:
printf("Insert record\n");
printf("table name: ");
char buf2[100];
scanf("%s", buf2);
char* tmp2 = (char *) malloc(sizeof(char)*sizeof(buf2));
memcpy(tmp2,buf2,sizeof(buf2));
table = (RM_TableData *) malloc(sizeof(RM_TableData));
openTable(table, tmp2);
createRecord(&result, table->schema);
for(i=0;i<table->schema->numAttr;i++){
if(table->schema->dataTypes[i] == DT_INT)
{
printf("intput int attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
int intt = 0;
scanf("%d",&intt);
//printf("\n%d is %d\n", i, intt);
MAKE_VALUE(value,DT_INT,intt);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_STRING)
{
printf("intput string attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
char instrbuf[100];
scanf("%s",instrbuf);
char* instr = (char *) malloc(sizeof(char)*sizeof(instrbuf));
memcpy(instr,instrbuf,sizeof(instrbuf));
MAKE_STRING_VALUE(value,instr);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_FLOAT)
{
printf("intput float attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
float infl;
scanf("%fl",&infl);
MAKE_VALUE(value,DT_FLOAT,infl);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_BOOL)
{
printf("intput boolean attr[%d/%d] value -- %s [T/F]: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
char inbl;
scanf(" %c",&inbl);
if(inbl == 'T'){
MAKE_VALUE(value,DT_BOOL,true);
}
else if (inbl == 'F'){
MAKE_VALUE(value,DT_BOOL,false);
}
setAttr(result, table->schema, i, value);
freeVal(value);
}
if(i==table->schema->numAttr-1){
printf("\n");
}
}
insertRecord(table,result);
break;
case 3:
printf("update record\n");
printf("table name: ");
char buf3[100];
scanf("%s", buf3);
char* tmp3 = (char *) malloc(sizeof(char)*sizeof(buf3));
memcpy(tmp3,buf3,sizeof(buf3));
table = (RM_TableData *) malloc(sizeof(RM_TableData));
openTable(table, tmp3);
createRecord(&result, table->schema);
r = (Record *) malloc(sizeof(Record));
RM_ScanHandle *sc = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
startScan(table, sc, NULL);
printf("Select record to update: \n");
j=0;
int rids[100][2];
while(next(sc, r) == RC_OK){
printf("(%d) ",j+1);
rids[j][0] = r->id.page;
rids[j][1] = r->id.slot;
for(i=0;i<table->schema->numAttr;i++){
if(table->schema->dataTypes[i] == DT_INT)
{
getAttr(r,table->schema,i,&v);
printf("%-15d",v->v.intV);
} else if(table->schema->dataTypes[i] == DT_STRING)
{
getAttr(r,table->schema,i,&v);
printf("%-15s",v->v.stringV);
} else if(table->schema->dataTypes[i] == DT_FLOAT)
{
getAttr(r,table->schema,i,&v);
printf("%-15fl",v->v.floatV);
} else if(table->schema->dataTypes[i] == DT_BOOL)
{
getAttr(r,table->schema,i,&v);
printf("%-15s",v->v.boolV ? "true" : "false");
}
}
printf("\n");
j++;
}
int sele;
scanf("%d", &sele);
sele--;
if(sele >= 0 && sele < j+1){
createRecord(&result, table->schema);
for(i=0;i<table->schema->numAttr;i++){
if(table->schema->dataTypes[i] == DT_INT)
{
printf("intput int attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
int intt = 0;
scanf("%d",&intt);
//printf("\n%d is %d\n", i, intt);
MAKE_VALUE(value,DT_INT,intt);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_STRING)
{
printf("intput string attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
char instrbuf[100];
scanf("%s",instrbuf);
char* instr = (char *) malloc(sizeof(char)*sizeof(instrbuf));
memcpy(instr,instrbuf,sizeof(instrbuf));
MAKE_STRING_VALUE(value,instr);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_FLOAT)
{
printf("intput float attr[%d/%d] value -- %s: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
float infl;
scanf("%fl",&infl);
MAKE_VALUE(value,DT_FLOAT,infl);
setAttr(result, table->schema, i, value);
freeVal(value);
} else if (table->schema->dataTypes[i] == DT_BOOL)
{
printf("intput boolean attr[%d/%d] value -- %s [T/F]: ", i+1,table->schema->numAttr,table->schema->attrNames[i]);
char inbl;
scanf(" %c",&inbl);
if(inbl == 'T'){
MAKE_VALUE(value,DT_BOOL,true);
}
else if (inbl == 'F'){
MAKE_VALUE(value,DT_BOOL,false);
}
setAttr(result, table->schema, i, value);
freeVal(value);
}
if(i==table->schema->numAttr-1){
printf("\n");
}
}
result->id.page = rids[sele][0];
result->id.slot = rids[sele][1];
updateRecord(table,result);
}
break;
case 4:
printf("delete record\n");
printf("table name: ");
char buf4[100];
scanf("%s", buf4);
char* tmp4 = (char *) malloc(sizeof(char)*sizeof(buf3));
memcpy(tmp4,buf4,sizeof(buf4));
table = (RM_TableData *) malloc(sizeof(RM_TableData));
openTable(table, tmp4);
createRecord(&result, table->schema);
r = (Record *) malloc(sizeof(Record));
RM_ScanHandle *sc2 = (RM_ScanHandle *) malloc(sizeof(RM_ScanHandle));
startScan(table, sc2, NULL);
printf("Select record to delete: \n");
j=0;
int rids2[100][2];
while(next(sc2, r) == RC_OK){
printf("(%d) ",j+1);
rids2[j][0] = r->id.page;
rids2[j][1] = r->id.slot;
for(i=0;i<table->schema->numAttr;i++){
if(table->schema->dataTypes[i] == DT_INT)
{
getAttr(r,table->schema,i,&v);
printf("%-15d",v->v.intV);
} else if(table->schema->dataTypes[i] == DT_STRING)
{
getAttr(r,table->schema,i,&v);
printf("%-15s",v->v.stringV);
} else if(table->schema->dataTypes[i] == DT_FLOAT)
{
getAttr(r,table->schema,i,&v);
printf("%-15fl",v->v.floatV);
} else if(table->schema->dataTypes[i] == DT_BOOL)
{
getAttr(r,table->schema,i,&v);
printf("%-15s",v->v.boolV ? "true" : "false");
}
}
printf("\n");
j++;
}
int sele2;
scanf("%d", &sele2);
sele2--;
if(sele2 >= 0 && sele2 < j+1){
RID rid;
rid.page = rids[sele2][0];
rid.slot = rids[sele2][1];
deleteRecord(table,rid);
}
break;
case 5:
printf("scan\n");
break;
case 6:
printf("table name: ");
char buf[100];
scanf("%s", buf);
char* tmp = (char *) malloc(sizeof(char)*sizeof(buf));
memcpy(tmp,buf,sizeof(buf));
printtable(buf);
break;
case 7:
cl = 1;
break;
default:
printf("invalid input\n");
break;
}
}
}