From 1fc1c3d0dabca009da359b005722a7452129d875 Mon Sep 17 00:00:00 2001
From: tyiuhc <tim.yiu@amplitude.com>
Date: Tue, 6 Aug 2024 14:30:49 -0700
Subject: [PATCH] make cohort storage private

---
 pkg/experiment/local/client.go            | 2 +-
 pkg/experiment/local/cohort_loader.go     | 4 ++--
 pkg/experiment/local/cohort_storage.go    | 2 +-
 pkg/experiment/local/deployment_runner.go | 4 ++--
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/pkg/experiment/local/client.go b/pkg/experiment/local/client.go
index 989b710..07286a7 100644
--- 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
index 3fa597b..d325315 100644
--- 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
index 66981e5..7a81f84 100644
--- 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
index 040ea0f..2b85437 100644
--- 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{