Feat: support passing do key in http header

Signed-off-by: Yin Da <yd219913@alibaba-inc.com>
kubevela · Feb 22, 2023 · 5635c3a · 5635c3a
1 parent 48b398d
commit 5635c3a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/cue/cuex/runtime/package.go b/cue/cuex/runtime/package.go
@@ -80,8 +80,10 @@ func (in *externalPackage) GetProviderFn(do string) ProviderFn {
 	if in.src.Spec.Provider == nil {
 		return nil
 	}
-	fn := ExternalProviderFn(*in.src.Spec.Provider)
-	return &fn
+	return &ExternalProviderFn{
+		Provider: *in.src.Spec.Provider,
+		Do:       do,
+	}
 }
 
 func (in *externalPackage) GetName() string {

diff --git a/cue/cuex/runtime/provider.go b/cue/cuex/runtime/provider.go
@@ -59,7 +59,10 @@ func (fn GenericProviderFn[T, U]) Call(ctx context.Context, value cue.Value) (cu
 var _ ProviderFn = (*ExternalProviderFn)(nil)
 
 // ExternalProviderFn external provider that implements ProviderFn interface
-type ExternalProviderFn v1alpha1.Provider
+type ExternalProviderFn struct {
+	v1alpha1.Provider
+	Do string
+}
 
 // DefaultClientInsecureSkipVerify set if the default external provider client
 // use insecure-skip-verify
@@ -74,6 +77,9 @@ var DefaultClient = singleton.NewSingleton(func() *http.Client {
 	}
 })
 
+// FunctionHeaderKey http header for recording cuex provider function
+const FunctionHeaderKey = "CueX-External-Provider-Function"
+
 // Call dial external endpoints by passing the json data of the input parameter,
 // then fill back returned values
 func (in *ExternalProviderFn) Call(ctx context.Context, value cue.Value) (cue.Value, error) {
@@ -89,6 +95,7 @@ func (in *ExternalProviderFn) Call(ctx context.Context, value cue.Value) (cue.Va
 			return value, err
 		}
 		req.Header.Set("Content-Type", runtime.ContentTypeJSON)
+		req.Header.Set(FunctionHeaderKey, in.Do)
 		resp, err := DefaultClient.Get().Do(req.WithContext(ctx))
 		if err != nil {
 			return value, err