Refactor duplicate mapping of response.

Author: yajra

src/Services/DataTable.php

@@ -144,13 +144,9 @@ protected function filename()
      */
     protected function getDataForExport()
     {
-        return array_map(function ($row) {
-            if ($columns = $this->exportColumns()) {
-                return (new DataTransformer())->transform($row, $columns, 'exportable');
-            }
+        $columns = $this->exportColumns();
 
-            return $row;
-        }, $this->getAjaxResponseData());
+        return $this->mapResponseToColumns($columns, 'exportable');
     }
 
     /**
@@ -193,6 +189,24 @@ public function builder()
         return $this->datatables->getHtmlBuilder();
     }
 
+    /**
+     * Map ajax response to columns definition.
+     *
+     * @param mixed $columns
+     * @param string $type
+     * @return array
+     */
+    protected function mapResponseToColumns($columns, $type)
+    {
+        return array_map(function ($row) use ($columns, $type) {
+            if ($columns) {
+                return (new DataTransformer())->transform($row, $columns, $type);
+            }
+
+            return $row;
+        }, $this->getAjaxResponseData());
+    }
+
     /**
      * Get decorated data as defined in datatables ajax response.
      *
@@ -248,13 +262,9 @@ public function printPreview()
      */
     protected function getDataForPrint()
     {
-        return array_map(function ($row) {
-            if ($columns = $this->printColumns()) {
-                return (new DataTransformer())->transform($row, $columns, 'printable');
-            }
+        $columns = $this->printColumns();
 
-            return $row;
-        }, $this->getAjaxResponseData());
+        return $this->mapResponseToColumns($columns, 'printable');
     }
 
     /**