PMM-12971 Add pg exporter connection limit

public/app/percona/add-instance/components/AddRemoteInstance/AddRemoteInstance.types.ts

@@ -103,6 +103,7 @@ export interface PostgreSQLPayload extends RemoteCommonPayload, TLSCommon {
   agent_password: string;
   max_query_length: number;
   auto_discovery_limit: number;
+  max_exporter_connections: number;
 }
 
 export interface MySQLPayload extends RemoteCommonPayload, TLSCommon {
@@ -196,6 +197,7 @@ export interface RDSPayload extends CommonRDSAzurePayload {
   metrics_mode: string;
   qan_postgresql_pgstatements: boolean;
   agent_password: string;
+  max_exporter_connections: number;
 }
 
 export interface MSAzurePayload extends CommonRDSAzurePayload {

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/AdditionalOptions/AdditionalOptions.constants.ts

@@ -1,5 +1,9 @@
 import { Messages } from './AdditionalOptions.messages';
-import { AutoDiscoveryOptionsInterface, TablestatOptionsInterface } from './AdditionalOptions.types';
+import {
+  AutoDiscoveryOptionsInterface,
+  TablestatOptionsInterface,
+  MaxConnectionLimitOptionsInterface,
+} from './AdditionalOptions.types';
 
 export const tablestatOptions = [
   {
@@ -30,3 +34,14 @@ export const autoDiscoveryOptions = [
     label: Messages.form.autoDiscoveryOptions.custom,
   },
 ];
+
+export const maxConnectionLimitOptions = [
+  {
+    value: MaxConnectionLimitOptionsInterface.enabled,
+    label: Messages.form.maxConnectionLimitOptions.enabled,
+  },
+  {
+    value: MaxConnectionLimitOptionsInterface.disabled,
+    label: Messages.form.maxConnectionLimitOptions.disabled,
+  },
+];

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/AdditionalOptions/AdditionalOptions.messages.ts

@@ -10,5 +10,9 @@ export const Messages = {
       disabled: 'Disabled',
       custom: 'Custom',
     },
+    maxConnectionLimitOptions: {
+      enabled: 'Enabled',
+      disabled: 'Disabled',
+    },
   },
 };

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/AdditionalOptions/AdditionalOptions.tsx

@@ -14,8 +14,12 @@ import { Messages } from '../FormParts.messages';
 import { getStyles } from '../FormParts.styles';
 import { AdditionalOptionsFormPartProps, PostgreSQLAdditionalOptionsProps } from '../FormParts.types';
 
-import { autoDiscoveryOptions, tablestatOptions } from './AdditionalOptions.constants';
-import { AutoDiscoveryOptionsInterface, TablestatOptionsInterface } from './AdditionalOptions.types';
+import { autoDiscoveryOptions, tablestatOptions, maxConnectionLimitOptions } from './AdditionalOptions.constants';
+import {
+  AutoDiscoveryOptionsInterface,
+  MaxConnectionLimitOptionsInterface,
+  TablestatOptionsInterface,
+} from './AdditionalOptions.types';
 import { MongodbTLSCertificate } from './MongodbTLSCertificate';
 import { MysqlTLSCertificate } from './MysqlTLSCertificate';
 import { PostgreTLSCertificate } from './PostgreTLSCertificate';
@@ -43,19 +47,31 @@ export const AdditionalOptionsFormPart: FC<AdditionalOptionsFormPartProps> = ({
 
 export const PostgreSQLAdditionalOptions: FC<PostgreSQLAdditionalOptionsProps> = ({ form, isRDS, isAzure }) => {
   const selectedOption = form.getState()?.values?.autoDiscoveryOptions;
+  const maxConnSelectedOption = form.getState()?.values?.maxConnectionLimitOptions;
   const [selectedValue, setSelectedValue] = useState<string>(selectedOption || AutoDiscoveryOptionsInterface.enabled);
+  const [maxConnSelectedValue, setMaxConnSelectedValue] = useState<string>(
+    maxConnSelectedOption || MaxConnectionLimitOptionsInterface.disabled
+  );
   const styles = useStyles2(getStyles);
   const validators = [platformCoreValidators.containsNumber, ...platformCoreValidators.int32];
 
   const getAutoDiscoveryLimitValue = (type: AutoDiscoveryOptionsInterface) =>
     type === AutoDiscoveryOptionsInterface.disabled ? -1 : 10;
 
+  const getMaxConnectionLimitValue = (type: MaxConnectionLimitOptionsInterface) => 5;
+
   useEffect(() => {
     setSelectedValue(selectedOption);
     form.change('autoDiscoveryLimit', getAutoDiscoveryLimitValue(selectedOption));
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [selectedOption]);
 
+  useEffect(() => {
+    setMaxConnSelectedValue(maxConnSelectedOption);
+    form.change('maxExporterConnections', getMaxConnectionLimitValue(maxConnSelectedOption));
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [maxConnSelectedOption]);
+
   return (
     <>
       <h4>{Messages.form.labels.trackingOptions}</h4>
@@ -87,6 +103,31 @@ export const PostgreSQLAdditionalOptions: FC<PostgreSQLAdditionalOptionsProps> =
           tooltipText={Messages.form.tooltips.postgresqlDetails.autoDiscoveryLimit}
         />
       </div>
+      {!isAzure && (
+        <div>
+          <h4>{Messages.form.labels.postgresqlDetails.maxConnection}</h4>
+          <div className={styles.group}>
+            <RadioButtonGroupField
+              name="maxConnectionLimitOptions"
+              data-testid="max-connection-limit-radio-button-group"
+              defaultValue={maxConnSelectedValue}
+              options={maxConnectionLimitOptions}
+              className={styles.radioField}
+              label={Messages.form.labels.postgresqlDetails.maxConnectionLimitOptions}
+              fullWidth
+            />
+            <NumberInputField
+              key="maxExporterConnections"
+              name="maxExporterConnections"
+              defaultValue={5}
+              disabled={maxConnSelectedValue !== MaxConnectionLimitOptionsInterface.enabled}
+              validators={validators}
+              label={Messages.form.labels.postgresqlDetails.maxConnectionLimit}
+              tooltipText={Messages.form.tooltips.postgresqlDetails.maxConnectionLimit}
+            />
+          </div>
+        </div>
+      )}
     </>
   );
 };

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/AdditionalOptions/AdditionalOptions.types.ts

@@ -9,3 +9,9 @@ export enum AutoDiscoveryOptionsInterface {
   disabled = 'disabled',
   custom = 'custom',
 }
+
+export enum MaxConnectionLimitOptionsInterface {
+  enabled = 'enabled',
+  disabled = 'disabled',
+  custom = 'custom',
+}

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/FormParts.messages.ts

@@ -32,6 +32,9 @@ export const Messages = {
         autoDiscovery: 'Auto-discovery',
         autoDiscoveryLimit: 'Auto-discovery limit',
         autoDiscoveryLimitOptions: 'State',
+        maxConnection: 'Maximum connection limit',
+        maxConnectionLimit: 'Maximum connection limit per PostgreSQL instance',
+        maxConnectionLimitOptions: 'State',
       },
       mysqlDetails: {
         maxQueryLength: 'Max query length',
@@ -91,6 +94,7 @@ export const Messages = {
         database: 'Database (default: postgres)',
         maxQueryLength: 'Max query length',
         autoDiscoveryLimit: 'Auto-discovery limit',
+        maxConnectionLimit: 'Maximum connection limit per PostgreSQL instance',
       },
       mysqlDetails: {
         maxQueryLength: 'Max query length',
@@ -131,6 +135,8 @@ export const Messages = {
         database: 'Database name',
         maxQueryLength: 'Full Example (Fingerprint) storage is not allowed by default to have the best performance',
         autoDiscoveryLimit: 'Turn off auto-discovery when the total count of databases exceeds the limit.',
+        maxConnectionLimit:
+          'The maximum connections the Exporter can open to each PostgreSQL instance for monitoring. Higher values might be needed for larger or busier instances, but avoid setting it too high as it can impact performance. Adjust based on your instance size and monitoring needs.',
       },
       mysqlDetails: {
         maxQueryLength: 'Full Example (Fingerprint) storage is not allowed by default to have the best performance',

public/app/percona/add-instance/components/AddRemoteInstance/FormParts/PostgreSQLConnectionDetails/PostgreSQLConnectionDetails.test.tsx

@@ -22,4 +22,26 @@ describe('PostgreSQL connection details:: ', () => {
 
     expect(screen.getByTestId('maxQueryLength-text-input')).toHaveValue('1000');
   });
+
+  it('should have max connection field for pg and rds pg', () => {
+    render(
+      <Form
+        onSubmit={jest.fn()}
+        render={() => <PostgreSQLConnectionDetails remoteInstanceCredentials={{ isAzure: false }} />}
+      />
+    );
+    const connectionlimitInput = screen.queryByTestId('maxExporterConnections-text-input');
+    expect(connectionlimitInput).toBeInTheDocument();
+  });
+
+  it('should not have max connection field for Azure', () => {
+    render(
+      <Form
+        onSubmit={jest.fn()}
+        render={() => <PostgreSQLConnectionDetails remoteInstanceCredentials={{ isAzure: true }} />}
+      />
+    );
+    const connectionlimitInput = screen.queryByTestId('maxExporterConnections-text-input');
+    expect(connectionlimitInput).not.toBeInTheDocument();
+  });
 });