[DOP-22196] add clickhouse, mssql, mysql connection types

Zabilsya · Zabilsya · commit 292bf074fc60 · 2024-12-10T15:48:10.000+06:00
diff --git a/src/entities/connection/api/types.ts b/src/entities/connection/api/types.ts
@@ -7,7 +7,15 @@ export type Connection = {
   description: string;
 } & ConnectionData;
 
-export type ConnectionData = ConnectionHive | ConnectionHdfs | ConnectionOracle | ConnectionPostgres | ConnectionS3;
+export type ConnectionData =
+  | ConnectionHive
+  | ConnectionHdfs
+  | ConnectionOracle
+  | ConnectionPostgres
+  | ConnectionClickhouse
+  | ConnectionMySql
+  | ConnectionMsSql
+  | ConnectionS3;
 
 export type ConnectionBucketStyle = 'domain' | 'path';
 
@@ -66,6 +74,48 @@ export interface ConnectionPostgres {
   };
 }
 
+export interface ConnectionClickhouse {
+  auth_data: {
+    type: ConnectionType.CLICKHOUSE;
+    user: string;
+    password?: string;
+  };
+  connection_data: {
+    type: ConnectionType.CLICKHOUSE;
+    host: string;
+    port: number;
+    database_name: string;
+  };
+}
+
+export interface ConnectionMySql {
+  auth_data: {
+    type: ConnectionType.MY_SQL;
+    user: string;
+    password?: string;
+  };
+  connection_data: {
+    type: ConnectionType.MY_SQL;
+    host: string;
+    port: number;
+    database_name: string;
+  };
+}
+
+export interface ConnectionMsSql {
+  auth_data: {
+    type: ConnectionType.MS_SQL;
+    user: string;
+    password?: string;
+  };
+  connection_data: {
+    type: ConnectionType.MS_SQL;
+    host: string;
+    port: number;
+    database_name: string;
+  };
+}
+
 export interface ConnectionS3 {
   auth_data: {
     type: ConnectionType.S3;
diff --git a/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionClickhouse/index.tsx b/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionClickhouse/index.tsx
@@ -0,0 +1,29 @@
+import React from 'react';
+import { Form, Input, InputNumber } from 'antd';
+
+import { useSensitiveFields } from '../../hooks';
+import { MAX_ALLOWED_PORT, MIN_ALLOWED_PORT } from '../../constants';
+
+export const ConnectionClickhouse = () => {
+  const { isRequired } = useSensitiveFields();
+
+  return (
+    <>
+      <Form.Item label="Database name" name="database_name" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Host" name="host" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Port" name="port" rules={[{ required: true }]}>
+        <InputNumber size="large" min={MIN_ALLOWED_PORT} max={MAX_ALLOWED_PORT} />
+      </Form.Item>
+      <Form.Item label="User" name="user" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Password" name="password" rules={[{ required: isRequired }]}>
+        <Input.Password size="large" />
+      </Form.Item>
+    </>
+  );
+};
diff --git a/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionMsSql/index.tsx b/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionMsSql/index.tsx
@@ -0,0 +1,29 @@
+import React from 'react';
+import { Form, Input, InputNumber } from 'antd';
+
+import { useSensitiveFields } from '../../hooks';
+import { MAX_ALLOWED_PORT, MIN_ALLOWED_PORT } from '../../constants';
+
+export const ConnectionMsSql = () => {
+  const { isRequired } = useSensitiveFields();
+
+  return (
+    <>
+      <Form.Item label="Database name" name="database_name" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Host" name="host" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Port" name="port" rules={[{ required: true }]}>
+        <InputNumber size="large" min={MIN_ALLOWED_PORT} max={MAX_ALLOWED_PORT} />
+      </Form.Item>
+      <Form.Item label="User" name="user" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Password" name="password" rules={[{ required: isRequired }]}>
+        <Input.Password size="large" />
+      </Form.Item>
+    </>
+  );
+};
diff --git a/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionMySql/index.tsx b/src/entities/connection/ui/ConnectionTypeForm/components/ConnectionMySql/index.tsx
@@ -0,0 +1,29 @@
+import React from 'react';
+import { Form, Input, InputNumber } from 'antd';
+
+import { useSensitiveFields } from '../../hooks';
+import { MAX_ALLOWED_PORT, MIN_ALLOWED_PORT } from '../../constants';
+
+export const ConnectionMySql = () => {
+  const { isRequired } = useSensitiveFields();
+
+  return (
+    <>
+      <Form.Item label="Database name" name="database_name" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Host" name="host" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Port" name="port" rules={[{ required: true }]}>
+        <InputNumber size="large" min={MIN_ALLOWED_PORT} max={MAX_ALLOWED_PORT} />
+      </Form.Item>
+      <Form.Item label="User" name="user" rules={[{ required: true }]}>
+        <Input size="large" />
+      </Form.Item>
+      <Form.Item label="Password" name="password" rules={[{ required: isRequired }]}>
+        <Input.Password size="large" />
+      </Form.Item>
+    </>
+  );
+};
diff --git a/src/entities/connection/ui/ConnectionTypeForm/components/index.ts b/src/entities/connection/ui/ConnectionTypeForm/components/index.ts
@@ -3,3 +3,6 @@ export * from './ConnectionHive';
 export * from './ConnectionOracle';
 export * from './ConnectionPostgres';
 export * from './ConnectionS3';
+export * from './ConnectionClickhouse';
+export * from './ConnectionMySql';
+export * from './ConnectionMsSql';
diff --git a/src/entities/connection/ui/ConnectionTypeForm/constants.tsx b/src/entities/connection/ui/ConnectionTypeForm/constants.tsx
@@ -3,7 +3,16 @@ import { createContext } from 'react';
 import { ConnectionType } from '@shared/types';
 
 import { SensitiveFieldsContextProps } from './types';
-import { ConnectionHdfs, ConnectionHive, ConnectionOracle, ConnectionPostgres, ConnectionS3 } from './components';
+import {
+  ConnectionClickhouse,
+  ConnectionHdfs,
+  ConnectionHive,
+  ConnectionMsSql,
+  ConnectionMySql,
+  ConnectionOracle,
+  ConnectionPostgres,
+  ConnectionS3,
+} from './components';
 
 const SENSITIVE_FIELDS_CONTEXT_INITIAL_VALUE: SensitiveFieldsContextProps = {
   isRequired: true,
@@ -21,5 +30,8 @@ export const CONNECTION_TYPE_COMPONENT = {
   [ConnectionType.HIVE]: <ConnectionHive />,
   [ConnectionType.ORACLE]: <ConnectionOracle />,
   [ConnectionType.POSTGRES]: <ConnectionPostgres />,
+  [ConnectionType.CLICKHOUSE]: <ConnectionClickhouse />,
+  [ConnectionType.MY_SQL]: <ConnectionMySql />,
+  [ConnectionType.MS_SQL]: <ConnectionMsSql />,
   [ConnectionType.S3]: <ConnectionS3 />,
 };
diff --git a/src/entities/connection/utils/adaptConnectionClickhouse/index.ts b/src/entities/connection/utils/adaptConnectionClickhouse/index.ts
@@ -0,0 +1,20 @@
+import { ConnectionClickhouse } from '../../api';
+
+import { AdaptConnectionClickhouseProps } from './types';
+
+/* Util for creating a ConnectionData object for "Clickhouse" connection by its fields */
+export const adaptConnectionClickhouse = (data: AdaptConnectionClickhouseProps): ConnectionClickhouse => {
+  return {
+    auth_data: {
+      type: data.type,
+      user: data.user,
+      password: data.password,
+    },
+    connection_data: {
+      type: data.type,
+      database_name: data.database_name,
+      host: data.host,
+      port: data.port,
+    },
+  };
+};
diff --git a/src/entities/connection/utils/adaptConnectionClickhouse/types.ts b/src/entities/connection/utils/adaptConnectionClickhouse/types.ts
@@ -0,0 +1,4 @@
+import { ConnectionClickhouse } from '../../api';
+
+export type AdaptConnectionClickhouseProps = ConnectionClickhouse['auth_data'] &
+  ConnectionClickhouse['connection_data'];
diff --git a/src/entities/connection/utils/adaptConnectionMsSql/index.ts b/src/entities/connection/utils/adaptConnectionMsSql/index.ts
@@ -0,0 +1,20 @@
+import { ConnectionMsSql } from '../../api';
+
+import { AdaptConnectionMsSqlProps } from './types';
+
+/* Util for creating a ConnectionData object for "MsSQL" connection by its fields */
+export const adaptConnectionMsSql = (data: AdaptConnectionMsSqlProps): ConnectionMsSql => {
+  return {
+    auth_data: {
+      type: data.type,
+      user: data.user,
+      password: data.password,
+    },
+    connection_data: {
+      type: data.type,
+      database_name: data.database_name,
+      host: data.host,
+      port: data.port,
+    },
+  };
+};
diff --git a/src/entities/connection/utils/adaptConnectionMsSql/types.ts b/src/entities/connection/utils/adaptConnectionMsSql/types.ts
@@ -0,0 +1,3 @@
+import { ConnectionMsSql } from '../../api';
+
+export type AdaptConnectionMsSqlProps = ConnectionMsSql['auth_data'] & ConnectionMsSql['connection_data'];
diff --git a/src/entities/connection/utils/adaptConnectionMySql/index.ts b/src/entities/connection/utils/adaptConnectionMySql/index.ts
@@ -0,0 +1,20 @@
+import { ConnectionMySql } from '../../api';
+
+import { AdaptConnectionMySqlProps } from './types';
+
+/* Util for creating a ConnectionData object for "MySQL" connection by its fields */
+export const adaptConnectionMySql = (data: AdaptConnectionMySqlProps): ConnectionMySql => {
+  return {
+    auth_data: {
+      type: data.type,
+      user: data.user,
+      password: data.password,
+    },
+    connection_data: {
+      type: data.type,
+      database_name: data.database_name,
+      host: data.host,
+      port: data.port,
+    },
+  };
+};
diff --git a/src/entities/connection/utils/adaptConnectionMySql/types.ts b/src/entities/connection/utils/adaptConnectionMySql/types.ts
@@ -0,0 +1,3 @@
+import { ConnectionMySql } from '../../api';
+
+export type AdaptConnectionMySqlProps = ConnectionMySql['auth_data'] & ConnectionMySql['connection_data'];
diff --git a/src/entities/connection/utils/index.ts b/src/entities/connection/utils/index.ts
@@ -3,3 +3,6 @@ export * from './adaptConnectionHive';
 export * from './adaptConnectionOracle';
 export * from './adaptConnectionPostgres';
 export * from './adaptConnectionS3';
+export * from './adaptConnectionClickhouse';
+export * from './adaptConnectionMySql';
+export * from './adaptConnectionMsSql';
diff --git a/src/features/connection/ConnectionDetailInfo/utils/getConnectionAuthData/index.ts b/src/features/connection/ConnectionDetailInfo/utils/getConnectionAuthData/index.ts
@@ -16,6 +16,9 @@ export const getConnectionAuthData = ({ data }: GetDescriptionItemsProps): Descr
     case ConnectionType.HDFS:
     case ConnectionType.ORACLE:
     case ConnectionType.POSTGRES:
+    case ConnectionType.CLICKHOUSE:
+    case ConnectionType.MY_SQL:
+    case ConnectionType.MS_SQL:
       return [
         {
           label: 'User',
diff --git a/src/features/connection/ConnectionDetailInfo/utils/getConnectionData/index.ts b/src/features/connection/ConnectionDetailInfo/utils/getConnectionData/index.ts
@@ -21,6 +21,9 @@ export const getConnectionData = ({ data }: GetDescriptionItemsProps): Descripti
         },
       ];
     case ConnectionType.POSTGRES:
+    case ConnectionType.CLICKHOUSE:
+    case ConnectionType.MY_SQL:
+    case ConnectionType.MS_SQL:
       return [
         {
           label: 'Database name',
diff --git a/src/features/connection/utils/adaptConnectionTypeRequest/index.ts b/src/features/connection/utils/adaptConnectionTypeRequest/index.ts
@@ -1,7 +1,10 @@
 import { ConnectionType } from '@shared/types';
 import {
+  adaptConnectionClickhouse,
   adaptConnectionHdfs,
   adaptConnectionHive,
+  adaptConnectionMsSql,
+  adaptConnectionMySql,
   adaptConnectionOracle,
   adaptConnectionPostgres,
   adaptConnectionS3,
@@ -21,6 +24,12 @@ export const adaptConnectionTypeRequest = (values: AdaptConnectionTypeRequestPro
       return adaptConnectionOracle(values);
     case ConnectionType.POSTGRES:
       return adaptConnectionPostgres(values);
+    case ConnectionType.CLICKHOUSE:
+      return adaptConnectionClickhouse(values);
+    case ConnectionType.MY_SQL:
+      return adaptConnectionMySql(values);
+    case ConnectionType.MS_SQL:
+      return adaptConnectionMsSql(values);
     case ConnectionType.S3:
       return adaptConnectionS3(values);
   }
diff --git a/src/features/transfer/MutateTransferForm/components/SourceParams/constants.tsx b/src/features/transfer/MutateTransferForm/components/SourceParams/constants.tsx
@@ -6,11 +6,17 @@ import { TransferConnectionHive } from '../TransferConnectionHive';
 import { TransferConnectionOracle } from '../TransferConnectionOracle';
 import { TransferConnectionPostgres } from '../TransferConnectionPostgres';
 import { TransferConnectionS3 } from '../TransferConnectionS3';
+import { TransferConnectionClickhouse } from '../TransferConnectionClickhouse';
+import { TransferConnectionMySql } from '../TransferConnectionMySql';
+import { TransferConnectionMsSql } from '../TransferConnectionMsSql';
 
 export const TRANSFER_SOURCE_CONNECTION_TYPE_COMPONENT = {
   [ConnectionType.HDFS]: <TransferConnectionHdfs name="source_params" />,
   [ConnectionType.HIVE]: <TransferConnectionHive name="source_params" />,
   [ConnectionType.ORACLE]: <TransferConnectionOracle name="source_params" />,
   [ConnectionType.POSTGRES]: <TransferConnectionPostgres name="source_params" />,
+  [ConnectionType.CLICKHOUSE]: <TransferConnectionClickhouse name="source_params" />,
+  [ConnectionType.MY_SQL]: <TransferConnectionMySql name="source_params" />,
+  [ConnectionType.MS_SQL]: <TransferConnectionMsSql name="source_params" />,
   [ConnectionType.S3]: <TransferConnectionS3 name="source_params" />,
 };
diff --git a/src/features/transfer/MutateTransferForm/components/TargetParams/constants.tsx b/src/features/transfer/MutateTransferForm/components/TargetParams/constants.tsx
@@ -6,11 +6,17 @@ import { TransferConnectionHive } from '../TransferConnectionHive';
 import { TransferConnectionOracle } from '../TransferConnectionOracle';
 import { TransferConnectionPostgres } from '../TransferConnectionPostgres';
 import { TransferConnectionS3 } from '../TransferConnectionS3';
+import { TransferConnectionClickhouse } from '../TransferConnectionClickhouse';
+import { TransferConnectionMySql } from '../TransferConnectionMySql';
+import { TransferConnectionMsSql } from '../TransferConnectionMsSql';
 
 export const TRANSFER_TARGET_CONNECTION_TYPE_COMPONENT = {
   [ConnectionType.HDFS]: <TransferConnectionHdfs name="target_params" />,
   [ConnectionType.HIVE]: <TransferConnectionHive name="target_params" />,
   [ConnectionType.ORACLE]: <TransferConnectionOracle name="target_params" />,
   [ConnectionType.POSTGRES]: <TransferConnectionPostgres name="target_params" />,
+  [ConnectionType.CLICKHOUSE]: <TransferConnectionClickhouse name="target_params" />,
+  [ConnectionType.MY_SQL]: <TransferConnectionMySql name="target_params" />,
+  [ConnectionType.MS_SQL]: <TransferConnectionMsSql name="target_params" />,
   [ConnectionType.S3]: <TransferConnectionS3 name="target_params" />,
 };
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionClickhouse/index.tsx b/src/features/transfer/MutateTransferForm/components/TransferConnectionClickhouse/index.tsx
@@ -0,0 +1,12 @@
+import React from 'react';
+import { Form, Input } from 'antd';
+
+import { TransferConnectionClickhouseProps } from './types';
+
+export const TransferConnectionClickhouse = ({ name }: TransferConnectionClickhouseProps) => {
+  return (
+    <Form.Item label="Table name" name={[name, 'table_name']} rules={[{ required: true }]}>
+      <Input size="large" />
+    </Form.Item>
+  );
+};
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionClickhouse/types.ts b/src/features/transfer/MutateTransferForm/components/TransferConnectionClickhouse/types.ts
@@ -0,0 +1,3 @@
+import { ConnectionParamFieldName } from '../../types';
+
+export interface TransferConnectionClickhouseProps extends ConnectionParamFieldName {}
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionMsSql/index.tsx b/src/features/transfer/MutateTransferForm/components/TransferConnectionMsSql/index.tsx
@@ -0,0 +1,12 @@
+import React from 'react';
+import { Form, Input } from 'antd';
+
+import { TransferConnectionMsSqlProps } from './types';
+
+export const TransferConnectionMsSql = ({ name }: TransferConnectionMsSqlProps) => {
+  return (
+    <Form.Item label="Table name" name={[name, 'table_name']} rules={[{ required: true }]}>
+      <Input size="large" />
+    </Form.Item>
+  );
+};
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionMsSql/types.ts b/src/features/transfer/MutateTransferForm/components/TransferConnectionMsSql/types.ts
@@ -0,0 +1,3 @@
+import { ConnectionParamFieldName } from '../../types';
+
+export interface TransferConnectionMsSqlProps extends ConnectionParamFieldName {}
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionMySql/index.tsx b/src/features/transfer/MutateTransferForm/components/TransferConnectionMySql/index.tsx
@@ -0,0 +1,12 @@
+import React from 'react';
+import { Form, Input } from 'antd';
+
+import { TransferConnectionMySqlProps } from './types';
+
+export const TransferConnectionMySql = ({ name }: TransferConnectionMySqlProps) => {
+  return (
+    <Form.Item label="Table name" name={[name, 'table_name']} rules={[{ required: true }]}>
+      <Input size="large" />
+    </Form.Item>
+  );
+};
diff --git a/src/features/transfer/MutateTransferForm/components/TransferConnectionMySql/types.ts b/src/features/transfer/MutateTransferForm/components/TransferConnectionMySql/types.ts
diff --git a/src/shared/types/connection.ts b/src/shared/types/connection.ts

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+import { ConnectionMsSql } from '../../api';`
	`2`	`+`
	`3`	`+export type AdaptConnectionMsSqlProps = ConnectionMsSql['auth_data'] & ConnectionMsSql['connection_data'];`