JCS-pub/internal/event/check_cache.go

package event

import (
	shell "github.com/ipfs/go-ipfs-api"
	"github.com/samber/lo"
	"gitlink.org.cn/cloudream/agent/internal/config"
	"gitlink.org.cn/cloudream/common/consts"
	evcst "gitlink.org.cn/cloudream/common/consts/event"
	"gitlink.org.cn/cloudream/common/utils/logger"
	"gitlink.org.cn/cloudream/db/model"
	agtevt "gitlink.org.cn/cloudream/rabbitmq/message/agent/event"
	scmsg "gitlink.org.cn/cloudream/rabbitmq/message/scanner"
	scevt "gitlink.org.cn/cloudream/rabbitmq/message/scanner/event"
)

type CheckCache struct {
	IsComplete bool
	Caches     []model.Cache
}

func NewCheckCache(isComplete bool, caches []model.Cache) *CheckCache {
	return &CheckCache{
		IsComplete: isComplete,
		Caches:     caches,
	}
}

func (t *CheckCache) TryMerge(other Event) bool {
	event, ok := other.(*CheckCache)
	if !ok {
		return false
	}

	if event.IsComplete {
		t.IsComplete = true
		t.Caches = event.Caches
		return true
	}

	if !t.IsComplete {
		t.Caches = append(t.Caches, event.Caches...)
		t.Caches = lo.UniqBy(t.Caches, func(ch model.Cache) string { return ch.HashValue })
		return true
	}

	return false
}

func (t *CheckCache) Execute(execCtx ExecuteContext) {
	filesMap, err := execCtx.Args.IPFS.GetPinnedFiles()
	if err != nil {
		logger.Warnf("get pinned files from ipfs failed, err: %s", err.Error())
		return
	}

	// TODO 根据锁定清单过滤被锁定的文件的记录
	if t.IsComplete {
		t.checkComplete(filesMap, execCtx)
	} else {
		t.checkIncrement(filesMap, execCtx)
	}
}

func (t *CheckCache) checkIncrement(filesMap map[string]shell.PinInfo, execCtx ExecuteContext) {
	for _, cache := range t.Caches {
		_, ok := filesMap[cache.HashValue]
		if ok {
			if cache.State == consts.CACHE_STATE_PINNED {
				// 不处理
			} else if cache.State == consts.CACHE_STATE_TEMP {
				execCtx.Args.IPFS.Unpin(cache.HashValue)
			}

			// 删除map中的记录，表示此记录已被检查过
			delete(filesMap, cache.HashValue)

		} else {
			if cache.State == consts.CACHE_STATE_PINNED {
				// 需要考虑此处是否是同步的过程
				execCtx.Args.IPFS.Pin(cache.HashValue)
			} else if cache.State == consts.CACHE_STATE_TEMP {
				// 不处理
			}
		}
	}

	// 增量情况下，不需要对filesMap中没检查的记录进行处理
}

func (t *CheckCache) checkComplete(filesMap map[string]shell.PinInfo, execCtx ExecuteContext) {
	for _, cache := range t.Caches {
		_, ok := filesMap[cache.HashValue]
		if ok {
			if cache.State == consts.CACHE_STATE_PINNED {
				// 不处理
			} else if cache.State == consts.CACHE_STATE_TEMP {
				execCtx.Args.IPFS.Unpin(cache.HashValue)
			}

			// 删除map中的记录，表示此记录已被检查过
			delete(filesMap, cache.HashValue)

		} else {
			if cache.State == consts.CACHE_STATE_PINNED {
				// 需要考虑此处是否是同步的过程
				execCtx.Args.IPFS.Pin(cache.HashValue)
			} else if cache.State == consts.CACHE_STATE_TEMP {
				// 不处理
			}
		}
	}

	var updateCacheOps []scevt.UpdateCacheEntry
	// map中剩下的数据是没有被遍历过，即Cache中没有记录的
	for hash, _ := range filesMap {
		updateCacheOps = append(updateCacheOps, scevt.NewUpdateCacheEntry(hash, evcst.UPDATE_CACHE_CREATE_TEMP))
	}
	execCtx.Args.Scanner.PostEvent(scmsg.NewPostEventBody(
		scevt.NewUpdateCache(config.Cfg().ID, updateCacheOps),
		execCtx.Option.IsEmergency,
		execCtx.Option.DontMerge,
	))
}

func init() {
	Register(func(val agtevt.CheckCache) Event { return NewCheckCache(val.IsComplete, val.Caches) })
}