Merge pull request #23 from mhh1422/master

elastic index method

Author: mhh1422

src/Query.php

@@ -190,14 +190,50 @@ protected function __create(array $data, $index, $type, $id = null, array $param
             'index' => $index,
             'type' => $type,
             'body' => $data] + ($id ? ['id' => $id] : []) + $parameters);
-        if ($result['_shards']['failed'] > 0) {
+        if (array_key_exists('_shards', $result) && array_key_exists('failed', $result['_shards']) && $result['_shards']['failed'] > 0) {
             throw new Exception('Failed to create the document. Serialized results: ' + json_encode($result));
         } else {
             $result = $this->__find($result['_index'], $result['_type'], $result['_id']);
         }
         return $result;
     }
 
+    /**
+     * Index a new document
+     * 
+     * @param array $data the document data
+     * @param string $type
+     * @param string|number $id
+     * @param array $parameters
+     * @return array
+     */
+    protected function _index(array $data, $type, $id = null, array $parameters = []) {
+        return $this->__index($data, $this->index(), $type, $id, $parameters);
+    }
+
+    /**
+     * Index a new document
+     * 
+     * @param array $data the document data
+     * @param string $index
+     * @param string $type
+     * @param string|number $id
+     * @param array $parameters
+     * @return array
+     */
+    protected function __index(array $data, $index, $type, $id = null, array $parameters = []) {
+        $result = $this->client->index([
+            'index' => $index,
+            'type' => $type,
+            'body' => $data] + ($id ? ['id' => $id] : []) + $parameters);
+        if (array_key_exists('_shards', $result) && array_key_exists('failed', $result['_shards']) && $result['_shards']['failed'] > 0) {
+            throw new Exception('Failed to index the document. Serialized results: ' + json_encode($result));
+        } else {
+            $result = $this->__find($result['_index'], $result['_type'], $result['_id']);
+        }
+        return $result;
+    }
+
     /**
      * Deletes a document
      * 
@@ -428,6 +464,7 @@ protected static function _allowPublicAccess() {
             'find',
             'meta',
             'create',
+            'index',
             'delete',
             'builder',
             'metaSettings',

src/TypeQueryTrait.php

@@ -67,6 +67,17 @@ protected function _create(array $data, $id = null, array $parameters = []) {
         return parent::_create($data, $this->type(), $id, $parameters);
     }
 
+    /**
+     * Overrides the parent class _index method to omit the type in parameters.
+     * @see Query::_index for details.
+     * 
+     * @param mixed $id
+     * @return Result|Model[]
+     */
+    protected function _index(array $data, $id = null, array $parameters = []) {
+        return parent::_index($data, $this->type(), $id, $parameters);
+    }
+
     /**
      * Creates the full model class name
      *