Skip to content

Commit 35c393d

Browse files
committed
fix typo and add INSERT statement generator to the routing
1 parent b586a28 commit 35c393d

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

src/include/statement_generator.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ class StatementGenerator {
105105
unique_ptr<DropStatement> GenerateDrop();
106106
unique_ptr<PrepareStatement> GeneratePrepare();
107107
unique_ptr<ExportStatement> GenerateExport();
108+
unique_ptr<InsertStatement> GenerateInsert();
108109

109110
unique_ptr<OrderModifier> GenerateOrderBy();
110111
unique_ptr<OrderModifier> GenerateOrderByAll();

src/statement_generator.cpp

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,9 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement() {
142142
if (RandomPercentage(20)) {
143143
return GenerateExport();
144144
}
145+
if (RandomPercentage(20)) {
146+
return GenerateInsert();
147+
}
145148
return GenerateStatement(StatementType::CREATE_STATEMENT);
146149
}
147150

@@ -171,7 +174,9 @@ unique_ptr<SQLStatement> StatementGenerator::GenerateStatement(StatementType typ
171174
case StatementType::DROP_STATEMENT:
172175
return GenerateDrop();
173176
case StatementType::EXPORT_STATEMENT:
174-
return GenerateDrop();
177+
return GenerateExport();
178+
case StatementType::INSERT_STATEMENT:
179+
return GenerateInsert();
175180
default:
176181
throw InternalException("Unsupported type");
177182
}
@@ -364,7 +369,7 @@ unique_ptr<ExportStatement> StatementGenerator::GenerateExport() {
364369
}
365370

366371
//===--------------------------------------------------------------------===//
367-
// Export Statement
372+
// Insert Statement
368373
//===--------------------------------------------------------------------===//
369374

370375
unique_ptr<InsertStatement> StatementGenerator::GenerateInsert() {

0 commit comments

Comments
 (0)