siyuan/kernel/av/relation.go

package av

import (
	"os"
	"path/filepath"
	"sync"

	"github.com/88250/gulu"
	"github.com/siyuan-note/filelock"
	"github.com/siyuan-note/logging"
	"github.com/siyuan-note/siyuan/kernel/util"
	"github.com/vmihailenco/msgpack/v5"
)

var (
	attributeViewRelationsLock = sync.Mutex{}
)

func RemoveAvRel(avID, destAvID string) {
	attributeViewRelationsLock.Lock()
	defer attributeViewRelationsLock.Unlock()

	relations := filepath.Join(util.DataDir, "storage", "av", "relations.msgpack")
	if !filelock.IsExist(relations) {
		return
	}

	data, err := filelock.ReadFile(relations)
	if nil != err {
		logging.LogErrorf("read attribute view relations failed: %s", err)
		return
	}

	avRels := map[string][]string{}
	if err = msgpack.Unmarshal(data, &avRels); nil != err {
		logging.LogErrorf("unmarshal attribute view relations failed: %s", err)
		return
	}

	destAvIDs := avRels[avID]
	if nil == destAvIDs {
		return
	}

	var newAvIDs []string
	for _, v := range destAvIDs {
		if v != destAvID {
			newAvIDs = append(newAvIDs, v)
		}
	}
	avRels[avID] = newAvIDs

	data, err = msgpack.Marshal(avRels)
	if nil != err {
		logging.LogErrorf("marshal attribute view relations failed: %s", err)
		return
	}
	if err = filelock.WriteFile(relations, data); nil != err {
		logging.LogErrorf("write attribute view relations failed: %s", err)
		return
	}
}

func UpsertAvRel(avID, destAvID string) {
	attributeViewRelationsLock.Lock()
	defer attributeViewRelationsLock.Unlock()

	avRelations := map[string][]string{}
	relations := filepath.Join(util.DataDir, "storage", "av", "relations.msgpack")
	if !filelock.IsExist(relations) {
		if err := os.MkdirAll(filepath.Dir(relations), 0755); nil != err {
			logging.LogErrorf("create attribute view dir failed: %s", err)
			return
		}
	} else {
		data, err := filelock.ReadFile(relations)
		if nil != err {
			logging.LogErrorf("read attribute view relations failed: %s", err)
			return
		}

		if err = msgpack.Unmarshal(data, &avRelations); nil != err {
			logging.LogErrorf("unmarshal attribute view relations failed: %s", err)
			return
		}
	}

	destAvIDs := avRelations[avID]
	destAvIDs = append(destAvIDs, destAvID)
	destAvIDs = gulu.Str.RemoveDuplicatedElem(destAvIDs)
	avRelations[avID] = destAvIDs

	data, err := msgpack.Marshal(avRelations)
	if nil != err {
		logging.LogErrorf("marshal attribute view relations failed: %s", err)
		return
	}
	if err = filelock.WriteFile(relations, data); nil != err {
		logging.LogErrorf("write attribute view relations failed: %s", err)
		return
	}
}