Added logic for creating and managing DAG invocations

NotAnAddictz · leokondrashov · commit 06e0651a9e29 · 2024-12-23T10:33:26.000+08:00
Signed-off-by: Kway Yi Shen &lt;kwayyishen@gmail.com&gt;
diff --git a/.github/configs/wordlist.txt b/.github/configs/wordlist.txt
@@ -23,6 +23,7 @@ workerNodeNum
 durations
 dur
 ACM
+Acyclic
 addr
 adservice
 AdService
@@ -153,6 +154,7 @@ Daglis
 DAGMode
 datacenter
 Datacenter
+DAGs
 dataflows
 dataset
 david
@@ -710,6 +712,7 @@ cgroups
 noop
 YAMLs
 cgo
+EnableDAGDataset
 EnableMetricsScrapping
 EnableZipkinTracing
 EndpointPort
diff --git a/cmd/config_knative_trace.json b/cmd/config_knative_trace.json
@@ -24,5 +24,8 @@
 
   "GRPCConnectionTimeoutSeconds": 15,
   "GRPCFunctionTimeoutSeconds": 900,
-  "DAGMode": false
+  "DAGMode": false,
+  "EnableDAGDataset": true,
+  "Width": 2,
+  "Depth": 2
 }
diff --git a/cmd/loader.go b/cmd/loader.go
@@ -32,11 +32,12 @@ import (
 	"strings"
 	"time"
 
+	"golang.org/x/exp/slices"
+
 	"github.com/vhive-serverless/loader/pkg/common"
 	"github.com/vhive-serverless/loader/pkg/config"
 	"github.com/vhive-serverless/loader/pkg/driver"
 	"github.com/vhive-serverless/loader/pkg/trace"
-	"golang.org/x/exp/slices"
 
 	log "github.com/sirupsen/logrus"
 	tracer "github.com/vhive-serverless/vSwarm/utils/tracing/go"
diff --git a/data/traces/example/dag_structure.csv b/data/traces/example/dag_structure.csv
@@ -0,0 +1,6 @@
+﻿Width,WidthPercentile,Depth,DepthPercentile,TotalNodes,TotalNodes
+1,0.00%,1,0.00%,2,0.00%
+1,78.66%,1,12.51%,2,55.98%
+2,92.13%,2,67.96%,3,79.20%
+3,95.24%,3,86.84%,4,86.63%
+4,100.00%,4,100.00%,5,100.00%
diff --git a/data/traces/reference/.gitattributes b/data/traces/reference/.gitattributes
@@ -1 +1,2 @@
 *.tar.gz filter=lfs diff=lfs merge=lfs -text
+dag_structure.csv filter=lfs diff=lfs merge=lfs -text
diff --git a/data/traces/reference/dag_structure.csv b/data/traces/reference/dag_structure.csv
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4454eacfd79dffef7be37728c442d3f5cdfa183bd219feef8a9036f782806178
+size 76204
diff --git a/docs/configuration.md b/docs/configuration.md
@@ -33,7 +33,10 @@
 | MetricScrapingPeriodSeconds  | int       | > 0                                                                 | 15                  | Period of Prometheus metrics scrapping                                               |
 | GRPCConnectionTimeoutSeconds | int       | > 0                                                                 | 60                  | Timeout for establishing a gRPC connection                                           |
 | GRPCFunctionTimeoutSeconds   | int       | > 0                                                                 | 90                  | Maximum time given to function to execute[^5]                                        |
-| DAGMode                      | bool      | true/false                                                          | false               | Sequential invocation of all functions one after another                             |
+| DAGMode                      | bool      | true/false                                                          | false               | Generates DAG workflows iteratively with functions in TracePath [^8]. Frequency and IAT of the DAG follows their respective entry function, while Duration and Memory of each function will follow their respective values in TracePath.                                                                                                              |                            
+| EnableDAGDataset             | bool      | true/false                                                          | true                |  Generate width and depth from dag_structure.csv in TracePath[^9]                                                                                                      |
+| Width                        | int       | > 0                                                                 | 2                   | Default width of DAG                                                                 |
+| Depth                        | int       | > 0                                                                 | 2                   | Default depth of DAG                                                                 |
 
 [^1]: To run RPS experiments add suffix `-RPS`.
 
@@ -57,6 +60,10 @@ Lambda; https://aws.amazon.com/about-aws/whats-new/2018/10/aws-lambda-supports-f
 
 [^7] It is recommended that the first 10% of cold starts are discarded from the experiment results for low cold start RPS.
 
+[^8]: The generated DAGs consist of unique functions. The shape of each DAG is determined either ```Width,Depth``` or calculated based on ```EnableDAGDAtaset```.
+
+[^9]: A [data sample](https://github.com/icanforce/Orion-OSDI22/blob/main/Public_Dataset/dag_structure.xlsx) of DAG structures has been created based on past Microsoft Azure traces. Width and Depth are determined based on probabilities of this sample.
+
 ---
 
 InVitro can cause failure on cluster manager components. To do so, please configure the `cmd/failure.json`. Make sure
diff --git a/docs/loader.md b/docs/loader.md
@@ -179,6 +179,53 @@ For more options, please see the `Makefile`.
 
 For instructions on how to use the loader with OpenWhisk go to `openwhisk_setup/README.md`.
 
+## Workflow Invocation
+Generation of a Directed Acyclic Graph (DAG) workflow is supported by setting `"DAGMode: true"` in `cmd/config_knative_trace.json` (as specified in [`docs/configuration.md`](../docs/configuration.md)). 
+
+Before invocation, DAGs will be iteratively generated based on the parameters: `width`,`depth`,`EnableDAGDataset`, until the remaining functions are insufficient to maintain the desired DAG structure. The remaining functions will be unused for the rest of the experiment. 
+
+An example of the generated workflow can be seen here:
+
+```bash
+Functions available: 20
+Width: 3
+Depth: 4
+EnableDAGDataset: false
+
+DAG 1: f(0) -> f(1) -> f(3) -> f(5)
+         \        
+          \ -> f(2) -> f(4) -> f(6)
+                        \
+                         \ -> f(7)
+
+DAG 2: f(8) -> f(9) -> f(12) -> f(15)
+        \        
+         \ -> f(10) -> f(13) -> f(16)
+          \
+           \-> f(11) -> f(14) -> f(17)
+```
+In the example, a single invocation of DAG 1 will result in 8 total functions invoked, with parallel invocations per branch. Invocation Frequency and IAT of DAGs 1 and 2 will depend on entry functions f(0) and f(8) respectively.
+
+To obtain [reference traces](https://github.com/vhive-serverless/invitro/blob/main/docs/sampler.md#reference-traces) for DAG execution, use the following command:
+```bash
+git lfs pull 
+tar -xzf data/traces/reference/sampled_150.tar.gz -C data/traces 
+```
+Microsoft has publicly released Microsoft Azure traces of function invocations from 10/18/2021 to 10/31/2021. From this trace, a [data sample](https://github.com/icanforce/Orion-OSDI22/blob/main/Public_Dataset) of DAG structures, representing the cumulative distribution of width and depth of DAGs during that period, was generated. Probabilities were applied to the data to derive the shape of the DAGs. The file `data/traces/example/dag_structure.csv` provides a simplified sample of the publicly released traces.
+
+By default, the shape of the DAGs are automatically calculated at every iteration using the above mentioned cumulative distribution.
+To manually set the shape of the DAGs, change the following parameters in `cmd/config_knative_trace.json`. Note that the number of functions in `TracePath` must be large enough to support the maximum size of the DAG. This ensures all DAGs generated will have the same width and depth.
+```bash
+"EnableDAGDataset": false,
+"Width": <width>,
+"Depth": <depth>
+```
+
+Lastly, start the experiment. This invokes all the generated DAGs with their respective frequencies.
+```bash
+go run cmd/loader.go --config cmd/config_knative_trace.json
+```
+
 ## Running on Cloud Using Serverless Framework
 
 **Currently supported vendors:** AWS
@@ -204,7 +251,6 @@ For instructions on how to use the loader with OpenWhisk go to `openwhisk_setup/
     ```bash
     go run cmd/loader.go --config cmd/config_knative_trace.json
     ```
-
 ---
 Note:
 - Current deployment is via container image.
diff --git a/pkg/common/trace_types.go b/pkg/common/trace_types.go
@@ -24,6 +24,8 @@
 
 package common
 
+import "container/list"
+
 type FunctionInvocationStats struct {
 	HashOwner    string
 	HashApp      string
@@ -103,3 +105,10 @@ type Function struct {
 
 	Specification *FunctionSpecification
 }
+
+type Node struct {
+	Function *Function
+	Branches []*list.List
+	Depth    int
+	DAG      string
+}
diff --git a/pkg/common/utilities.go b/pkg/common/utilities.go
@@ -135,3 +135,15 @@ func SumNumberOfInvocations(withWarmup bool, totalDuration int, functions []*Fun
 
 	return result
 }
+
+func GetName(function *Function) int {
+	parts := strings.Split(function.Name, "-")
+	if parts[0] == "test" {
+		return 0
+	}
+	functionId, err := strconv.Atoi(parts[2])
+	if err != nil {
+		log.Fatal(err)
+	}
+	return functionId
+}
diff --git a/pkg/config/parser.go b/pkg/config/parser.go
@@ -80,6 +80,9 @@ type LoaderConfiguration struct {
 	GRPCConnectionTimeoutSeconds int  `json:"GRPCConnectionTimeoutSeconds"`
 	GRPCFunctionTimeoutSeconds   int  `json:"GRPCFunctionTimeoutSeconds"`
 	DAGMode                      bool `json:"DAGMode"`
+	EnableDAGDataset             bool `json:"EnableDAGDataset"`
+	Width                        int  `json:"Width"`
+	Depth                        int  `json:"Depth"`
 }
 
 func ReadConfigurationFile(path string) LoaderConfiguration {
diff --git a/pkg/config/parser_test.go b/pkg/config/parser_test.go
@@ -61,7 +61,11 @@ func TestConfigParser(t *testing.T) {
 		config.MetricScrapingPeriodSeconds != 15 ||
 		config.AutoscalingMetric != "concurrency" ||
 		config.GRPCConnectionTimeoutSeconds != 15 ||
-		config.GRPCFunctionTimeoutSeconds != 900 {
+		config.GRPCFunctionTimeoutSeconds != 900 ||
+		config.DAGMode != false ||
+		config.EnableDAGDataset != true ||
+		config.Width != 2 ||
+		config.Depth != 2 {
 
 		t.Error("Unexpected configuration read.")
 	}
diff --git a/pkg/config/test_config.json b/pkg/config/test_config.json
@@ -22,5 +22,8 @@
 
   "GRPCConnectionTimeoutSeconds": 15,
   "GRPCFunctionTimeoutSeconds": 900,
-  "DAGMode": false
+  "DAGMode": false,
+  "EnableDAGDataset": true,
+  "Width": 2,
+  "Depth": 2
 }
diff --git a/pkg/config/test_config_aws.json b/pkg/config/test_config_aws.json
@@ -21,5 +21,8 @@
 
   "GRPCConnectionTimeoutSeconds": 15,
   "GRPCFunctionTimeoutSeconds": 900,
-  "DAGMode": false
-}
+  "DAGMode": false,
+  "EnableDAGDataset": true,
+  "Width": 2,
+  "Depth": 2
+}
diff --git a/pkg/driver/trace_driver.go b/pkg/driver/trace_driver.go
diff --git a/pkg/driver/trace_driver_test.go b/pkg/driver/trace_driver_test.go
diff --git a/pkg/generator/dag_generation.go b/pkg/generator/dag_generation.go
diff --git a/pkg/generator/dag_generation_test.go b/pkg/generator/dag_generation_test.go

Original file line number	Diff line number	Diff line change
`@@ -24,5 +24,8 @@`
`24`	`24`
`25`	`25`	`"GRPCConnectionTimeoutSeconds": 15,`
`26`	`26`	`"GRPCFunctionTimeoutSeconds": 900,`
`27`		`- "DAGMode": false`
	`27`	`+ "DAGMode": false,`
	`28`	`+ "EnableDAGDataset": true,`
	`29`	`+ "Width": 2,`
	`30`	`+ "Depth": 2`
`28`	`31`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1 +1,2 @@`
`1`	`1`	`*.tar.gz filter=lfs diff=lfs merge=lfs -text`
	`2`	`+dag_structure.csv filter=lfs diff=lfs merge=lfs -text`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+version https://git-lfs.github.com/spec/v1`
	`2`	`+oid sha256:4454eacfd79dffef7be37728c442d3f5cdfa183bd219feef8a9036f782806178`
	`3`	`+size 76204`