QRunIO/qqq
2
0
Fork 0
mirror of https://github.com/Kingsrook/qqq.git synced 2025-07-18 13:10:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implemented delete

Browse Source
This commit is contained in:
Darin Kelkhoff
2021-11-30 18:10:22 -06:00
parent 01142017a3
commit c0ad05d119
3 changed files with 110 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/main/java/com/kingsrook/qqq/frontend/picocli/QPicoCliImplementation.java
View File

@ -5,11 +5,14 @@ import java.io.File;
import java.io.IOException;
import java.io.PrintStream;
import java.io.PrintWriter;
import java.io.Serializable;
import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import com.kingsrook.qqq.backend.core.actions.DeleteAction;
import com.kingsrook.qqq.backend.core.actions.InsertAction;
import com.kingsrook.qqq.backend.core.actions.MetaDataAction;
import com.kingsrook.qqq.backend.core.actions.QueryAction;
@ -19,6 +22,8 @@ import com.kingsrook.qqq.backend.core.adapters.JsonToQFieldMappingAdapter;
import com.kingsrook.qqq.backend.core.adapters.JsonToQRecordAdapter;
import com.kingsrook.qqq.backend.core.exceptions.QException;
import com.kingsrook.qqq.backend.core.model.actions.AbstractQFieldMapping;
import com.kingsrook.qqq.backend.core.model.actions.DeleteRequest;
import com.kingsrook.qqq.backend.core.model.actions.DeleteResult;
import com.kingsrook.qqq.backend.core.model.actions.InsertRequest;
import com.kingsrook.qqq.backend.core.model.actions.InsertResult;
import com.kingsrook.qqq.backend.core.model.actions.MetaDataRequest;
@ -140,6 +145,7 @@ public class QPicoCliImplementation
tableCommand.addSubcommand("meta-data", defineMetaDataCommand(table));
tableCommand.addSubcommand("query", defineQueryCommand(table));
tableCommand.addSubcommand("insert", defineInsertCommand(table));
tableCommand.addSubcommand("delete", defineDeleteCommand(table));
});
CommandLine commandLine = new CommandLine(topCommandSpec);
@ -230,7 +236,6 @@ public class QPicoCliImplementation
/*******************************************************************************
**
*******************************************************************************/
@SuppressWarnings("checkstyle:Indentation")
private CommandSpec defineInsertCommand(QTableMetaData table)
{
CommandSpec insertCommand = CommandSpec.create();
@ -254,18 +259,7 @@ public class QPicoCliImplementation
for(QFieldMetaData field : table.getFields().values())
{
insertCommand.addOption(OptionSpec.builder("--field-" + field.getName())
.type(
switch(field.getType())
{
case STRING, TEXT, HTML, PASSWORD -> String.class;
case INTEGER -> Integer.class;
case DECIMAL -> BigDecimal.class;
case DATE -> LocalDate.class;
// case TIME -> LocalTime.class;
case DATE_TIME -> LocalDateTime.class;
default -> throw new IllegalStateException("Unsupported field type: " + field.getType());
}
)
.type(getClassForField(field))
.build());
}
return insertCommand;
@ -273,6 +267,43 @@ public class QPicoCliImplementation
/*******************************************************************************
**
*******************************************************************************/
private CommandSpec defineDeleteCommand(QTableMetaData table)
{
CommandSpec deleteCommand = CommandSpec.create();
QFieldMetaData primaryKeyField = table.getField(table.getPrimaryKeyField());
deleteCommand.addOption(OptionSpec.builder("--primaryKey")
.type(String.class) // todo - mmm, better as picocli's "compound" thing, w/ the actual pkey's type?
.build());
return deleteCommand;
}
/*******************************************************************************
**
*******************************************************************************/
@SuppressWarnings("checkstyle:Indentation")
private Class<?> getClassForField(QFieldMetaData field)
{
return switch(field.getType())
{
case STRING, TEXT, HTML, PASSWORD -> String.class;
case INTEGER -> Integer.class;
case DECIMAL -> BigDecimal.class;
case DATE -> LocalDate.class;
// case TIME -> LocalTime.class;
case DATE_TIME -> LocalDateTime.class;
default -> throw new IllegalStateException("Unsupported field type: " + field.getType());
};
}
/*******************************************************************************
**
*******************************************************************************/
@ -316,6 +347,10 @@ public class QPicoCliImplementation
{
return runTableInsert(commandLine, tableName, subParseResult);
}
case "delete":
{
return runTableDelete(commandLine, tableName, subParseResult);
}
default:
{
commandLine.getErr().println("Unknown command: " + subParseResult.commandSpec().name());
@ -467,6 +502,31 @@ public class QPicoCliImplementation
}
/*******************************************************************************
**
*******************************************************************************/
private int runTableDelete(CommandLine commandLine, String tableName, ParseResult subParseResult) throws QException
{
DeleteRequest deleteRequest = new DeleteRequest(qInstance);
deleteRequest.setTableName(tableName);
QTableMetaData table = qInstance.getTable(tableName);
/////////////////////////////////////////////
// get the pKeys that the user specified //
/////////////////////////////////////////////
List<Serializable> primaryKeyList = null;
String primaryKeyOption = subParseResult.matchedOptionValue("--primaryKey", "");
String[] primaryKeyValues = primaryKeyOption.split(",");
deleteRequest.setPrimaryKeys(Arrays.asList(primaryKeyValues));
DeleteAction deleteAction = new DeleteAction();
DeleteResult deleteResult = deleteAction.execute(deleteRequest);
commandLine.getOut().println(JsonUtils.toPrettyJson(deleteResult));
return commandLine.getCommandSpec().exitCodeOnSuccess();
}
/*******************************************************************************
**

25
src/test/java/com/kingsrook/qqq/frontend/picocli/QPicoCliImplementationTest.java
View File

@ -160,6 +160,31 @@ class QPicoCliImplementationTest
/*******************************************************************************
**
*******************************************************************************/
@Test
public void test_tableDelete() throws Exception
{
TestOutput testOutput = testCli("person", "delete", "--primaryKey", "2,4");
JSONObject deleteResult = JsonUtils.toJSONObject(testOutput.getOutput());
assertNotNull(deleteResult);
assertEquals(2, deleteResult.getJSONArray("records").length());
assertEquals(2, deleteResult.getJSONArray("records").getJSONObject(0).getInt("primaryKey"));
assertEquals(4, deleteResult.getJSONArray("records").getJSONObject(1).getInt("primaryKey"));
TestUtils.runTestSql("SELECT id FROM person", (rs -> {
int rowsFound = 0;
while(rs.next())
{
rowsFound++;
assertTrue(rs.getInt(1) == 1 || rs.getInt(1) == 3 || rs.getInt(1) == 5);
}
assertEquals(3, rowsFound);
}));
}
/*******************************************************************************
**
*******************************************************************************/

12
src/test/java/com/kingsrook/qqq/frontend/picocli/TestUtils.java
View File

@ -42,6 +42,18 @@ public class TestUtils
/*******************************************************************************
**
*******************************************************************************/
public static void runTestSql(String sql, QueryManager.ResultSetProcessor resultSetProcessor) throws Exception
{
ConnectionManager connectionManager = new ConnectionManager();
Connection connection = connectionManager.getConnection(new RDBSMBackendMetaData(defineBackend()));
QueryManager.executeStatement(connection, sql, resultSetProcessor);
}
/*******************************************************************************
**
*******************************************************************************/