Run ECS launch in goroutine, and add local cluster paymodel (#61)

* Run ECS launch in goroutine, to prevent timeout issues

* Fix context for launchEcsWorkspace

* Fix status

* Fix transitgateway setup

* Fix routing

* Change order of execution for launch

* Prevent launch if paymodel is not active

* Add 500 error on launch instead of 403

* Log when launch is forbidden

* Add more failure logging

* Add in enhancement for paymodels check.

* Remove comment

* return 404 if no paymodel is found

* Remove error returns from ecsLaunchWorkspace

* Remove error returns from ecsLaunchWorkspace

* Update hatchery/hatchery.go

Co-authored-by: Sai Shanmukha Narumanchi <nss10@outlook.com>

* Making needed changes

* Updating `status` endpoint to work with local launch

* Reverting local changes

* Updating comments

* Fix nil pointer reference

---------

Co-authored-by: Sai Shanmukha Narumanchi <nss10@outlook.com>

.gitignore

@@ -25,3 +25,4 @@
 
 # End of https://www.toptal.com/developers/gitignore/api/go
 .dccache
+hatchery_qabrh.json

.secrets.baseline

@@ -1,9 +1,9 @@
 {
   "exclude": {
-    "files": "go.sum",
+    "files": "go.sum|^.secrets.baseline$",
     "lines": null
   },
-  "generated_at": "2021-10-13T19:11:36Z",
+  "generated_at": "2023-04-07T16:22:23Z",
   "plugins_used": [
     {
       "name": "AWSKeyDetector"
@@ -61,6 +61,7 @@
     "doc/explanation/dockstore.md": [
       {
         "hashed_secret": "9b5925ea817163740dfb287a9894e8ab3aba2c18",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 91,
         "type": "Secret Keyword"
@@ -69,6 +70,7 @@
     "doc/howto/configuration.md": [
       {
         "hashed_secret": "e94cc2a86b04ad4ddc98fcbf91ed236437939d47",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 30,
         "type": "Secret Keyword"
@@ -77,6 +79,7 @@
     "doc/howto/jupyterNotebook.md": [
       {
         "hashed_secret": "e94cc2a86b04ad4ddc98fcbf91ed236437939d47",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 35,
         "type": "Secret Keyword"
@@ -85,14 +88,25 @@
     "doc/howto/noVNCFirefox.md": [
       {
         "hashed_secret": "0da0e0005ca04acb407af2681d0bede6d9406039",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 51,
         "type": "Secret Keyword"
       }
     ],
+    "hatchery/ecs.go": [
+      {
+        "hashed_secret": "96e77458817d8486fab4c56c8d4c51b83f85c79d",
+        "is_secret": false,
+        "is_verified": false,
+        "line_number": 391,
+        "type": "Secret Keyword"
+      }
+    ],
     "testData/dockstore/docker-compose.yml": [
       {
         "hashed_secret": "770fc9a9befcecb410cd10a29eec487beb70009c",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 81,
         "type": "Secret Keyword"
@@ -101,6 +115,7 @@
     "testData/testConfig.json": [
       {
         "hashed_secret": "0da0e0005ca04acb407af2681d0bede6d9406039",
+        "is_secret": false,
         "is_verified": false,
         "line_number": 55,
         "type": "Secret Keyword"

hatchery/alb.go

@@ -217,6 +217,47 @@ func (creds *CREDS) CreateLoadBalancer(userName string) (*elbv2.CreateLoadBalanc
 	return loadBalancer, targetGroup.TargetGroups[0].TargetGroupArn, listener, nil
 }
 
+func (creds *CREDS) terminateLoadBalancerTargetGroup(userName string) error {
+	svc := elbv2.New(session.Must(session.NewSession(&aws.Config{
+		Credentials: creds.creds,
+		Region:      aws.String("us-east-1"),
+	})))
+	tgName := truncateString(strings.ReplaceAll(os.Getenv("GEN3_ENDPOINT"), ".", "-")+userToResourceName(userName, "service")+"tg", 32)
+	Config.Logger.Printf("Deleting target group: %s", tgName)
+	tgArn, err := svc.DescribeTargetGroups(&elbv2.DescribeTargetGroupsInput{
+		Names: []*string{aws.String(tgName)},
+	})
+	if err != nil {
+		if aerr, ok := err.(awserr.Error); ok {
+			switch aerr.Code() {
+			case elbv2.ErrCodeTargetGroupNotFoundException:
+				// Target group not found, nothing to do
+				return nil
+			}
+		} else {
+			Config.Logger.Printf("Error describing target group: %s", err.Error())
+			return err
+		}
+	}
+	input := &elbv2.DeleteTargetGroupInput{
+		TargetGroupArn: tgArn.TargetGroups[0].TargetGroupArn,
+	}
+
+	_, err = svc.DeleteTargetGroup(input)
+	if err != nil {
+		if aerr, ok := err.(awserr.Error); ok {
+			switch aerr.Code() {
+			case elbv2.ErrCodeResourceInUseException:
+				// Target group in use, nothing to do
+				return nil
+			}
+		} else {
+			Config.Logger.Printf("Error deleting target group: %s", err.Error())
+		}
+	}
+	return nil
+}
+
 func (creds *CREDS) terminateLoadBalancer(userName string) error {
 	svc := elbv2.New(session.Must(session.NewSession(&aws.Config{
 		Credentials: creds.creds,
@@ -229,16 +270,32 @@ func (creds *CREDS) terminateLoadBalancer(userName string) error {
 	}
 	result, err := svc.DescribeLoadBalancers(getInput)
 	if err != nil {
-		return err
+		if aerr, ok := err.(awserr.Error); ok {
+			switch aerr.Code() {
+			case elbv2.ErrCodeLoadBalancerNotFoundException:
+				// Load balancer doesn't exist, we are happy! :)
+				return nil
+			}
+		} else {
+			return err
+		}
 	}
 	if len(result.LoadBalancers) == 1 {
 		delInput := &elbv2.DeleteLoadBalancerInput{
 			LoadBalancerArn: result.LoadBalancers[0].LoadBalancerArn,
 		}
 		_, err := svc.DeleteLoadBalancer(delInput)
 		if err != nil {
-			return err
+			if aerr, ok := err.(awserr.Error); ok {
+				switch aerr.Code() {
+				case elbv2.ErrCodeLoadBalancerNotFoundException:
+					fmt.Println(elbv2.ErrCodeLoadBalancerNotFoundException, aerr.Error())
+				}
+			} else {
+				return err
+			}
 		}
 	}
+
 	return nil
 }

hatchery/config.go

@@ -60,6 +60,8 @@ type PayModel struct {
 	Name            string  `json:"workspace_type"`
 	User            string  `json:"user_id"`
 	AWSAccountId    string  `json:"account_id"`
+	Status          string  `json:"request_status"`
+	Local           bool    `json:"local"`
 	Region          string  `json:"region"`
 	Ecs             bool    `json:"ecs"`
 	Subnet          int     `json:"subnet"`