fix(submitBtn): added setQueryToLoading for Submit component

influxdata · Jun 9, 2020 · a585d98 · a585d98
1 parent f926775
commit a585d98
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/ui/src/notebooks/components/header/Submit.tsx b/ui/src/notebooks/components/header/Submit.tsx
@@ -18,7 +18,7 @@ const COMMENT_REMOVER = /(\/\*([\s\S]*?)\*\/)|(\/\/(.*)$)/gm
 
 export const Submit: FC = () => {
   const {query} = useContext(QueryContext)
-  const {onNotify} = useContext(SubmitQueryButtonContext)
+  const {onNotify, setQueryToLoading} = useContext(SubmitQueryButtonContext)
   const {id, pipes, updateResult, updateMeta} = useContext(NotebookContext)
   const {timeContext} = useContext(TimeContext)
   const time = timeContext[id]
@@ -86,6 +86,7 @@ export const Submit: FC = () => {
     <SubmitQueryButton
       text="Run Flow"
       onNotify={onNotify}
+      setQueryToLoading={setQueryToLoading}
       icon={IconFont.Play}
       submitButtonDisabled={false}
       queryStatus={RemoteDataState.NotStarted}

diff --git a/ui/src/notebooks/context/query.tsx b/ui/src/notebooks/context/query.tsx
@@ -35,10 +35,12 @@ export const QueryContext = React.createContext<QueryContextType>(
 
 export interface SubmitQueryButtonContextType {
   onNotify: () => void
+  setQueryToLoading: () => void
 }
 
 export const DEFAULT_SUBMIT_BTN_CONTEXT: SubmitQueryButtonContextType = {
   onNotify: () => {},
+  setQueryToLoading: () => {},
 }
 
 export const SubmitQueryButtonContext = React.createContext<

diff --git a/ui/src/timeMachine/components/SubmitQueryButton.tsx b/ui/src/timeMachine/components/SubmitQueryButton.tsx
@@ -32,7 +32,7 @@ interface StateProps {
 }
 
 interface DispatchProps {
-  setQueryToLoading: typeof setQueryToLoading
+  setQueryToLoading: typeof setQueryToLoading | (() => void)
   onSubmit: typeof saveAndExecuteQueries | (() => void)
   onNotify: typeof notify | (() => void)
 }