feat:新增数组字段 --story=128495448 字段类型： 数组

Author: elizabevil

src/common/metadata/attribute_array.go

@@ -0,0 +1,102 @@
+/*
+ * Tencent is pleased to support the open source community by making 蓝鲸 available.
+ * Copyright (C) 2017-2018 THL A29 Limited, a Tencent company. All rights reserved.
+ * Licensed under the MIT License (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ * http://opensource.org/licenses/MIT
+ * 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 metadata
+
+import (
+	"configcenter/src/common/util"
+	"encoding/json"
+	"fmt"
+	"math"
+
+	"github.com/tidwall/gjson"
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+// ArrayOption len cap ,option is basic type 's option
+type ArrayOption[T any] struct {
+	Len    int `bson:"len" json:"len"`
+	Cap    int `bson:"cap" json:"cap"`
+	Option T   `bson:"option" json:"option"`
+}
+
+// Valid ArrayOption
+func (a *ArrayOption[T]) Valid() error {
+	if a.Len < 0 || a.Len > a.Cap {
+		return fmt.Errorf("invalid array option,len:%d cap:%d", a.Len, a.Cap)
+	}
+	return nil
+}
+
+// ParseArrayOption len cap ,option is basic type 's option
+func ParseArrayOption[T any](option any, handle func(v any) (T, error)) (ArrayOption[T], error) {
+	if option == nil {
+		return ArrayOption[T]{Len: math.MaxInt, Cap: math.MaxInt}, nil
+	}
+
+	var result ArrayOption[T]
+
+	optMap := map[string]interface{}{
+		"len": math.MaxInt,
+		"cap": math.MaxInt,
+	}
+	switch value := option.(type) {
+	case ArrayOption[T]:
+		return value, nil
+	case bson.M:
+		optMap = value
+	case map[string]interface{}:
+		optMap = value
+	default:
+		marshal, err := json.Marshal(option)
+		if err != nil {
+			return result, fmt.Errorf("invalid array option,type:%v,value:%v,err:%w", option, option, err)
+		}
+
+		lenItem := gjson.GetBytes(marshal, "len")
+		capItem := gjson.GetBytes(marshal, "cap")
+		if !lenItem.Exists() || !capItem.Exists() {
+			return result, fmt.Errorf("invalid array option,type:%v,value:%v,err: not exist len or cap", option, option)
+		}
+		optMap["len"] = lenItem.Int()
+		optMap["cap"] = capItem.Int()
+		optMap["option"] = gjson.GetBytes(marshal, "option").Value()
+	}
+
+	lenn, lenOk := optMap["len"]
+	capp, capOk := optMap["cap"]
+	if !lenOk || !capOk {
+		return result, fmt.Errorf("invalid array option,type:%v,value:%v,err: not exist len or cap", option, option)
+	}
+
+	lenOpt, err := util.GetIntByInterface(lenn)
+	if err != nil {
+		return result, err
+	}
+	result.Len = lenOpt
+	capOpt, err := util.GetIntByInterface(capp)
+	if err != nil {
+		return result, err
+	}
+	result.Cap = capOpt
+
+	var defaultOption T
+	result.Option = defaultOption
+	if handle != nil {
+		t, err := handle(optMap["option"])
+		if err != nil {
+			return ArrayOption[T]{}, err
+		}
+		result.Option = t
+	}
+	return result, result.Valid()
+}

src/common/metadata/attribute_array_bool.go

@@ -0,0 +1,161 @@
+/*
+ * Tencent is pleased to support the open source community by making 蓝鲸 available.
+ * Copyright (C) 2017-2018 THL A29 Limited, a Tencent company. All rights reserved.
+ * Licensed under the MIT License (the "License"); you may not use this file except
+ * in compliance with the License. You may obtain a copy of the License at
+ * http://opensource.org/licenses/MIT
+ * 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 metadata
+
+import (
+	"context"
+	"fmt"
+
+	"configcenter/src/common"
+	"configcenter/src/common/blog"
+	"configcenter/src/common/mapstr"
+	"configcenter/src/common/util"
+	"configcenter/src/common/valid/attribute/manager/register"
+)
+
+func init() {
+	// Register the arrayBool attribute type
+	register.Register(arrayBool{})
+}
+
+type arrayBool struct {
+}
+
+// Name returns the name of the arrayBool attribute.
+func (a arrayBool) Name() string {
+	return "array_bool"
+}
+
+// DisplayName returns the display name for user.
+func (a arrayBool) DisplayName() string {
+	return "布尔数组"
+}
+
+// RealType returns the db type of the arrayBool attribute.
+// Flattened array uses LongChar as storage type
+func (a arrayBool) RealType() string {
+	return common.FieldTypeLongChar
+}
+
+// Info returns the tips for user.
+func (a arrayBool) Info() string {
+	return "布尔值的扁平化数组字段，存储多个true/false值"
+}
+
+// Validate validates the arrayBool attribute value
+func (a arrayBool) Validate(ctx context.Context, objID string, propertyType string, required bool,
+	option, value interface{}) error {
+
+	rid := util.ExtractRequestIDFromContext(ctx)
+
+	if value == nil {
+		if required {
+			blog.Errorf("array_bool attribute %s.%s value is required but got nil, rid: %s",
+				objID, propertyType, rid)
+			return fmt.Errorf("array_bool attribute %s.%s value is required but got nil",
+				objID, propertyType)
+		}
+		return nil
+	}
+	opts, err := ParseArrayOption[any](option, nil)
+	if err != nil {
+		blog.Errorf("array_bool parse option failed: %v, rid: %s", err, rid)
+		return fmt.Errorf("array_bool invalid option: %v", err)
+	}
+
+	// Validate that value is a slice of any
+	boolArray, ok := util.ConvertAnyToSlice(value)
+	if !ok {
+		blog.Errorf("array_bool attribute %s.%s value must be []interface{}, got %T, rid: %s",
+			objID, propertyType, value, rid)
+		return fmt.Errorf("array_bool attribute %s.%s value must be []interface{}, got %T",
+			objID, propertyType, value)
+	}
+	if opts.Cap < len(boolArray) {
+		return fmt.Errorf("array_bool invalid cap %d, rid: %s", opts.Cap, rid)
+	}
+	// Validate each item in the array is a boolean
+	for i, item := range boolArray {
+		if _, ok := item.(bool); !ok {
+			blog.Errorf("array_bool attribute %s.%s array item [%d] type %T is not bool, rid: %s",
+				objID, propertyType, i, item, rid)
+			return fmt.Errorf("array_bool attribute %s.%s array item [%v] type %T is not bool",
+				objID, propertyType, item, item)
+		}
+	}
+
+	return nil
+}
+
+// FillLostValue fills the lost value with default value
+func (a arrayBool) FillLostValue(ctx context.Context, valData mapstr.MapStr, propertyId string,
+	defaultValue, option interface{}) error {
+
+	rid := util.ExtractRequestIDFromContext(ctx)
+
+	valData[propertyId] = nil
+	if defaultValue == nil {
+		return nil
+	}
+
+	// Validate default value
+	defaultArray, ok := util.ConvertAnyToSlice(defaultValue)
+	if !ok {
+		blog.Errorf("array_bool default value must be []interface{}, got %T, rid: %s", defaultValue, rid)
+		return fmt.Errorf("array_bool default value must be []interface{}, got %T", defaultValue)
+	}
+
+	// Validate each item in default array
+	for i, item := range defaultArray {
+		if _, ok := item.(bool); !ok {
+			blog.Errorf("array_bool default value array item [%d] type %T is not bool, rid: %s", i, item, rid)
+			return fmt.Errorf("array_bool default value array item [%d] type %T is not bool", i, item)
+		}
+	}
+
+	valData[propertyId] = defaultArray
+	return nil
+}
+
+// ValidateOption validates the option field
+func (a arrayBool) ValidateOption(ctx context.Context, option interface{}, defaultVal interface{}) error {
+
+	rid := util.ExtractRequestIDFromContext(ctx)
+
+	_, err := ParseArrayOption[any](option, nil)
+	if err != nil {
+		return err
+	}
+	if defaultVal == nil {
+		return nil
+	}
+
+	// Validate default value
+	defaultArray, ok := util.ConvertAnyToSlice(defaultVal)
+	if !ok {
+		blog.Errorf("array_bool default value must be []interface{}, got %T, rid: %s", defaultVal, rid)
+		return fmt.Errorf("array_bool default value must be []interface{}, got %T", defaultVal)
+	}
+
+	// Validate each item in default array
+	for i, item := range defaultArray {
+		if _, ok := item.(bool); !ok {
+			blog.Errorf("array_bool default value array item [%d] type %T is not bool, rid: %s", i, item, rid)
+			return fmt.Errorf("array_bool default value array item [%d] type %T is not bool", i, item)
+		}
+	}
+
+	return nil
+}
+
+var _ register.AttributeTypeI = &arrayBool{}