feat(entities) Adds batch dependent store interface and mysql implementation

Author: manjari25

extension/storage/BUILD.bazel

@@ -3,6 +3,7 @@ load("@rules_go//go:def.bzl", "go_library")
 go_library(
     name = "storage",
     srcs = [
+        "batch_dependent_store.go",
         "batch_store.go",
         "change_provider_store.go",
         "request_store.go",

extension/storage/mysql/BUILD.bazel

@@ -3,6 +3,7 @@ load("@rules_go//go:def.bzl", "go_library")
 go_library(
     name = "mysql",
     srcs = [
+        "batch_dependent_store.go",
         "batch_store.go",
         "change_provider_store.go",
         "request_store.go",

extension/storage/mysql/storage.go

@@ -28,10 +28,11 @@ type MySQLParameters struct {
 }
 
 type mysqlStorage struct {
-	db                  *sql.DB
-	requestStore        storage.RequestStore
-	changeProviderStore storage.ChangeProviderStore
-	batchStore          storage.BatchStore
+	db                    *sql.DB
+	requestStore          storage.RequestStore
+	changeProviderStore   storage.ChangeProviderStore
+	batchStore            storage.BatchStore
+	batchDependentStore   storage.BatchDependentStore
 }
 
 // NewStorage creates a new MySQL storage.
@@ -52,10 +53,11 @@ func NewStorage(p MySQLParameters) (storage.Storage, error) {
 	}
 
 	return &mysqlStorage{
-		db:                  db,
-		requestStore:        NewRequestStore(db),
-		changeProviderStore: NewChangeProviderStore(db),
-		batchStore:          NewBatchStore(db),
+		db:                    db,
+		requestStore:          NewRequestStore(db),
+		changeProviderStore:   NewChangeProviderStore(db),
+		batchStore:            NewBatchStore(db),
+		batchDependentStore:   NewBatchDependentStore(db),
 	}, nil
 }
 
@@ -74,6 +76,11 @@ func (f *mysqlStorage) GetBatchStore() storage.BatchStore {
 	return f.batchStore
 }
 
+// GetBatchDependentStore returns the MySQL-backed BatchDependentStore.
+func (f *mysqlStorage) GetBatchDependentStore() storage.BatchDependentStore {
+	return f.batchDependentStore
+}
+
 // Close closes the underlying database connection.
 func (f *mysqlStorage) Close() error {
 	return f.db.Close()

extension/storage/storage.go

@@ -35,6 +35,9 @@ type Storage interface {
 	// GetBatchStore returns the BatchStore instance.
 	GetBatchStore() BatchStore
 
+	// GetBatchDependentStore returns the BatchDependentStore instance.
+	GetBatchDependentStore() BatchDependentStore
+
 	// Close closes the storage and all underlying connections. Should only be called once at the end of the program.
 	Close() error
 }

gateway/controller/land_test.go

@@ -78,10 +78,30 @@ func (m *mockBatchStore) UpdateState(ctx context.Context, id string, version int
 	return nil
 }
 
+type mockBatchDependentStore struct {
+	createFunc func(ctx context.Context, batchDependent entity.BatchDependent) error
+	getFunc    func(ctx context.Context, batchID string) (entity.BatchDependent, error)
+}
+
+func (m *mockBatchDependentStore) Get(ctx context.Context, batchID string) (entity.BatchDependent, error) {
+	if m.getFunc != nil {
+		return m.getFunc(ctx, batchID)
+	}
+	return entity.BatchDependent{}, nil
+}
+
+func (m *mockBatchDependentStore) Create(ctx context.Context, batchDependent entity.BatchDependent) error {
+	if m.createFunc != nil {
+		return m.createFunc(ctx, batchDependent)
+	}
+	return nil
+}
+
 type mockStorage struct {
-	requestStore        storage.RequestStore
-	changeProviderStore storage.ChangeProviderStore
-	batchStore          storage.BatchStore
+	requestStore          storage.RequestStore
+	changeProviderStore   storage.ChangeProviderStore
+	batchStore            storage.BatchStore
+	batchDependentStore   storage.BatchDependentStore
 }
 
 func (m *mockStorage) GetRequestStore() storage.RequestStore {
@@ -96,6 +116,10 @@ func (m *mockStorage) GetBatchStore() storage.BatchStore {
 	return m.batchStore
 }
 
+func (m *mockStorage) GetBatchDependentStore() storage.BatchDependentStore {
+	return m.batchDependentStore
+}
+
 func (m *mockStorage) Close() error {
 	return nil
 }