Add support for new VECTOR type

dbussink · dbussink · commit fcb28a6db68e · 2024-07-18T14:28:43.000+02:00
MySQL 9.0.0 added support for the VECTOR type. This adds basic support
so it can be handled at the protocol level.

Signed-off-by: Dirkjan Bussink &lt;d.bussink@gmail.com&gt;
diff --git a/AUTHORS b/AUTHORS
@@ -33,6 +33,7 @@ Daniel Montoya <dsmontoyam at gmail.com>
 Daniel Nichter <nil at codenode.com>
 Daniël van Eeden <git at myname.nl>
 Dave Protasowski <dprotaso at gmail.com>
+Dirkjan Bussink <d.bussink at gmail.com>
 DisposaBoy <disposaboy at dby.me>
 Egor Smolyakov <egorsmkv at gmail.com>
 Erwan Martin <hello at erwan.io>
diff --git a/const.go b/const.go
@@ -125,7 +125,10 @@ const (
 	fieldTypeBit
 )
 const (
-	fieldTypeJSON fieldType = iota + 0xf5
+	fieldTypeVector fieldType = iota + 0xf2
+	fieldTypeInvalid
+	fieldTypeBool
+	fieldTypeJSON
 	fieldTypeNewDecimal
 	fieldTypeEnum
 	fieldTypeSet
diff --git a/fields.go b/fields.go
@@ -112,6 +112,8 @@ func (mf *mysqlField) typeDatabaseName() string{
 		return "VARCHAR"
 	case fieldTypeYear:
 		return "YEAR"
+	case fieldTypeVector:
+		return "VECTOR"
 	default:
 		return ""
 	}
@@ -198,7 +200,7 @@ func (mf *mysqlField) scanType() reflect.Type{
 		return scanTypeNullFloat
 
 	case fieldTypeBit, fieldTypeTinyBLOB, fieldTypeMediumBLOB, fieldTypeLongBLOB,
-		fieldTypeBLOB, fieldTypeVarString, fieldTypeString, fieldTypeGeometry:
+		fieldTypeBLOB, fieldTypeVarString, fieldTypeString, fieldTypeGeometry, fieldTypeVector:
 		if mf.charSet == binaryCollationID{
 			return scanTypeBytes
 		}
diff --git a/packets.go b/packets.go
@@ -1329,7 +1329,8 @@ func (rows *binaryRows) readRow(dest []driver.Value) error{
 		case fieldTypeDecimal, fieldTypeNewDecimal, fieldTypeVarChar,
 			fieldTypeBit, fieldTypeEnum, fieldTypeSet, fieldTypeTinyBLOB,
 			fieldTypeMediumBLOB, fieldTypeLongBLOB, fieldTypeBLOB,
-			fieldTypeVarString, fieldTypeString, fieldTypeGeometry, fieldTypeJSON:
+			fieldTypeVarString, fieldTypeString, fieldTypeGeometry, fieldTypeJSON,
+			fieldTypeVector:
 			var isNull bool
 			var n int
 			dest[i], isNull, n, err = readLengthEncodedString(data[pos:])

-Original file line number
+Diff line change
 return"VARCHAR"
 casefieldTypeYear:
 return"YEAR"
 +casefieldTypeVector:
 +return"VECTOR"
 default:
 return""
+ }
 returnscanTypeNullFloat
 casefieldTypeBit, fieldTypeTinyBLOB, fieldTypeMediumBLOB, fieldTypeLongBLOB,
 -fieldTypeBLOB, fieldTypeVarString, fieldTypeString, fieldTypeGeometry:
 +fieldTypeBLOB, fieldTypeVarString, fieldTypeString, fieldTypeGeometry, fieldTypeVector:
 ifmf.charSet==binaryCollationID{
 returnscanTypeBytes
+ }