eth-tracker/internal/cache/xmap.go

package cache

import (
	"context"

	"github.com/puzpuzpuz/xsync/v3"
)

type mapCache struct {
	xmap *xsync.MapOf[string, bool]
}

func NewMapCache() Cache {
	return &mapCache{
		xmap: xsync.NewMapOf[string, bool](),
	}
}

func (c *mapCache) Add(_ context.Context, key string) error {
	c.xmap.Store(key, true)
	return nil
}

func (c *mapCache) Remove(_ context.Context, key string) error {
	c.xmap.Delete(key)
	return nil
}

func (c *mapCache) Exists(_ context.Context, key string) (bool, error) {
	_, ok := c.xmap.Load(key)
	if ok {
		return true, nil

	}

	return false, nil
}

func (c *mapCache) ExistsNetwork(_ context.Context, token string, addresses ...string) (bool, error) {
	_, ok := c.xmap.Load(token)
	if !ok {
		return false, nil
	}

	for _, v := range addresses {
		_, ok := c.xmap.Load(v)
		if ok {
			return true, nil
		}
	}
	return false, nil
}

func (c *mapCache) Size(_ context.Context) (int64, error) {
	return int64(c.xmap.Size()), nil
}