mirror of
https://github.com/openziti/zrok.git
synced 2024-11-23 00:23:48 +01:00
36 lines
631 B
Go
36 lines
631 B
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
|
|
}
|