From b8fb9e6af86a2e603d347d93290cff77b0296e9f Mon Sep 17 00:00:00 2001
From: Dave Augustus <95467821+dave-augustus@users.noreply.github.com>
Date: Mon, 7 Feb 2022 20:27:14 +0530
Subject: [PATCH] Phase 2 (#6667)

Signed-off-by: Pappu Kumar <pappu.kumar@progress.com>
---
 .../migrations/pipeline/phasetwomigration.go  | 187 ++++++++++++++++++
 1 file changed, 187 insertions(+)
 create mode 100644 components/infra-proxy-service/migrations/pipeline/phasetwomigration.go

diff --git a/components/infra-proxy-service/migrations/pipeline/phasetwomigration.go b/components/infra-proxy-service/migrations/pipeline/phasetwomigration.go
new file mode 100644
index 00000000000..e504284c2d5
--- /dev/null
+++ b/components/infra-proxy-service/migrations/pipeline/phasetwomigration.go
@@ -0,0 +1,187 @@
+package pipeline
+
+import (
+	"context"
+	"fmt"
+)
+
+type PhaseTwoPipleine struct {
+	in chan<- PipelineData
+}
+
+type PhaseTwoPipelineProcessor func(<-chan PipelineData) <-chan PipelineData
+
+// PopulateOrgs returns PhaseTwoPipelineProcessor
+func PopulateOrgs() PhaseTwoPipelineProcessor {
+	return func(result <-chan PipelineData) <-chan PipelineData {
+		return populateOrgs(result)
+	}
+}
+
+func populateOrgs(result <-chan PipelineData) <-chan PipelineData {
+	fmt.Println("Starting populateOrgs routine")
+	out := make(chan PipelineData, 100)
+
+	go func() {
+		for res := range result {
+			fmt.Println("Processing to populateOrgs...")
+			select {
+			case out <- res:
+			case <-res.Ctx.Done():
+				res.Done <- nil
+			}
+		}
+		fmt.Println("Closing populateOrgs routine")
+		close(out)
+	}()
+	return out
+}
+
+// PopulateOrgs returns PhaseTwoPipelineProcessor
+func CreateProject() PhaseTwoPipelineProcessor {
+	return func(result <-chan PipelineData) <-chan PipelineData {
+		return createProject(result)
+	}
+}
+
+func createProject(result <-chan PipelineData) <-chan PipelineData {
+	fmt.Println("Starting CreateProject routine")
+	out := make(chan PipelineData, 100)
+
+	go func() {
+		for res := range result {
+			fmt.Println("Processing to createProject...")
+			select {
+			case out <- res:
+			case <-res.Ctx.Done():
+				res.Done <- nil
+			}
+		}
+		fmt.Println("Closing CreateProject routine")
+		close(out)
+	}()
+	return out
+}
+
+// PopulateUsers returns PhaseTwoPipelineProcessor
+func PopulateUsers() PhaseTwoPipelineProcessor {
+	return func(result <-chan PipelineData) <-chan PipelineData {
+		return populateUsers(result)
+	}
+}
+
+func populateUsers(result <-chan PipelineData) <-chan PipelineData {
+	fmt.Println("Starting PopulateUsers routine")
+	out := make(chan PipelineData, 100)
+
+	go func() {
+		for res := range result {
+			fmt.Println("Processing to populateUsers...")
+			select {
+			case out <- res:
+			case <-res.Ctx.Done():
+				res.Done <- nil
+			}
+		}
+		fmt.Println("Closing PopulateUsers routine")
+		close(out)
+	}()
+	return out
+}
+
+// PopulateORGUser returns PhaseTwoPipelineProcessor
+func PopulateORGUser() PhaseTwoPipelineProcessor {
+	return func(result <-chan PipelineData) <-chan PipelineData {
+		return populateORGUser(result)
+	}
+}
+
+func populateORGUser(result <-chan PipelineData) <-chan PipelineData {
+	fmt.Println("Starting PopulateORGUser routine")
+	out := make(chan PipelineData, 100)
+
+	go func() {
+		for res := range result {
+			fmt.Println("Processing to populateORGUser...")
+			select {
+			case out <- res:
+			case <-res.Ctx.Done():
+				res.Done <- nil
+			}
+		}
+		fmt.Println("Closing PopulateORGUser routine")
+		close(out)
+	}()
+	return out
+}
+
+// PopulateMembersPolicy returns PhaseTwoPipelineProcessor
+func PopulateMembersPolicy() PhaseTwoPipelineProcessor {
+	return func(result <-chan PipelineData) <-chan PipelineData {
+		return populateMembersPolicy(result)
+	}
+}
+
+func populateMembersPolicy(result <-chan PipelineData) <-chan PipelineData {
+	fmt.Println("Starting PopulateMembersPolicy routine")
+	out := make(chan PipelineData, 100)
+
+	go func() {
+		for res := range result {
+			fmt.Println("Processing to populateMembersPolicy...")
+			select {
+			case out <- res:
+			case <-res.Ctx.Done():
+				res.Done <- nil
+			}
+		}
+		fmt.Println("Closing PopulateMembersPolicy routine")
+		close(out)
+	}()
+	return out
+}
+
+func migrationTwoPipeline(source <-chan PipelineData, pipes ...PhaseTwoPipelineProcessor) {
+	fmt.Println("Pipeline started...")
+
+	go func() {
+		for _, pipe := range pipes {
+			source = pipe(source)
+		}
+
+		for s := range source {
+			s.Done <- nil
+		}
+	}()
+}
+
+func SetupPhaseTwoPipeline() PhaseTwoPipleine {
+	c := make(chan PipelineData, 100)
+	migrationTwoPipeline(c,
+		PopulateOrgs(),
+		CreateProject(),
+		PopulateUsers(),
+		PopulateORGUser(),
+		PopulateMembersPolicy(),
+	)
+	return PhaseTwoPipleine{in: c}
+}
+
+func (p *PhaseTwoPipleine) Run(result Result) {
+	status := make(chan string)
+	go func() {
+		ctx, cancel := context.WithCancel(context.Background())
+		defer cancel()
+		done := make(chan error)
+		select {
+		case p.in <- PipelineData{Result: result, Done: done, Ctx: ctx}:
+		}
+		err := <-done
+		if err != nil {
+			fmt.Println("received error")
+		}
+		fmt.Println("received done")
+		status <- "Done"
+	}()
+	<-status
+}