From 9eccb8e416786c607d683998cf0f82a21927205f Mon Sep 17 00:00:00 2001
From: Duke Manh <51073515+DukeManh@users.noreply.github.com>
Date: Mon, 18 Oct 2021 06:36:13 -0400
Subject: [PATCH] CodeEditor: Trigger the lastest getSuggestions() passed to
 CodeEditor (#40544)

---
 packages/grafana-ui/src/components/Monaco/CodeEditor.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/grafana-ui/src/components/Monaco/CodeEditor.tsx b/packages/grafana-ui/src/components/Monaco/CodeEditor.tsx
index ec056ad9cc5..2c265b83618 100644
--- a/packages/grafana-ui/src/components/Monaco/CodeEditor.tsx
+++ b/packages/grafana-ui/src/components/Monaco/CodeEditor.tsx
@@ -31,7 +31,10 @@ class UnthemedCodeEditor extends React.PureComponent<Props> {
   componentDidUpdate(oldProps: Props) {
     const { getSuggestions, language } = this.props;
 
-    if (language !== oldProps.language) {
+    const newLanguage = oldProps.language !== language;
+    const newGetSuggestions = oldProps.getSuggestions !== getSuggestions;
+
+    if (newGetSuggestions || newLanguage) {
       if (this.completionCancel) {
         this.completionCancel.dispose();
       }
@@ -44,7 +47,9 @@ class UnthemedCodeEditor extends React.PureComponent<Props> {
       if (getSuggestions) {
         this.completionCancel = registerSuggestions(this.monaco, language, getSuggestions);
       }
+    }
 
+    if (newLanguage) {
       this.loadCustomLanguage();
     }
   }