From 20bd5bbb0950f1adf827875695deb1d8d6dd5309 Mon Sep 17 00:00:00 2001 From: Michael Quigley Date: Wed, 15 Mar 2023 15:16:48 -0400 Subject: [PATCH] detail cache (#270) --- controller/metrics2/cache.go | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 controller/metrics2/cache.go diff --git a/controller/metrics2/cache.go b/controller/metrics2/cache.go new file mode 100644 index 00000000..7dc8b047 --- /dev/null +++ b/controller/metrics2/cache.go @@ -0,0 +1,33 @@ +package metrics2 + +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 +}