Fix getTotalCount()

Author: lav45

src/ActiveDataProvider.php

@@ -53,6 +53,17 @@ public function getQueryResults()
         return $this->_queryResults;
     }
 
+    /**
+     * @inheritdoc
+     */
+    public function prepare($forcePrepare = false)
+    {
+        if ($forcePrepare) {
+            $this->setTotalCount(null);
+        }
+        parent::prepare($forcePrepare);
+    }
+
     /**
      * @return array all aggregations results
      */
@@ -122,6 +133,9 @@ protected function prepareModels()
 
         if (is_array(($results = $query->search($this->db)))) {
             $this->setQueryResults($results);
+            if ($pagination !== false) {
+                $pagination->totalCount = $this->getTotalCount();
+            }
             return $results['hits']['hits'];
         }
         $this->setQueryResults([]);

src/ElasticsearchTarget.php

@@ -139,6 +139,7 @@ public function collect($messages, $final)
     public function prepareMessage($message)
     {
         list($text, $level, $category, $timestamp) = $message;
+        $timestamp = (int)round($timestamp);
 
         $result = [
             'category' => $category,

tests/ActiveDataProviderTest.php

@@ -150,11 +150,15 @@ public function testTotalCountAfterSearch()
         ]);
 
         $pagination = $provider->getPagination();
+        $this->assertEquals(0, $pagination->getPageCount());
+        $this->assertCount(2, $provider->getModels());
         $this->assertEquals(2, $pagination->getPageCount());
-        $this->assertEquals(3, $pagination->getTotalCount());
+        $this->assertEquals(3, $provider->getTotalCount());
 
         $query->andWhere(['name' => 'user2']);
+        $provider->prepare(true);
+        $this->assertCount(1, $provider->getModels());
         $this->assertEquals(1, $pagination->getPageCount());
-        $this->assertEquals(1, $pagination->getTotalCount());
+        $this->assertEquals(1, $provider->getTotalCount());
     }
 }