@finos/legend-extension-dsl-data-quality 2.0.16 → 2.0.18

package/src/graph-manager/protocol/pure/v1/V1_DSL_Data_Quality_PureGraphManagerExtension.ts

@@ -17,14 +17,17 @@
 import {
   type AbstractPureGraphManager,
   type ExecutionResult,
+  type EXECUTION_SERIALIZATION_FORMAT,
+  type ExecutionOptions,
   type GraphDataOrigin,
   type PureModel,
   type RawExecutionPlan,
   type RootGraphFetchTree,
   type V1_ExecutionResult,
+  type V1_ParameterValue,
   type V1_PureModelContext,
   type V1_RootGraphFetchTree,
-  type V1_ParameterValue,
+  V1_getEngineSerializationFormat,
   LegendSDLC,
   PureClientVersion,
   V1_buildExecutionError,
@@ -32,26 +35,27 @@ import {
   V1_ExecutionError,
   V1_GraphBuilderContextBuilder,
   V1_LegendSDLC,
+  V1_parameterValueModelSchema,
   V1_ProcessingContext,
   V1_Protocol,
   V1_PureGraphManager,
   V1_PureModelContextPointer,
   V1_pureModelContextPropSchema,
+  V1_RemoteEngine,
   V1_serializeExecutionResult,
-  V1_parameterValueModelSchema,
   V1_transformParameterValue,
-  V1_RemoteEngine,
 } from '@finos/legend-graph';
 import { createModelSchema, optional, primitive } from 'serializr';
 import {
   type PlainObject,
   assertErrorThrown,
+  customListWithSchema,
+  guaranteeNonNullable,
   guaranteeType,
   NetworkClientError,
   returnUndefOnError,
   SerializationFactory,
   UnsupportedOperationError,
-  customListWithSchema,
 } from '@finos/legend-shared';
 import { DSL_DataQuality_PureGraphManagerExtension } from '../DSL_DataQuality_PureGraphManagerExtension.js';
 import {
@@ -123,6 +127,7 @@ export class V1_DSL_Data_Quality_PureGraphManagerExtension extends DSL_DataQuali
     input: PlainObject<V1_DQExecuteInput>,
     options?: {
       returnAsResponse?: boolean;
+      serializationFormat?: EXECUTION_SERIALIZATION_FORMAT | undefined;
     },
   ): Promise<PlainObject<V1_ExecutionResult> | Response> => {
     // TODO: improve abstraction so that we do not need to access the engine server client directly
@@ -137,7 +142,11 @@ export class V1_DSL_Data_Quality_PureGraphManagerExtension extends DSL_DataQuali
       input,
       {},
       undefined,
-      undefined,
+      {
+        serializationFormat: options?.serializationFormat
+          ? V1_getEngineSerializationFormat(options.serializationFormat)
+          : undefined,
+      },
       { enableCompression: true },
       {
         skipProcessing: Boolean(options?.returnAsResponse),
@@ -158,6 +167,33 @@ export class V1_DSL_Data_Quality_PureGraphManagerExtension extends DSL_DataQuali
     ).text();
   }
 
+  private async export(
+    input: V1_DQExecuteInput,
+    options?: ExecutionOptions,
+  ): Promise<Response> {
+    try {
+      return guaranteeNonNullable(
+        (await this.executeValidation(
+          V1_DQExecuteInput.serialization.toJson(input),
+          {
+            serializationFormat: options?.serializationFormat,
+            returnAsResponse: true,
+          },
+        )) as Response,
+      );
+    } catch (error) {
+      assertErrorThrown(error);
+      if (error instanceof NetworkClientError) {
+        throw V1_buildExecutionError(
+          V1_ExecutionError.serialization.fromJson(
+            error.payload as PlainObject<V1_ExecutionError>,
+          ),
+        );
+      }
+      throw error;
+    }
+  }
+
   createExecutionInput(
     graph: PureModel,
     packagePath: string,
@@ -250,6 +286,20 @@ export class V1_DSL_Data_Quality_PureGraphManagerExtension extends DSL_DataQuali
     }
   };
 
+  exportData = async (
+    graph: PureModel,
+    packagePath: string,
+    options: DQExecuteInputOptions,
+  ): Promise<Response> => {
+    const input = this.createExecutionInput(
+      graph,
+      packagePath,
+      new V1_DQExecuteInput(),
+      options,
+    );
+    return this.export(input, options);
+  };
+
   debugExecutionPlanGeneration = async (
     graph: PureModel,
     packagePath: string,

package/style/_data-quality-relation-validation-builder.scss

@@ -191,6 +191,19 @@
 
         @include flexVCenter;
         @include flexHSpaceBetween;
+
+        &__action {
+          display: flex;
+          height: 2.8rem;
+        }
+      }
+
+      &__header-group {
+        display: flex;
+        flex-direction: row;
+        color: var(--color-light-grey-300);
+
+        @include flexVCenter;
       }
 
       &__header-right {

package/style/_data-quality-validation-builder.scss

@@ -1316,11 +1316,17 @@
     width: 100%;
   }
 
+  .panel__header {
+    margin-right: 0.5rem;
+  }
+
   &__content {
     position: relative;
   }
 
   &__analytics {
+    display: flex;
+    justify-content: center;
     font-size: 1.2rem;
     color: var(--color-dark-grey-400);
     margin-left: 0.5rem;