zrok/controller/metrics/cache.go

package metrics

import (
	"github.com/openziti/zrok/controller/store"
)

type cache struct {
	str *store.Store
}

func newShareCache(str *store.Store) *cache {
	return &cache{str}
}

func (c *cache) addZrokDetail(u *Usage) error {
	tx, err := c.str.Begin()
	if err != nil {
		return err
	}
	defer func() { _ = tx.Rollback() }()

	shr, err := c.str.FindShareWithZIdAndDeleted(u.ZitiServiceId, tx)
	if err != nil {
		return err
	}
	u.ShareToken = shr.Token
	env, err := c.str.GetEnvironment(shr.EnvironmentId, tx)
	if err != nil {
		return err
	}
	u.EnvironmentId = int64(env.Id)
	u.AccountId = int64(*env.AccountId)

	return nil
}