From 6fc51b029526ca514f1b967807e9375371993705 Mon Sep 17 00:00:00 2001
From: Brandon Farmer <bthesorceror@gmail.com>
Date: Mon, 30 Apr 2018 15:08:23 -0700
Subject: [PATCH] Load table graph state on componentDidMount

Co-authored-by: Deniz Kusefoglu <denizk@gmail.com>
Co-authored-by: Brandon Farmer <bthesorceror@gmail.com>
---
 .../components/CellEditorOverlay.tsx          | 13 +----
 ui/src/shared/components/TableGraph.js        | 55 ++++++++++++++++---
 2 files changed, 51 insertions(+), 17 deletions(-)

diff --git a/ui/src/dashboards/components/CellEditorOverlay.tsx b/ui/src/dashboards/components/CellEditorOverlay.tsx
index 161747421..dfabef295 100644
--- a/ui/src/dashboards/components/CellEditorOverlay.tsx
+++ b/ui/src/dashboards/components/CellEditorOverlay.tsx
@@ -102,12 +102,8 @@ class CellEditorOverlay extends Component<Props, State> {
   constructor(props) {
     super(props)
 
-    const {
-      cell: {legend},
-    } = props
-    let {
-      cell: {queries},
-    } = props
+    const {cell: {legend}} = props
+    let {cell: {queries}} = props
 
     // Always have at least one query
     if (_.isEmpty(queries)) {
@@ -403,10 +399,7 @@ class CellEditorOverlay extends Component<Props, State> {
   }
 
   private get sourceLink(): string {
-    const {
-      cell: {queries},
-      source: {links},
-    } = this.props
+    const {cell: {queries}, source: {links}} = this.props
     return _.get(queries, '0.source.links.self', links.self)
   }
 
diff --git a/ui/src/shared/components/TableGraph.js b/ui/src/shared/components/TableGraph.js
index 098424352..73a65be5a 100644
--- a/ui/src/shared/components/TableGraph.js
+++ b/ui/src/shared/components/TableGraph.js
@@ -53,6 +53,52 @@ class TableGraph extends Component {
     }
   }
 
+  componentDidMount() {
+    const sortField = _.get(
+      this.props,
+      ['tableOptions', 'sortBy', 'internalName'],
+      DEFAULT_TIME_FIELD.internalName
+    )
+    const sort = {field: sortField, direction: DEFAULT_SORT_DIRECTION}
+    const {
+      data,
+      tableOptions,
+      queryASTs,
+      timeFormat,
+      fieldOptions,
+      decimalPlaces,
+    } = this.props
+    const result = timeSeriesToTableGraph(data, queryASTs)
+    const sortedLabels = result.sortedLabels
+
+    const computedFieldOptions = computeFieldOptions(fieldOptions, sortedLabels)
+    const {
+      transformedData,
+      sortedTimeVals,
+      columnWidths,
+      totalWidths,
+    } = transformTableData(
+      data,
+      sort,
+      computedFieldOptions,
+      tableOptions,
+      timeFormat,
+      decimalPlaces
+    )
+
+    this.setState({
+      transformedData,
+      sortedTimeVals,
+      columnWidths,
+      data: result.data,
+      sortedLabels,
+      totalColumnWidths: totalWidths,
+      hoveredColumnIndex: NULL_ARRAY_INDEX,
+      hoveredRowIndex: NULL_ARRAY_INDEX,
+      sort,
+    })
+  }
+
   handleUpdateFieldOptions = fieldOptions => {
     const {isInCEO} = this.props
     if (!isInCEO) {
@@ -161,10 +207,7 @@ class TableGraph extends Component {
   }
 
   handleHover = (columnIndex, rowIndex) => () => {
-    const {
-      handleSetHoverTime,
-      tableOptions: {verticalTimeAxis},
-    } = this.props
+    const {handleSetHoverTime, tableOptions: {verticalTimeAxis}} = this.props
     const {sortedTimeVals} = this.state
     if (verticalTimeAxis && rowIndex === 0) {
       return
@@ -220,9 +263,7 @@ class TableGraph extends Component {
 
   calculateColumnWidth = columnSizerWidth => column => {
     const {index} = column
-    const {
-      tableOptions: {fixFirstColumn},
-    } = this.props
+    const {tableOptions: {fixFirstColumn}} = this.props
     const {transformedData, columnWidths, totalColumnWidths} = this.state
     const columnCount = _.get(transformedData, ['0', 'length'], 0)
     const columnLabel = transformedData[0][index]