types: throw error when input exceeds the range of float32 in vector (#58841)

ref #58379

Author: EricZequan

pkg/expression/integration_test/integration_test.go

@@ -961,9 +961,12 @@ func TestVectorArithmatic(t *testing.T) {
 	tk.MustQueryToErr(`SELECT VEC_FROM_TEXT('[1]') + 2;`)
 	tk.MustQueryToErr(`SELECT VEC_FROM_TEXT('[1]') + '2';`)
 
+	// Input outside the float32 value range will result in an error.
 	tk.MustQueryToErr(`SELECT VEC_FROM_TEXT('[3e38]') + '[3e38]';`)
 	tk.MustQuery(`SELECT VEC_FROM_TEXT('[1,2,3]') * '[4,5,6]';`).Check(testkit.Rows("[4,10,18]"))
 	tk.MustQueryToErr(`SELECT VEC_FROM_TEXT('[1e37]') * '[1e37]';`)
+	tk.MustQueryToErr("select VEC_L2_NORM('[1e39]') + 1")
+	tk.MustQueryToErr("select VEC_L2_NORM('[1e39]')*0 + 1")
 }
 
 func TestVectorFunctions(t *testing.T) {

pkg/types/vector.go

@@ -233,6 +233,11 @@ func ParseVectorFloat32(s string) (VectorFloat32, error) {
 			valueError = errors.Errorf("infinite value not allowed in vector")
 			return false
 		}
+		// Check if the value can be safely converted to float32
+		if v < -math.MaxFloat32 || v > math.MaxFloat32 {
+			valueError = errors.Errorf("value %v out of range for float32", v)
+			return false
+		}
 		values = append(values, float32(v))
 		return true
 	})

pkg/types/vector_test.go

@@ -101,6 +101,9 @@ func TestVectorParse(t *testing.T) {
 	require.False(t, v.IsZeroValue())
 	require.Equal(t, 1, v.Compare(types.ZeroVectorFloat32))
 	require.Equal(t, -1, types.ZeroVectorFloat32.Compare(v))
+
+	v, err = types.ParseVectorFloat32(`[-1e39, 1e39]`)
+	require.EqualError(t, err, "value -1e+39 out of range for float32")
 }
 
 func TestVectorDatum(t *testing.T) {