pkg/common/utils/doris/storage_conf.go

// Licensed to the Apache Software Foundation (ASF) 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 doris

import (
	"strings"
)

// ResolveStorageRootPath transforms a string of storage paths into a slice of StorageRootPathInfo.
func ResolveStorageRootPath(configPath string) []string {
	var res []string

	if configPath == "" {
		return res
	}

	// Separate multiple paths by ';'
	configPathSplit := strings.Split(configPath, ";")

	// Remove empty elements
	for _, c := range configPathSplit {
		if path := parseSinglePath(c); path != "" {
			res = append(res, path)
		}
	}

	return res
}

// Resolving a single storage path
func parseSinglePath(pathConfig string) string {
	if pathConfig == "" {
		return ""
	}
	path := strings.Split(strings.Split(pathConfig, ".")[0], ",")[0]
	path = strings.TrimSpace(path)
	if strings.HasSuffix(path, "/") {
		path = path[:len(path)-1]
	}
	return path
}

// GetNameOfEachPath is used to parse a set of paths to obtain unique and concise names for each path.
// If the paths are repeated, the returned names may also be repeated.
// And the order of each name in the array is consistent with the input paths.
// For example:
//
//	["/path1"] >> ["path1"]
//	["/opt/doris/path1"] >> ["path1"]
//	["/path1", "/path2"] >> ["path1", "path2"]
//	["/home/disk1/doris", "/home/disk2/doris"] >> ["doris", "disk2-doris"]
//	["/home/doris/disk1", "/home/doris/disk2"] >> ["disk1", "disk2"]
//	["/home/disk1/doris", "/home/disk1/doris", "/home/disk2/doris"] >> ["disk1-doris", "disk1-doris", "disk2-doris"]
func GetNameOfEachPath(paths []string) []string {
	namePath := map[string]string{}
	pathName := map[string]string{}
	for _, path := range paths {
		//use unix path separator.
		sp := strings.Split(path, "/")
		name := ""
		for i := 1; i <= len(sp); i++ {
			if sp[len(sp)-i] == "" {
				continue
			}

			if name == "" {
				name = sp[len(sp)-i]
			} else {
				name = sp[len(sp)-i] + "-" + name
			}

			if _, ok := namePath[name]; !ok {
				break
			}
		}

		namePath[name] = path
		pathName[path] = name
	}
	res := make([]string, len(paths))
	for k := range paths {
		res[k] = strings.ReplaceAll(pathName[paths[k]], "_", "-")
	}
	return res
}