make cohort storage private

amplitude · Aug 6, 2024 · 1fc1c3d · 1fc1c3d
1 parent 82cc98a
commit 1fc1c3d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/pkg/experiment/local/client.go b/pkg/experiment/local/client.go
@@ -30,7 +30,7 @@ type Client struct {
 	flagsMutex        *sync.RWMutex
 	engine            *evaluation.Engine
 	assignmentService *assignmentService
-	cohortStorage     CohortStorage
+	cohortStorage     cohortStorage
 	flagConfigStorage flagConfigStorage
 	cohortLoader      *cohortLoader
 	deploymentRunner  *deploymentRunner

diff --git a/pkg/experiment/local/cohort_loader.go b/pkg/experiment/local/cohort_loader.go
@@ -7,13 +7,13 @@ import (
 
 type cohortLoader struct {
 	cohortDownloadApi cohortDownloadApi
-	cohortStorage     CohortStorage
+	cohortStorage     cohortStorage
 	jobs              sync.Map
 	executor          *sync.Pool
 	lockJobs          sync.Mutex
 }
 
-func newCohortLoader(cohortDownloadApi cohortDownloadApi, cohortStorage CohortStorage) *cohortLoader {
+func newCohortLoader(cohortDownloadApi cohortDownloadApi, cohortStorage cohortStorage) *cohortLoader {
 	return &cohortLoader{
 		cohortDownloadApi: cohortDownloadApi,
 		cohortStorage:     cohortStorage,

diff --git a/pkg/experiment/local/cohort_storage.go b/pkg/experiment/local/cohort_storage.go
@@ -4,7 +4,7 @@ import (
 	"sync"
 )
 
-type CohortStorage interface {
+type cohortStorage interface {
 	getCohort(cohortID string) *Cohort
 	getCohorts() map[string]*Cohort
 	getCohortsForUser(userID string, cohortIDs map[string]struct{}) map[string]struct{}

diff --git a/pkg/experiment/local/deployment_runner.go b/pkg/experiment/local/deployment_runner.go
@@ -12,7 +12,7 @@ type deploymentRunner struct {
 	config            *Config
 	flagConfigApi     flagConfigApi
 	flagConfigStorage flagConfigStorage
-	cohortStorage     CohortStorage
+	cohortStorage     cohortStorage
 	cohortLoader      *cohortLoader
 	lock              sync.Mutex
 	poller            *poller
@@ -23,7 +23,7 @@ func newDeploymentRunner(
 	config *Config,
 	flagConfigApi flagConfigApi,
 	flagConfigStorage flagConfigStorage,
-	cohortStorage CohortStorage,
+	cohortStorage cohortStorage,
 	cohortLoader *cohortLoader,
 ) *deploymentRunner {
 	dr := &deploymentRunner{