Method ExtendEnumCases

controller/mutators/etcd/enum_mutator.go:57–113

(namespace, tableName, columnName string, enumCases []string)

}

func (e *enumMutator) ExtendEnumCases(namespace, tableName, columnName string, enumCases []string) ([]int, error) {
	schema, err := e.schemaMutator.GetTable(namespace, tableName)
	if err != nil {
		return nil, err
	}
	columnID := -1
	for id, column := range schema.Columns {
		if column.Name == columnName && !column.Deleted && column.IsEnumColumn() {
			columnID = id
			break
		}
	}
	if columnID < 0 {
		return nil, metaCom.ErrColumnDoesNotExist
	}

	e.RLock()
	enumCache, exist := e.enumCacheMap[getCacheKey(namespace, tableName, schema.Incarnation, columnID)]
	if !exist {
		e.RUnlock()
		// fetch enum case from etcd and update cache
		return e.extendEnumCase(namespace, tableName, schema.Incarnation, columnID, 0, enumCases)
	}

	currentNodeID := enumCache.currentNodeID
	// check enum cache for existing enums
	enumIDs := make([]int, len(enumCases))
	newEnumCases := make([]string, 0)
	for index, enum := range enumCases {
		enumID, exist := enumCache.enumCases[enum]
		if !exist {
			enumIDs[index] = -1
			newEnumCases = append(newEnumCases, enum)
		} else {
			enumIDs[index] = enumID
		}
	}
	e.RUnlock()

	if len(newEnumCases) > 0 {
		// fetch enum cases from etcd and update cache
		missingIDs, err := e.extendEnumCase(namespace, tableName, schema.Incarnation, columnID, currentNodeID, newEnumCases)
		if err != nil {
			return nil, err
		}
		// missingID is ordered the same as newEnumCases
		// which is the same order in enumIDs
		i := 0
		for index, enumID := range enumIDs {
			if enumID < 0 {
				enumIDs[index] = missingIDs[i]
				i++
			}
		}
	}
	return enumIDs, nil
}