support dump for vecf32 and vecf64 (#21281)

https://github.com/matrixorigin/MO-Cloud/issues/4765

Approved by: @qingxinhome, @heni02

Author: YANGGMM

pkg/frontend/export.go

@@ -593,7 +593,16 @@ func exportDataFromResultSetToCSVFile(oq *ExportConfig) error {
 			if err != nil {
 				return err
 			}
-			value = addEscapeToString(value.([]byte))
+			if _, ok := value.([]byte); ok {
+				value = addEscapeToString(value.([]byte))
+			} else if arr, ok := value.([]float32); ok {
+				// this is for T_array_float32 type
+				value = []byte(types.ArrayToString[float32](arr))
+			} else if arr, ok := value.([]float64); ok {
+				// this is for T_array_float64 type
+				value = []byte(types.ArrayToString[float64](arr))
+			}
+
 			if err = formatOutputString(oq, value.([]byte), symbol[i], closeby, true, buffer); err != nil {
 				return err
 			}

pkg/frontend/export_test.go

@@ -207,12 +207,26 @@ func Test_exportDataToCSVFile(t *testing.T) {
 			mrs: &MysqlResultSet{},
 		}
 
-		var col = make([]MysqlColumn, 13)
+		var col = make([]MysqlColumn, 15)
 		col[5].flag = 0
 		col[6].flag = 1 << 5
-		var colType = []defines.MysqlType{defines.MYSQL_TYPE_YEAR, defines.MYSQL_TYPE_YEAR, defines.MYSQL_TYPE_YEAR, defines.MYSQL_TYPE_SHORT, defines.MYSQL_TYPE_DOUBLE,
-			defines.MYSQL_TYPE_LONGLONG, defines.MYSQL_TYPE_LONGLONG, defines.MYSQL_TYPE_VARCHAR, defines.MYSQL_TYPE_DATE, defines.MYSQL_TYPE_DATETIME,
-			defines.MYSQL_TYPE_BOOL, defines.MYSQL_TYPE_DECIMAL, defines.MYSQL_TYPE_JSON}
+		var colType = []defines.MysqlType{
+			defines.MYSQL_TYPE_YEAR,
+			defines.MYSQL_TYPE_YEAR,
+			defines.MYSQL_TYPE_YEAR,
+			defines.MYSQL_TYPE_SHORT,
+			defines.MYSQL_TYPE_DOUBLE,
+			defines.MYSQL_TYPE_LONGLONG,
+			defines.MYSQL_TYPE_LONGLONG,
+			defines.MYSQL_TYPE_VARCHAR,
+			defines.MYSQL_TYPE_DATE,
+			defines.MYSQL_TYPE_DATETIME,
+			defines.MYSQL_TYPE_BOOL,
+			defines.MYSQL_TYPE_DECIMAL,
+			defines.MYSQL_TYPE_JSON,
+			defines.MYSQL_TYPE_VARCHAR,
+			defines.MYSQL_TYPE_VARCHAR,
+		}
 		for i := 0; i < len(col); i++ {
 			col[i].SetColumnType(colType[i])
 			ep.mrs.AddColumn(&col[i])
@@ -230,6 +244,8 @@ func Test_exportDataToCSVFile(t *testing.T) {
 		data[10] = true
 		data[11] = 1.2
 		data[12], _ = types.ParseStringToByteJson(`{"a":1}`)
+		data[13] = []float32{8.560689, 6.790359, 821.9778}
+		data[13] = []float64{8.560689, 6.790359, 821.9778}
 
 		ep.mrs.AddRow(data)
 

test/distributed/cases/array/array.result

@@ -12,6 +12,14 @@ insert into vec_table values(1, "[1,2,3]", "[4,5,6]");
 select * from vec_table;
 a    b    c
 1    [1, 2, 3]    [4, 5, 6]
+set save_query_result = on;
+/* save_result */ select * from vec_table;
+a    b    c
+1    [1, 2, 3]    [4, 5, 6]
+select * from result_scan(last_query_id()) as u;
+a    b    c
+1    [1, 2, 3]    [4, 5, 6]
+set save_query_result = off;
 select b+b from vec_table;
 b + b
 [2, 4, 6]
@@ -192,6 +200,20 @@ a    b    c
 2    [0.8166459, 0.66616553, 0.4886152]    null
 3    [0.1726299, 3.2908857, 30.433094]    [0.45052445, 2.1984527, 9.579752, 123.48039, 4635.894]
 4    [8.560689, 6.790359, 821.9778]    [0.46323407, 23.498016, 563.923, 56.076736, 8732.958]
+set save_query_result = on;
+/* save_result */ select * from t7;
+a    b    c
+1    null    null
+2    [0.8166459, 0.66616553, 0.4886152]    null
+3    [0.1726299, 3.2908857, 30.433094]    [0.45052445, 2.1984527, 9.579752, 123.48039, 4635.894]
+4    [8.560689, 6.790359, 821.9778]    [0.46323407, 23.498016, 563.923, 56.076736, 8732.958]
+select * from result_scan(last_query_id()) as u;
+a    b    c
+1    null    null
+2    [0.8166459, 0.66616553, 0.4886152]    null
+3    [0.1726299, 3.2908857, 30.433094]    [0.45052445, 2.1984527, 9.579752, 123.48039, 4635.894]
+4    [8.560689, 6.790359, 821.9778]    [0.46323407, 23.498016, 563.923, 56.076736, 8732.958]
+set save_query_result = off;
 select a, b + b, c + c from t7;
 a    b + b    c + c
 1    null    null

test/distributed/cases/array/array.sql

@@ -10,6 +10,11 @@ desc vec_table;
 insert into vec_table values(1, "[1,2,3]", "[4,5,6]");
 select * from vec_table;
 
+set save_query_result = on;
+/* save_result */ select * from vec_table;
+select * from result_scan(last_query_id()) as u;
+set save_query_result = off;
+
 -- binary operators
 select b+b from vec_table;
 select b-b from vec_table;
@@ -106,6 +111,10 @@ insert into t7 values(2, "[0.8166459, 0.66616553, 0.4886152]", NULL);
 insert into t7 values(3, "[0.1726299, 3.2908857, 30.433094]","[0.45052445, 2.1984527, 9.579752, 123.48039, 4635.894]");
 insert into t7 values(4, "[8.560689, 6.790359, 821.9778]", "[0.46323407, 23.498016, 563.923, 56.076736, 8732.958]");
 select * from t7;
+set save_query_result = on;
+/* save_result */ select * from t7;
+select * from result_scan(last_query_id()) as u;
+set save_query_result = off;
 select a, b + b, c + c from t7;
 select a, b * b, c * c from t7;
 select l2_norm(b), l2_norm(c) from t7;