func()

in dao/feature_view_redis_dao.go [51:95]

• 38 lines of code
• 10 McCabe index (conditional complexity)

func (d *FeatureViewRedisDao) GetFeatures(keys []interface{}, selectFields []string) ([]map[string]interface{}, error) {
	var pkeys []string
	for _, key := range keys {
		pkey := utils.ToString(key, "")
		pkeys = append(pkeys, fmt.Sprintf("%s%s", d.redisPrefix, pkey))
	}

	ctx := context.Background()
	values, err := d.redisClinet.MGet(ctx, pkeys...).Result()
	if err != nil {
		return nil, err
	}

	result := make([]map[string]interface{}, 0, len(keys))
	for i, value := range values {
		if value == nil {
			result = append(result, map[string]interface{}{d.primaryKeyField: keys[i]})
		} else {
			if str, ok := value.(string); ok {
				m := make(map[string]interface{}, len(selectFields)+1)
				m[d.primaryKeyField] = keys[i]
				strs := strings.Split(str, d.redisDelimiter)
				for _, field := range selectFields {
					if index, ok := d.fieldIndexMap[field]; ok {
						switch d.fieldTypeMap[field] {
						case constants.FS_DOUBLE, constants.FS_FLOAT:
							m[field] = utils.ToFloat(strs[index], 0)
						case constants.FS_INT32, constants.FS_INT64:
							m[field] = utils.ToInt(strs[index], 0)
						default:
							m[field] = strs[index]
						}

					}
				}

				result = append(result, m)
			}

		}

	}

	return result, nil
}