Added workflow client tinkerbell#35

Cbkhare · Jul 9, 2020 · 51b3350 · 51b3350
1 parent 32f7149
commit 51b3350
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/packet/client.go b/packet/client.go
@@ -19,12 +19,16 @@ import (
 type hardwareGetter interface {
 }
 
+type workflowGetter interface {
+}
+
 type Client struct {
 	http           *http.Client
 	baseURL        *url.URL
 	consumerToken  string
 	authToken      string
 	hardwareClient hardwareGetter
+	workflowClient workflowGetter
 }
 
 func NewClient(consumerToken, authToken string, baseURL *url.URL) (*Client, error) {
@@ -44,6 +48,7 @@ func NewClient(consumerToken, authToken string, baseURL *url.URL) (*Client, erro
 	}
 
 	var hg hardwareGetter
+	var wg workflowGetter
 	var err error
 	dataModelVersion := os.Getenv("DATA_MODEL_VERSION")
 	switch dataModelVersion {
@@ -52,6 +57,12 @@ func NewClient(consumerToken, authToken string, baseURL *url.URL) (*Client, erro
 		if err != nil {
 			return nil, errors.Wrap(err, "connect to tink")
 		}
+
+		wg, err = tinkClient.TinkerbellWorkflowClient()
+		if err != nil {
+			return nil, errors.Wrap(err, "connect to tink")
+		}
+
 	default:
 		facility := os.Getenv("FACILITY_CODE")
 		if facility == "" {
@@ -62,6 +73,7 @@ func NewClient(consumerToken, authToken string, baseURL *url.URL) (*Client, erro
 		if err != nil {
 			return nil, errors.Wrap(err, "connect to cacher")
 		}
+		wg = hg
 	}
 
 	return &Client{
@@ -70,6 +82,7 @@ func NewClient(consumerToken, authToken string, baseURL *url.URL) (*Client, erro
 		consumerToken:  consumerToken,
 		authToken:      authToken,
 		hardwareClient: hg,
+		workflowClient: wg,
 	}, nil
 }