diff --git a/src/Plugins/BotSharp.Plugin.ExcelHandler/Services/MySqlService.cs b/src/Plugins/BotSharp.Plugin.ExcelHandler/Services/MySqlService.cs index f85b5ace..af08194c 100644 --- a/src/Plugins/BotSharp.Plugin.ExcelHandler/Services/MySqlService.cs +++ b/src/Plugins/BotSharp.Plugin.ExcelHandler/Services/MySqlService.cs @@ -90,6 +90,7 @@ namespace BotSharp.Plugin.ExcelHandler.Services { var numTables = workbook.NumberOfSheets; var commandList = new List(); + var state = _services.GetRequiredService(); for (int sheetIdx = 0; sheetIdx < numTables; sheetIdx++) { @@ -109,15 +110,17 @@ namespace BotSharp.Plugin.ExcelHandler.Services commandList.Add(commandResult); continue; } - var (isInsertSuccess, insertMessage) = SqlInsertDataFn(sheet); string table = $"{_database}.{_tableName}"; + state.SetState("tmp_table", table); + + var (isInsertSuccess, insertMessage) = SqlInsertDataFn(sheet); string exampleData = GetInsertExample(table); commandResult = new SqlContextOut { isSuccessful = isInsertSuccess, - Message = $"{insertMessage}\r\nExample Data: {exampleData}", + Message = $"{insertMessage}\r\nExample Data: {exampleData}. \r\n The remaining data contains different values. ", FileName = _currentFileName }; commandList.Add(commandResult); diff --git a/src/Plugins/BotSharp.Plugin.SqlDriver/Functions/ExecuteQueryFn.cs b/src/Plugins/BotSharp.Plugin.SqlDriver/Functions/ExecuteQueryFn.cs index 99b40a65..d1b21bdc 100644 --- a/src/Plugins/BotSharp.Plugin.SqlDriver/Functions/ExecuteQueryFn.cs +++ b/src/Plugins/BotSharp.Plugin.SqlDriver/Functions/ExecuteQueryFn.cs @@ -41,6 +41,12 @@ public class ExecuteQueryFn : IFunctionCallback _ => throw new NotImplementedException($"Database type {settings.DatabaseType} is not supported.") }; + if (refinedArgs.SqlStatements.Length == 1 && refinedArgs.SqlStatements[0].StartsWith("DROP TABLE")) + { + message.Content = "Drop table successfully"; + return true; + } + if (results.Count() == 0) { message.Content = "No record found";