golib/infra/ii/common.go

package ii

import (
	"context"
	"encoding/xml"
	"os"

	"golib/features/mo"
	"golib/gio"
)

const (
	DefaultConfigSuffix = ".xml"
)

// LoadItems 从 path 中读取并解析 XML 配置
func LoadItems(path string) (Items, error) {
	name, err := gio.ReadDir(path, DefaultConfigSuffix)
	if err != nil {
		return nil, err
	}
	items := make(map[Name]ItemInfo)
	for i := 0; i < len(name); i++ {
		var itemInfo ItemInfo
		itemInfo, err = ReadFile(name[i])
		if err != nil {
			return nil, err
		}
		items[itemInfo.Name] = itemInfo
	}
	return items, nil
}

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

func ReadFrom(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
}

// 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))
	}

	if len(needAdd) == 0 {
		return nil
	}
	_, err = operator.CreateMany(ctx, needAdd)
	return err
}

func SetItemsUnique(items Items, client *mo.Client) error {
	for _, item := range items {
		if err := SetUnique(item, client); err != nil {
			return err
		}
	}
	return nil
}