Function ParseVector

pkg/util/vector/vector.go:28–66

ParseVector parses the Postgres string representation of a vector.

(input string)

// ParseVector parses the Postgres string representation of a vector.
func ParseVector(input string) (T, error) {
	input = strings.TrimSpace(input)
	if !strings.HasPrefix(input, "[") || !strings.HasSuffix(input, "]") {
		return T{}, pgerror.Newf(pgcode.InvalidTextRepresentation,
			"malformed vector literal: Vector contents must start with \"[\" and"+
				" end with \"]\"")
	}

	input = strings.TrimPrefix(input, "[")
	input = strings.TrimSuffix(input, "]")
	parts := strings.Split(input, ",")

	if len(parts) > MaxDim {
		return T{}, pgerror.Newf(pgcode.ProgramLimitExceeded, "vector cannot have more than %d dimensions", MaxDim)
	}

	vector := make([]float32, len(parts))
	for i, part := range parts {
		part = strings.TrimSpace(part)
		if part == "" {
			return T{}, pgerror.New(pgcode.InvalidTextRepresentation, "invalid input syntax for type vector: empty string")
		}

		val, err := strconv.ParseFloat(part, 32)
		if err != nil {
			return T{}, pgerror.Newf(pgcode.InvalidTextRepresentation, "invalid input syntax for type vector: %s", part)
		}

		if math.IsInf(val, 0) {
			return T{}, pgerror.New(pgcode.DataException, "infinite value not allowed in vector")
		}
		if math.IsNaN(val) {
			return T{}, pgerror.New(pgcode.DataException, "NaN not allowed in vector")
		}
		vector[i] = float32(val)
	}

	return vector, nil
}

// AsSet returns this vector a set of one vector.
func (v T) AsSet() Set {