Add weight on field

Author: zawaze

Config/module.xml

@@ -13,7 +13,7 @@
         <language>en_US</language>
         <language>fr_FR</language>
     </languages>
-    <version>2.1.3</version>
+    <version>2.1.4</version>
     <authors>
         <author>
             <name>Nicolas Barbey</name>

Index/BaseIndex.php

@@ -15,16 +15,28 @@
 
 abstract class BaseIndex implements TntSearchIndexInterface
 {
+    const FIELD_WEIGHT = [
+        'title' => 10,
+        'chapo' => 5
+    ];
+
     public function __construct(
         protected EventDispatcherInterface $disptacher,
         protected TntSearchProvider        $tntSearchProvider
-    ) {}
+    )
+    {
+    }
 
     public function getTokenizer(): string
     {
         return Tokenizer::class;
     }
 
+    public function getFieldWeights($field): int
+    {
+        return self::FIELD_WEIGHT[$field] ?? 1;
+    }
+
     public function getIndexName(): string
     {
         $reflectionClass = new ReflectionClass(get_called_class());
@@ -75,8 +87,11 @@ protected function indexOneIndex(string $locale = null): void
 
             $this->disptacher->dispatch($extendQueryEvent, ExtendQueryEvent::EXTEND_QUERY . $indexName);
 
+            $tntIndexer->setIndexObject($this);
+
             $tntIndexer->query($extendQueryEvent->getQuery());
             $tntIndexer->run();
+
         } catch (Exception $ex) {
             Tlog::getInstance()->addError("Error indexation on index $indexFileName : " . $ex->getMessage());
         }

Index/Product.php

@@ -8,6 +8,10 @@
 
 class Product extends BaseIndex
 {
+    const FIELD_WEIGHT = [
+        'title' => 10,
+    ];
+
     public function isTranslatable(): bool
     {
         return true;

Index/TntSearchIndexInterface.php

@@ -4,6 +4,8 @@
 
 interface TntSearchIndexInterface
 {
+    public function getFieldWeights(string $field): int;
+
     public function isTranslatable(): bool;
 
     public function index(): void;

Service/Support/TntIndexer.php

@@ -4,9 +4,12 @@
 
 use TeamTNT\TNTSearch\Indexer\TNTIndexer as BaseTNTIndexer;
 use TntSearch\Connector\PropelConnector;
+use TntSearch\Index\TntSearchIndexInterface;
 
 class TntIndexer extends BaseTNTIndexer
 {
+    protected TntSearchIndexInterface $indexObject;
+
     /**
      * Override tu use propel instance instead of dsn.
      */
@@ -15,6 +18,63 @@ public function createConnector(array $config): PropelConnector
         return new PropelConnector();
     }
 
+    public function saveWordlist($stems): array
+    {
+        $terms = [];
+        $stems->map(function ($column, $key) use (&$terms) {
+            $weight = $this->indexObject->getFieldWeights($key);
+            foreach ($column as $term) {
+                if (array_key_exists($term, $terms)) {
+                    $terms[$term]['hits'] = (int) $terms[$term]['hits'] * $weight;
+                    $terms[$term]['docs'] = 1;
+                } else {
+                    $terms[$term] = [
+                        'hits' => 1 * $weight,
+                        'docs' => 1,
+                        'id' => 0
+                    ];
+                }
+            }
+        });
+
+        foreach ($terms as $key => $term) {
+            try {
+                $this->insertWordlistStmt->bindParam(":keyword", $key);
+                $this->insertWordlistStmt->bindParam(":hits", $term['hits']);
+                $this->insertWordlistStmt->bindParam(":docs", $term['docs']);
+                $this->insertWordlistStmt->execute();
+
+                $terms[$key]['id'] = $this->index->lastInsertId();
+                if ($this->inMemory) {
+                    $this->inMemoryTerms[$key] = $terms[$key]['id'];
+                }
+            } catch (\Exception $e) {
+                if ($e->getCode() == 23000) {
+                    $this->updateWordlistStmt->bindValue(':docs', $term['docs']);
+                    $this->updateWordlistStmt->bindValue(':hits', $term['hits']);
+                    $this->updateWordlistStmt->bindValue(':keyword', $key);
+                    $this->updateWordlistStmt->execute();
+                    if (!$this->inMemory) {
+                        $this->selectWordlistStmt->bindValue(':keyword', $key);
+                        $this->selectWordlistStmt->execute();
+                        $res = $this->selectWordlistStmt->fetch(PDO::FETCH_ASSOC);
+                        $terms[$key]['id'] = $res['id'];
+                    } else {
+                        $terms[$key]['id'] = $this->inMemoryTerms[$key];
+                    }
+                } else {
+                    echo "Error while saving wordlist: " . $e->getMessage() . "\n";
+                }
+
+                // Statements must be refreshed, because in this state they have error attached to them.
+                $this->statementsPrepared = false;
+                $this->prepareStatementsForIndex();
+
+            }
+        }
+        return $terms;
+    }
+
     /**
      * Allow to handle PDOConnection from propel.
      */
@@ -25,4 +85,9 @@ public function setDatabasePropelConnector($dbh): void
             $this->dbh->setAttribute(\PDO::MYSQL_ATTR_USE_BUFFERED_QUERY, false);
         }
     }
+
+    public function setIndexObject(TntSearchIndexInterface $indexObject): void
+    {
+        $this->indexObject = $indexObject;
+    }
 }