如何在存储过程MYSQL中将“SELECT *”的结果转换为数组?

时间:2021-08-01 22:03:42

标签: mysql stored-procedures

实际上,我试图在删除记录之前获取一些记录并将其存储到 log_record 中。 这是我的桌子。但问题是我不知道如何将“SELECT *”的结果转换为数组,所以我无法在存储过程(SP)中循环它以根据列名和列值将其插入到记录中。

CREATE TABLE IF NOT EXISTS account (
  id_account int(11) NOT NULL AUTO_INCREMENT,
  account_name VARCHAR(30) NOT NULL,
  account_location tinyint(4) NOT NULL,
  create_at timestamp NOT NULL DEFAULT current_timestamp(),
  PRIMARY KEY (id_account)
)ENGINE=InnoDB;

INSERT INTO account(account_name, account_location) VALUES('Me', 110),
('You', 110),
('Them', 2);

CREATE TABLE IF NOT EXISTS log_record (
  id_log INT AUTO_INCREMENT PRIMARY KEY,
  type tinyint not null,
  table_affected VARCHAR(10),
  column_name VARCHAR(30),
  new_value VARCHAR(255),
  create_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
  constraint C_TYPE_LOG check(type IN (0,1))
)ENGINE=INNODB;

DROP PROCEDURE IF EXISTS deleteProcedure;
CREATE PROCEDURE deleteProcedure(IN KeyValue INT, IN TableName VARCHAR(30), IN PrimaryKey VARCHAR(30), OUT messegeResult VARCHAR(30))
    BEGIN
        DECLARE ColumnName VARCHAR(30);
        DECLARE ColumnValue VARCHAR(30);
        DECLARE result TINYINT(1);
        DECLARE EXIT HANDLER FOR SQLEXCEPTION 
        BEGIN
            ROLLBACK;
            RESIGNAL;
            SET messegeResult = null;
        END;
        -- # THIS FOR TURN INTO ARRAY AREA
        -- # But i don't know how to turn result of "SELECT *" INTO array of value and array of column name
        SET @ArrayColumnValue = '';
        SET @ArrayColumnName = '';
        
        SELECT * FROM TableName WHERE PrimaryKey = KeyValue;
        
        START TRANSACTION;
            -- # LOOP AREA FOR INSERT 
            WHILE (LOCATE(',', @ArrayColumnValue) > 0)
            DO
                SET @ColumnName = ELT(1, @ArrayColumnName);
                SET @ArrayColumnName= SUBSTRING(@ArrayColumnName, LOCATE(',',@ArrayColumnName) + 1);
                SET @ColumnValue = ELT(1, @ArrayColumnValue);
                SET @ArrayColumnValue= SUBSTRING(@ArrayColumnValue, LOCATE(',',@ArrayColumnValue) + 1);

                INSERT INTO log_record(type, table_affected, column_name, old_value) 
                VALUES('DELETE', TableName, @ColumnName ,@ColumnValue);
            END WHILE;
            
            SELECT ROW_COUNT() INTO result;
            -- # CHECK IS SUCCESS BEING INSERT
            IF (@result = 0) THEN
                ROLLBACK;
                SET messegeResult = 'FAILED';
            ELSE
                DELETE FROM TableName WHERE PrimaryKey = KeyValue;
                SET messegeResult = 'SUCCESS';
            END IF;
        COMMIT;
    END

还是一开始就做错了?就像使用错误的方法来解决愚蠢的事情而不是应该轻松完成的事情。 我是这件事的新手,如果您知道我使用错误的方式,请告诉我使用其他方式。 TT

1 个答案:

答案 0 :(得分:0)

我正在 maria DB 上对其进行测试,并且工作正常。

CREATE PROCEDURE DeleteProcedure(IN LogTableName VARCHAR(100), IN TableName VARCHAR(100), in InputValues varchar(1000))
ProcLabel:BEGIN
    -- Local Variable
    DECLARE log_column_names, column_names, current_column_name, current_input_value, column_names_repeat, input_values_found, input_value_params VARCHAR(1000);
    DECLARE table_key_name, this_table_name, table_key_data_type VARCHAR(100);
    DECLARE column_start_pos, column_comma_pos, column_loop_count, row_start_pos, row_comma_pos, row_loop_count, total_column_loop_count, input_values_found_length SMALLINT;
    DECLARE column_end_loop, row_end_loop, table_key_data_type_char TINYINT;
    -- Declare Condition
    DECLARE TableNotFound CONDITION for 1146 ;
    
    -- Exit Handler
    DECLARE EXIT HANDLER FOR TableNotFound
    BEGIN
        SELECT CONCAT('Table ', @this_table_name ,' Are Not Found') MESSAGE_TEXT;
    END;
    
    -- Set Array Column
    SET @this_table_name = TableName;
    SET @SQL = "SELECT GROUP_CONCAT(column_name SEPARATOR ', ') FROM information_schema.columns WHERE table_name = ? INTO @column_names";
    PREPARE stmt FROM @SQL;
    EXECUTE stmt USING @this_table_name;
    
    SET @this_table_name = LogTableName;
    SET @SQL = "SELECT GROUP_CONCAT(column_name SEPARATOR ', ') FROM information_schema.columns WHERE table_name = ? AND column_key != 'PRI' AND column_name != 'create_at' INTO @log_column_names";
    PREPARE stmt FROM @SQL;
    EXECUTE stmt USING @this_table_name;
    
    SET @SQL = "SELECT column_name, data_type FROM information_schema.columns WHERE table_name = ? AND column_key = 'PRI' INTO @table_key_name, @table_key_data_type";
    PREPARE stmt FROM @SQL;
    EXECUTE stmt USING TableName;
    
    SELECT IF(LOWER(SUBSTRING(@table_key_data_type, -3, 3) != 'int'), TRUE, FALSE) INTO @table_key_data_type_char;
   
    SET @SQL = CONCAT("INSERT INTO ", LogTableName ," (", @log_column_names ,") VALUES");
    
    -- Set Default Value Before Loop
    SET @SQLVAL = '';
    SET @row_start_pos = 1;
    SET @row_comma_pos = LOCATE(',', InputValues);
    SET @row_loop_count = 0;
    SET @total_column_loop_count = 0;
    SET @input_values_found = '';
    SET @input_value_params = InputValues;
    
    InputValueRepeat:REPEAT
        IF @row_comma_pos > 0 THEN
            SET @current_input_value = SUBSTRING(InputValues, @row_start_pos, @row_comma_pos - @row_start_pos);
            SET @row_end_loop = 0;
        ELSE
            SET @current_input_value = SUBSTRING(InputValues, @row_start_pos);
            SET @row_end_loop = 1;
        END IF;
        
        SET @SQLC = CONCAT("SELECT COUNT(1) FROM ", TableName ," WHERE ", @table_key_name ," = ", @current_input_value ," INTO @record_found");
        PREPARE stmtC FROM @SQLC;
        EXECUTE stmtC;
        DEALLOCATE PREPARE stmtC;
        
        IF @record_found = 1 AND @row_loop_count > 0 AND @column_loop_count IS NOT NULL THEN
            set @SQLVAL = CONCAT(@SQLVAL,",");
        END IF;
        
        IF @record_found = 1 THEN
            SET @column_start_pos = 1;
            SET @column_comma_pos = LOCATE(',', @column_names);
            SET @column_names_repeat = @column_names;
            SET @column_loop_count = 0;
            
            ColumnNameRepeat:REPEAT
                IF @column_comma_pos > 0 THEN
                    SET @current_column_name = SUBSTRING(@column_names_repeat, @column_start_pos, @column_comma_pos - @column_start_pos);
                    SET @column_end_loop = 0;
                ELSE
                    SET @current_column_name = SUBSTRING(@column_names_repeat, @column_start_pos);
                    SET @column_end_loop = 1;
                END IF;
    
                SET @SQLSV = CONCAT("SELECT ", @current_column_name ," FROM ", TableName ," WHERE ", @table_key_name ," = ? INTO @current_column_value");
                PREPARE stmtSV FROM @SQLSV;
                EXECUTE stmtSV USING @current_input_value;
                DEALLOCATE PREPARE stmtSV;
            
                SET @SQLVAL = CONCAT(@SQLVAL, "('", CONCAT_WS("','", 'D', TableName, @current_column_name, @current_column_value, @current_input_value) ,"')");

                IF @column_end_loop = 0 THEN
                    SET @column_names_repeat = substring(@column_names_repeat, @column_comma_pos + 1);
                    SET @column_comma_pos = LOCATE(',', @column_names_repeat);
                    set @SQLVAL = CONCAT(@SQLVAL,",");
                END IF;
            
                SET @column_loop_count = @column_loop_count + 1;
            
                UNTIL @column_end_loop = 1

            END REPEAT;
            SET @total_column_loop_count = @total_column_loop_count + @column_loop_count;
            IF @input_values_found != '' THEN
                IF (@table_key_data_type_char = TRUE) THEN
                    SET @input_values_found = CONCAT_WS("','", @input_values_found, @current_input_value);
                ELSE 
                    SET @input_values_found = CONCAT_WS(",", @input_values_found, @current_input_value);
                END IF;
            ELSE 
                SET @input_values_found = @current_input_value;
            END IF;
        END IF;
        
        IF @row_end_loop = 0 THEN
            SET InputValues = substring(InputValues, @row_comma_pos + 1);
            SET @row_comma_pos = LOCATE(',', InputValues);
        ELSE 
            set @SQLVAL = CONCAT(@SQLVAL,";");
        END IF;
        
        SET @row_loop_count = @row_loop_count + 1;
                       
        UNTIL @row_end_loop = 1

    END REPEAT;
    
    IF (table_key_data_type_char = TRUE) THEN
        SET @input_values_found = CONCAT("'", @input_values_found ,"'");   
    END IF;
    
    SELECT CHAR_LENGTH(CONCAT(@input_values_found)) INTO @input_values_found_length;
    if @input_values_found_length < 3 AND @table_key_data_type_char = TRUE OR @table_key_data_type_char = FALSE AND @input_values_found_length < 1 THEN
        SELECT CONCAT("Can't find input value : [", @input_value_params ,"] on table ", TableName) MESSAGE_TEXT; 
        DEALLOCATE PREPARE stmt;
        LEAVE ProcLabel;
    END IF;
    
    START TRANSACTION;
    
    SET @SQL = CONCAT(@SQL, @SQLVAL);
    PREPARE stmt FROM @SQL;
    EXECUTE stmt;
    
    SELECT ROW_COUNT() into @affected_row;
    
    IF (@affected_row != @total_column_loop_count) THEN
        SELECT CONCAT("Failed to insert data expected row : ", @column_loop_count ,", affected row : ", @affected_row) MESSAGE_TEXT;
        ROLLBACK;
        DEALLOCATE PREPARE stmt;
        LEAVE ProcLabel;
    END IF;
    
    SET @affected_row = 0;
    
    SET @SQL = CONCAT("DELETE FROM ", TableName ," WHERE ", @table_key_name ," IN (", @input_values_found ,")");
    PREPARE stmt FROM @SQL;
    EXECUTE stmt;
    
    SELECT ROW_COUNT() INTO @affected_row;
    
    IF (@affected_row = 0) THEN
        SELECT CONCAT("Failed to delete ", TableName ," on record key ", @input_values_found) MESSAGE_TEXT;
        ROLLBACK;
        DEALLOCATE PREPARE stmt;
        LEAVE ProcLabel;
    END IF;
    
    SELECT CONCAT("Success ", @affected_row ," record on ", TableName ," was deleted") MESSAGE_TEXT;
    COMMIT;
    DEALLOCATE PREPARE stmt;
END

我们需要像 Call DeleteProcedure('[to table]','[from table]','[array of keys to delete]'); => Call DeleteProcedure('log_history','orders','1,3,4');

一样调用这个 SP

所以我们可以选择尽可能多的使用逗号分隔符,例如第三个参数上的“1,2,3”。但是我还没有在超过一百个键上尝试过,因为第三个参数数据类型是 varchar(1000)。 我希望在不久的将来有人可以解决这个问题。