IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

ServiceAccounts: Collect usage stat for service account forced expiry (#64833)

Browse Source 
collect usage stat for service account forced expiry
This commit is contained in:
Jo
2023-03-16 15:36:31 +00:00
committed by GitHub
parent 7784782512
commit 0e565a2e6c
5 changed files with 28 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/services/serviceaccounts/manager/stats.go
View File

@@ -42,13 +42,20 @@ func init() {
func (sa *ServiceAccountsService) getUsageMetrics(ctx context.Context) (map[string]interface{}, error) {
stats := map[string]interface{}{}
sqlStats, err := sa.store.GetUsageMetrics(ctx)
storeStats, err := sa.store.GetUsageMetrics(ctx)
if err != nil {
return nil, err
}
stats["stats.serviceaccounts.count"] = sqlStats.ServiceAccounts
stats["stats.serviceaccounts.tokens.count"] = sqlStats.Tokens
stats["stats.serviceaccounts.count"] = storeStats.ServiceAccounts
stats["stats.serviceaccounts.tokens.count"] = storeStats.Tokens
var forcedExpiryEnabled int64 = 0
if storeStats.ForcedExpiryEnabled {
forcedExpiryEnabled = 1
}
stats["stats.serviceaccounts.forced_expiry_enabled.count"] = forcedExpiryEnabled
var secretScanEnabled int64 = 0
if sa.secretScanEnabled {
@@ -57,8 +64,8 @@ func (sa *ServiceAccountsService) getUsageMetrics(ctx context.Context) (map[stri
stats["stats.serviceaccounts.secret_scan.enabled.count"] = secretScanEnabled
MStatTotalServiceAccountTokens.Set(float64(sqlStats.Tokens))
MStatTotalServiceAccounts.Set(float64(sqlStats.ServiceAccounts))
MStatTotalServiceAccountTokens.Set(float64(storeStats.Tokens))
MStatTotalServiceAccounts.Set(float64(storeStats.ServiceAccounts))
return stats, nil
}

7
pkg/services/serviceaccounts/manager/stats_test.go
View File

@@ -18,13 +18,16 @@ func Test_UsageStats(t *testing.T) {
require.NoError(t, err)
storeMock.ExpectedStats = &serviceaccounts.Stats{
ServiceAccounts: 1,
Tokens: 1,
ServiceAccounts: 1,
Tokens: 1,
ForcedExpiryEnabled: false,
}
stats, err := svc.getUsageMetrics(context.Background())
require.NoError(t, err)
assert.Len(t, stats, 4, stats)
assert.Equal(t, int64(1), stats["stats.serviceaccounts.count"].(int64))
assert.Equal(t, int64(1), stats["stats.serviceaccounts.tokens.count"].(int64))
assert.Equal(t, int64(1), stats["stats.serviceaccounts.secret_scan.enabled.count"].(int64))
assert.Equal(t, int64(0), stats["stats.serviceaccounts.forced_expiry_enabled.count"].(int64))
}