manual licese reporting persist snapshots CE (#25021)

vault/activity_log.go

@@ -1146,7 +1146,7 @@ func (c *Core) setupActivityLogLocked(ctx context.Context, wg *sync.WaitGroup) e
 			close(manager.retentionDone)
 		}(manager.retentionMonths)
 
-		manager.CensusReportDone = make(chan bool)
+		manager.CensusReportDone = make(chan bool, 1)
 		go c.activityLog.CensusReport(ctx, c.CensusAgent(), c.BillingStart())
 	}
 

vault/census.go

@@ -10,11 +10,12 @@ import "time"
 // CensusAgent is a stub for OSS
 type CensusReporter interface{}
 
-// setupCensusAgent is a stub for OSS.
-func (c *Core) setupCensusAgent() error                { return nil }
-func (c *Core) BillingStart() time.Time                { return time.Time{} }
-func (c *Core) AutomatedLicenseReportingEnabled() bool { return false }
-func (c *Core) CensusAgent() CensusReporter            { return nil }
-func (c *Core) ReloadCensus() error                    { return nil }
-func (c *Core) teardownCensusAgent() error             { return nil }
-func (c *Core) ManualLicenseReportingEnabled() bool    { return false }
+func (c *Core) setupCensusManager() error                   { return nil }
+func (c *Core) BillingStart() time.Time                     { return time.Time{} }
+func (c *Core) AutomatedLicenseReportingEnabled() bool      { return false }
+func (c *Core) CensusAgent() CensusReporter                 { return nil }
+func (c *Core) ReloadCensus() error                         { return nil }
+func (c *Core) teardownCensusManager() error                { return nil }
+func (c *Core) StartManualCensusSnapshots()                 {}
+func (c *Core) ManualLicenseReportingEnabled() bool         { return false }
+func (c *Core) ManualCensusSnapshotInterval() time.Duration { return time.Duration(0) }

vault/core.go

@@ -644,8 +644,6 @@ type Core struct {
 	// it is protected by activityLogLock
 	activityLogConfig ActivityLogCoreConfig
 
-	censusConfig atomic.Value
-
 	// activeTime is set on active nodes indicating the time at which this node
 	// became active.
 	activeTime time.Time
@@ -821,9 +819,6 @@ type CoreConfig struct {
 	LicensePath     string
 	LicensingConfig *LicensingConfig
 
-	// Configured Census Agent
-	CensusAgent CensusReporter
-
 	DisablePerformanceStandby bool
 	DisableIndexing           bool
 	DisableKeyEncodingChecks  bool
@@ -2455,15 +2450,22 @@ func (s standardUnsealStrategy) unseal(ctx context.Context, logger log.Logger, c
 			return err
 		}
 
-		if err := c.setupCensusAgent(); err != nil {
-			c.logger.Error("skipping reporting for nil agent", "error", err)
+		if !c.perfStandby {
+			if err := c.setupCensusManager(); err != nil {
+				logger.Error("skipping license reporting for nil agent", "error", err)
+			}
 		}
 
 		// not waiting on wg to avoid changing existing behavior
 		var wg sync.WaitGroup
 		if err := c.setupActivityLog(ctx, &wg); err != nil {
 			return err
 		}
+
+		if !c.perfStandby {
+			c.StartManualCensusSnapshots()
+		}
+
 	} else {
 		c.auditBroker = NewAuditBroker(c.logger)
 	}
@@ -2659,8 +2661,8 @@ func (c *Core) preSeal() error {
 		result = multierror.Append(result, fmt.Errorf("error stopping expiration: %w", err))
 	}
 	c.stopActivityLog()
-	// Clean up the censusAgent on seal
-	if err := c.teardownCensusAgent(); err != nil {
+	// Clean up census on seal
+	if err := c.teardownCensusManager(); err != nil {
 		result = multierror.Append(result, fmt.Errorf("error tearing down reporting agent: %w", err))
 	}
 

vault/testing.go

@@ -232,7 +232,6 @@ func TestCoreWithSealAndUINoCleanup(t testing.T, opts *CoreConfig) *Core {
 	conf.PluginDirectory = opts.PluginDirectory
 	conf.DetectDeadlocks = opts.DetectDeadlocks
 	conf.Experiments = []string{experiments.VaultExperimentEventsAlpha1}
-	conf.CensusAgent = opts.CensusAgent
 	conf.AdministrativeNamespacePath = opts.AdministrativeNamespacePath
 	conf.ImpreciseLeaseRoleTracking = opts.ImpreciseLeaseRoleTracking