SQL Datasources: Update behavior of default connection limits (#66687)

* Update behavior of defaults in connection limits

* Refactor to use config object instead

* Refactor remove unneeded function

---------

Co-authored-by: Zoltán Bedi <zoltan.bedi@gmail.com>

public/app/features/plugins/sql/components/configuration/ConnectionLimits.tsx

@@ -1,10 +1,10 @@
 import React from 'react';
 
 import { DataSourceSettings } from '@grafana/data';
+import { config } from '@grafana/runtime';
 import { FieldSet, InlineField, InlineFieldRow, InlineSwitch } from '@grafana/ui';
 import { NumberInput } from 'app/core/components/OptionsUI/NumberInput';
 
-import { SQLConnectionDefaults } from '../../constants';
 import { SQLConnectionLimits, SQLOptions } from '../../types';
 
 interface Props<T> {
@@ -47,7 +47,7 @@ export const ConnectionLimits = <T extends SQLConnectionLimits>(props: Props<T>)
         maxOpenConns: number,
         maxIdleConns: number,
       });
-    } else if (number !== undefined) {
+    } else {
       updateJsonData({
         maxOpenConns: number,
       });
@@ -68,10 +68,10 @@ export const ConnectionLimits = <T extends SQLConnectionLimits>(props: Props<T>)
       if (jsonData.maxOpenConns !== undefined) {
         maxConns = jsonData.maxOpenConns;
         idleConns = jsonData.maxOpenConns;
-      } else {
-        maxConns = SQLConnectionDefaults.MAX_CONNS;
-        idleConns = SQLConnectionDefaults.MAX_CONNS;
       }
+    } else {
+      maxConns = jsonData.maxOpenConns;
+      idleConns = jsonData.maxIdleConns;
     }
 
     updateJsonData({
@@ -124,7 +124,7 @@ export const ConnectionLimits = <T extends SQLConnectionLimits>(props: Props<T>)
             <span>
               If enabled, automatically set the number of <i>Maximum idle connections</i> to the same value as
               <i> Max open connections</i>. If the number of maximum open connections is not set it will be set to the
-              default ({SQLConnectionDefaults.MAX_CONNS}).
+              default ({config.sqlConnectionLimits.maxIdleConns}).
             </span>
           }
         >

public/app/features/plugins/sql/components/configuration/useMigrateDatabaseFields.test.ts

@@ -2,11 +2,23 @@ import { renderHook } from '@testing-library/react-hooks';
 
 import { DataSourceSettings } from '@grafana/data';
 
-import { SQLConnectionDefaults } from '../../constants';
 import { SQLOptions } from '../../types';
 
 import { useMigrateDatabaseFields } from './useMigrateDatabaseFields';
 
+jest.mock('@grafana/runtime', () => {
+  return {
+    config: {
+      sqlConnectionLimits: {
+        maxOpenConns: 10,
+        maxIdleConns: 11,
+        connMaxLifetime: 12,
+      },
+    },
+    logDebug: jest.fn(),
+  };
+});
+
 describe('Database Field Migration', () => {
   let defaultProps = {
     options: {
@@ -57,8 +69,9 @@ describe('Database Field Migration', () => {
       ...defaultProps,
       onOptionsChange: (options: DataSourceSettings) => {
         const jsonData = options.jsonData as SQLOptions;
-        expect(jsonData.maxOpenConns).toBe(SQLConnectionDefaults.MAX_CONNS);
-        expect(jsonData.maxIdleConns).toBe(Math.ceil(SQLConnectionDefaults.MAX_CONNS));
+        expect(jsonData.maxOpenConns).toBe(10);
+        expect(jsonData.maxIdleConns).toBe(11);
+        expect(jsonData.connMaxLifetime).toBe(12);
         expect(jsonData.maxIdleConnsAuto).toBe(true);
       },
     };

public/app/features/plugins/sql/components/configuration/useMigrateDatabaseFields.ts

@@ -1,9 +1,8 @@
 import { useEffect } from 'react';
 
 import { DataSourcePluginOptionsEditorProps } from '@grafana/data';
-import { logDebug } from '@grafana/runtime';
+import { logDebug, config } from '@grafana/runtime';
 
-import { SQLConnectionDefaults } from '../../constants';
 import { SQLOptions } from '../../types';
 
 /**
@@ -34,9 +33,7 @@ export function useMigrateDatabaseFields<T extends SQLOptions, S = {}>({
       jsonData.maxIdleConns === undefined &&
       jsonData.maxIdleConnsAuto === undefined
     ) {
-      // It's expected that the default will be greater than 4
-      const maxOpenConns = SQLConnectionDefaults.MAX_CONNS;
-      const maxIdleConns = maxOpenConns;
+      const { maxOpenConns, maxIdleConns } = config.sqlConnectionLimits;
 
       logDebug(
         `Setting default max open connections to ${maxOpenConns} and setting max idle connection to ${maxIdleConns}`
@@ -55,6 +52,21 @@ export function useMigrateDatabaseFields<T extends SQLOptions, S = {}>({
       optionsUpdated = true;
     }
 
+    // If the maximum connection lifetime hasn't been
+    // otherwise set fill in with the default from configuration
+    if (jsonData.connMaxLifetime === undefined) {
+      const { connMaxLifetime } = config.sqlConnectionLimits;
+
+      // Spread new options and add our value
+      newOptions.jsonData = {
+        ...newOptions.jsonData,
+        connMaxLifetime: connMaxLifetime,
+      };
+
+      // Note that we've updated the options
+      optionsUpdated = true;
+    }
+
     // Only issue an update if we changed options
     if (optionsUpdated) {
       onOptionsChange(newOptions);

public/app/features/plugins/sql/constants.ts

@@ -15,11 +15,3 @@ export const MACRO_NAMES = [
   '$__unixEpochGroup',
   '$__unixEpochGroupAlias',
 ];
-
-/**
- * Constants for SQL connection
- * parameters and automatic settings
- */
-export const SQLConnectionDefaults = {
-  MAX_CONNS: 100,
-};