mirror of https://github.com/milvus-io/milvus.git
75 lines
4.0 KiB
Go
75 lines
4.0 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package column
|
|
|
|
import "github.com/cockroachdb/errors"
|
|
|
|
var (
|
|
// scalars
|
|
NewNullableColumnBool NullableColumnCreateFunc[bool, *ColumnBool] = NewNullableColumnCreator(NewColumnBool).New
|
|
NewNullableColumnInt8 NullableColumnCreateFunc[int8, *ColumnInt8] = NewNullableColumnCreator(NewColumnInt8).New
|
|
NewNullableColumnInt16 NullableColumnCreateFunc[int16, *ColumnInt16] = NewNullableColumnCreator(NewColumnInt16).New
|
|
NewNullableColumnInt32 NullableColumnCreateFunc[int32, *ColumnInt32] = NewNullableColumnCreator(NewColumnInt32).New
|
|
NewNullableColumnInt64 NullableColumnCreateFunc[int64, *ColumnInt64] = NewNullableColumnCreator(NewColumnInt64).New
|
|
NewNullableColumnVarChar NullableColumnCreateFunc[string, *ColumnVarChar] = NewNullableColumnCreator(NewColumnVarChar).New
|
|
NewNullableColumnString NullableColumnCreateFunc[string, *ColumnString] = NewNullableColumnCreator(NewColumnString).New
|
|
NewNullableColumnFloat NullableColumnCreateFunc[float32, *ColumnFloat] = NewNullableColumnCreator(NewColumnFloat).New
|
|
NewNullableColumnDouble NullableColumnCreateFunc[float64, *ColumnDouble] = NewNullableColumnCreator(NewColumnDouble).New
|
|
NewNullableColumnJSONBytes NullableColumnCreateFunc[[]byte, *ColumnJSONBytes] = NewNullableColumnCreator(NewColumnJSONBytes).New
|
|
// array
|
|
NewNullableColumnBoolArray NullableColumnCreateFunc[[]bool, *ColumnBoolArray] = NewNullableColumnCreator(NewColumnBoolArray).New
|
|
NewNullableColumnInt8Array NullableColumnCreateFunc[[]int8, *ColumnInt8Array] = NewNullableColumnCreator(NewColumnInt8Array).New
|
|
NewNullableColumnInt16Array NullableColumnCreateFunc[[]int16, *ColumnInt16Array] = NewNullableColumnCreator(NewColumnInt16Array).New
|
|
NewNullableColumnInt32Array NullableColumnCreateFunc[[]int32, *ColumnInt32Array] = NewNullableColumnCreator(NewColumnInt32Array).New
|
|
NewNullableColumnInt64Array NullableColumnCreateFunc[[]int64, *ColumnInt64Array] = NewNullableColumnCreator(NewColumnInt64Array).New
|
|
NewNullableColumnVarCharArray NullableColumnCreateFunc[[]string, *ColumnVarCharArray] = NewNullableColumnCreator(NewColumnVarCharArray).New
|
|
NewNullableColumnFloatArray NullableColumnCreateFunc[[]float32, *ColumnFloatArray] = NewNullableColumnCreator(NewColumnFloatArray).New
|
|
NewNullableColumnDoubleArray NullableColumnCreateFunc[[]float64, *ColumnDoubleArray] = NewNullableColumnCreator(NewColumnDoubleArray).New
|
|
)
|
|
|
|
type NullableColumnCreateFunc[T any, Col interface {
|
|
Column
|
|
Data() []T
|
|
}] func(name string, values []T, validData []bool) (Col, error)
|
|
|
|
type NullableColumnCreator[col interface {
|
|
Column
|
|
withValidData([]bool)
|
|
}, T any] struct {
|
|
base func(name string, values []T) col
|
|
}
|
|
|
|
func (c NullableColumnCreator[col, T]) New(name string, values []T, validData []bool) (col, error) {
|
|
var result col
|
|
if len(values) != len(validData) {
|
|
return result, errors.New("values & validData slice has different length")
|
|
}
|
|
result = c.base(name, values)
|
|
result.withValidData(validData)
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func NewNullableColumnCreator[col interface {
|
|
Column
|
|
withValidData([]bool)
|
|
}, T any](base func(name string, values []T) col) NullableColumnCreator[col, T] {
|
|
return NullableColumnCreator[col, T]{
|
|
base: base,
|
|
}
|
|
}
|