golib/infra/ii/common.go

package ii

import (
	"context"
	"encoding/xml"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"golib/features/mo"
)

const (
	DefaultConfigSuffix = ".xml"
)

// ReadDir 从 path 中读取并解析 XML 配置
func ReadDir(path string) ([]ItemInfo, error) {
	name, err := readDir(path)
	if err != nil {
		return nil, err
	}
	item := make([]ItemInfo, len(name))
	for i := 0; i < len(name); i++ {
		var itemInfo ItemInfo
		itemInfo, err = UnmarshalFile(name[i])
		if err != nil {
			return nil, fmt.Errorf("unmarshal file: %s, filename: %s", err, name[i])
		}
		item[i] = itemInfo
	}
	return item, nil
}

func Unmarshal(b []byte) (ItemInfo, error) {
	var itemInfo ItemInfo
	if err := xml.Unmarshal(b, &itemInfo); err != nil {
		return ItemInfo{}, err
	}
	if err := itemInfo.init(); err != nil {
		return ItemInfo{}, err
	}
	return itemInfo, nil
}

// UnmarshalFile 解析 name 至 ItemInfo
// 如果需要 FieldInfo.Unique 生效, 需要调用 SetUnique
func UnmarshalFile(name string) (ItemInfo, error) {
	content, err := os.ReadFile(name)
	if err != nil {
		return ItemInfo{}, err
	}
	return Unmarshal(content)
}

// SetUnique 设置唯一键
// 注意: 为了降低初始化 XML 配置文件时的耦合度, 因此只能通过此方法设置唯一键. 如果通过软件实现唯一值, 那么将无法保证原子性
// 实现方法: 取出已存在的 index, 然后与 ItemInfo 中的 uniqueMap 比较:
// 删除 uniqueMap 中不存在的字段, 跳过 uniqueMap 中已存在的字段, 然后设置 uniqueMap 存在但 index 中不存在的字段为索引
func SetUnique(info ItemInfo, client *mo.Client) error {
	ctx, cancel := context.WithTimeout(context.Background(), mo.DefaultTimout)
	defer cancel()

	operator := info.Open(client).Indexes()

	cursor, err := operator.List(ctx)
	if err != nil {
		return err
	}

	indexMap, err := mo.ResolveIndexName(cursor)
	if err != nil {
		return err
	}

	for idx := range indexMap {
		if _, ok := info.uniqueMap[idx]; ok {
			continue
		}
		// 删除 info 中不存在的索引
		if _, err = operator.DropOne(ctx, idx); err != nil {
			return err
		}
	}

	var needAdd []mo.IndexModel

	for key := range info.uniqueMap {
		if _, ok := indexMap[mo.IndexName(key)]; ok {
			continue
		}
		needAdd = append(needAdd, mo.NewIndex(key))
	}

	_, err = operator.CreateMany(ctx, needAdd)
	return err
}

func readDir(path string) ([]string, error) {
	file, err := os.ReadDir(filepath.Join(path))
	if err != nil {
		return nil, err
	}
	fileList := make([]string, 0, 1024)
	for i := 0; i < len(file); i++ {
		if !strings.HasSuffix(file[i].Name(), DefaultConfigSuffix) {
			continue
		}
		if file[i].IsDir() {
			var fs []string
			fs, err = readDir(filepath.Join(path, file[i].Name()))
			if err != nil {
				return nil, err
			}
			fileList = append(fileList, fs...)
			continue
		}
		fileList = append(fileList, filepath.Join(path, file[i].Name()))
	}
	return fileList, nil
}