Support X-Trino-Role header

pkg/trino/datasource-context.go

@@ -14,6 +14,7 @@ import (
 const (
 	accessTokenKey     = "accessToken"
 	trinoUserHeader    = "X-Trino-User"
+	trinoRoleHeader    = "X-Trino-Role"
 	trinoClientTagsKey = "X-Trino-Client-Tags"
 	bearerPrefix       = "Bearer "
 )
@@ -41,6 +42,10 @@ func (ds *SQLDatasourceWithTrinoUserContext) QueryData(ctx context.Context, req
 		ctx = context.WithValue(ctx, trinoUserHeader, user)
 	}
 
+	if settings.Role != "" {
+		ctx = context.WithValue(ctx, trinoRoleHeader, "system=ROLE{"+settings.Role+"}")
+	}
+
 	if settings.ClientTags != "" {
 		ctx = context.WithValue(ctx, trinoClientTagsKey, settings.ClientTags)
 	}

pkg/trino/datasource.go

@@ -83,6 +83,7 @@ func (s *TrinoDatasource) SetQueryArgs(ctx context.Context, headers http.Header)
 
 	user := ctx.Value(trinoUserHeader)
 	accessToken := ctx.Value(accessTokenKey)
+	role := ctx.Value(trinoRoleHeader)
 	clientTags := ctx.Value(trinoClientTagsKey)
 
 	if user != nil {
@@ -93,6 +94,10 @@ func (s *TrinoDatasource) SetQueryArgs(ctx context.Context, headers http.Header)
 		args = append(args, sql.Named(accessTokenKey, accessToken.(string)))
 	}
 
+	if role != nil {
+		args = append(args, sql.Named(trinoRoleHeader, role.(string)))
+	}
+
 	if clientTags != nil {
 		args = append(args, sql.Named(trinoClientTagsKey, clientTags.(string)))
 	}

pkg/trino/models/settings.go

@@ -20,6 +20,7 @@ type TrinoDatasourceSettings struct {
 	ClientId            string             `json:"clientId"`
 	ClientSecret        string             `json:"clientSecret"`
 	ImpersonationUser   string             `json:"impersonationUser"`
+	Role                string             `json:"role"`
 	ClientTags          string             `json:"clientTags"`
 }
 

src/ConfigEditor.tsx

@@ -34,6 +34,9 @@ export class ConfigEditor extends PureComponent<Props, State> {
     const onImpersonationUserChange = (event: ChangeEvent<HTMLInputElement>) => {
       onOptionsChange({...options, jsonData: {...options.jsonData, impersonationUser: event.target.value}})
     };
+    const onRoleChange = (event: ChangeEvent<HTMLInputElement>) => {
+      onOptionsChange({...options, jsonData: {...options.jsonData, role: event.target.value}})
+    };
     const onClientTagsChange = (event: ChangeEvent<HTMLInputElement>) => {
       onOptionsChange({...options, jsonData: {...options.jsonData, clientTags: event.target.value}})
     };
@@ -75,6 +78,19 @@ export class ConfigEditor extends PureComponent<Props, State> {
                 />
             </InlineField>
           </div>
+          <div className="gf-form-inline">
+            <InlineField
+              label="Role"
+              tooltip="For X-Trino-Role"
+              labelWidth={26}
+            >
+              <Input
+                value={options.jsonData?.role ?? ''}
+                onChange={onRoleChange}
+                width={40}
+              />
+            </InlineField>
+          </div>
           <div className="gf-form-inline">
             <InlineField
               label="Client Tags"

src/types.ts

@@ -56,6 +56,7 @@ export interface TrinoDataSourceOptions extends DataSourceJsonData {
   tokenUrl?: string;
   clientId?: string;
   impersonationUser?: string;
+  role?: string;
   clientTags?: string;
 }
 /**