(cloud) Should make searching function again for

This commit is contained in:
2023-03-21 11:35:35 -07:00
parent ab9cb15e24
commit 4c90e72a1f
4 changed files with 52 additions and 42 deletions

View File

@@ -1,7 +1,6 @@
(ns auto-ap.graphql.accounts (ns auto-ap.graphql.accounts
(:require (:require
[auto-ap.datomic :refer [audit-transact conn]] [auto-ap.datomic :refer [audit-transact conn]]
[iol-ion.tx :refer [upsert-entity]]
[auto-ap.datomic.accounts :as d-accounts] [auto-ap.datomic.accounts :as d-accounts]
[auto-ap.graphql.utils [auto-ap.graphql.utils
:refer [->graphql :refer [->graphql
@@ -13,11 +12,10 @@
is-admin? is-admin?
result->page]] result->page]]
[auto-ap.search :as search] [auto-ap.search :as search]
[auto-ap.utils :refer [heartbeat]]
[datomic.client.api :as dc] [datomic.client.api :as dc]
[clojure.string :as str]) [mount.core :as mount]
(:import [yang.scheduler :as scheduler]))
(org.apache.lucene.search TermInSetQuery)
(org.apache.lucene.util BytesRef)))
(defn get-graphql [context args _] (defn get-graphql [context args _]
(assert-admin (:id context)) (assert-admin (:id context))
@@ -195,3 +193,8 @@
#_(dc/pull (dc/db conn) '[*] [:db/ident :account-client-override/k]) #_(dc/pull (dc/db conn) '[*] [:db/ident :account-client-override/k])
#_{:clj-kondo/ignore [:clojure-lsp/unused-public-var]}
(mount/defstate indexer
:start (scheduler/every (* 5 60 1000) (heartbeat rebuild-search-index "rebuild-search-index"))
:stop (scheduler/stop indexer))

View File

@@ -1,13 +1,17 @@
(ns auto-ap.graphql.clients (ns auto-ap.graphql.clients
(:require (:require
[amazonica.aws.s3 :as s3] [amazonica.aws.s3 :as s3]
[iol-ion.tx :refer [upsert-entity random-tempid]]
[auto-ap.datomic :refer [audit-transact conn]] [auto-ap.datomic :refer [audit-transact conn]]
[auto-ap.datomic.clients :as d-clients] [auto-ap.datomic.clients :as d-clients]
[auto-ap.square.core :as square]
[auto-ap.graphql.utils [auto-ap.graphql.utils
:refer [->graphql assert-admin can-see-client? is-admin? attach-tracing-resolvers]] :refer [->graphql
assert-admin
attach-tracing-resolvers
can-see-client?
is-admin?]]
[auto-ap.routes.queries :as q] [auto-ap.routes.queries :as q]
[auto-ap.square.core :as square]
[auto-ap.utils :refer [heartbeat]]
[clj-time.coerce :as coerce] [clj-time.coerce :as coerce]
[clojure.java.io :as io] [clojure.java.io :as io]
[clojure.set :as set] [clojure.set :as set]
@@ -15,8 +19,9 @@
[clojure.tools.logging :as log] [clojure.tools.logging :as log]
[datomic.client.api :as dc] [datomic.client.api :as dc]
[unilog.context :as lc] [unilog.context :as lc]
[com.brunobonacci.mulog :as mu] [mount.core :as mount]
[datomic.client.api :as dc]) [yang.scheduler :as scheduler]
[iol-ion.tx :refer [random-tempid]])
(:import (:import
(java.util UUID) (java.util UUID)
(org.apache.commons.codec.binary Base64))) (org.apache.commons.codec.binary Base64)))
@@ -598,3 +603,8 @@
:input-objects input-objects :input-objects input-objects
:enums enums}) :enums enums})
(attach-tracing-resolvers resolvers))) (attach-tracing-resolvers resolvers)))
#_{:clj-kondo/ignore [:clojure-lsp/unused-public-var]}
(mount/defstate indexer
:start (scheduler/every (* 5 60 1000) (heartbeat d-clients/rebuild-search-index "rebuild-search-index"))
:stop (scheduler/stop indexer))

View File

@@ -11,11 +11,14 @@
enum->keyword enum->keyword
is-admin? is-admin?
result->page]] result->page]]
[auto-ap.search :as search]
[auto-ap.utils :refer [heartbeat]]
[clojure.set :as set] [clojure.set :as set]
[clojure.string :as str] [clojure.string :as str]
[clojure.tools.logging :as log] [clojure.tools.logging :as log]
[datomic.client.api :as dc] [datomic.client.api :as dc]
[auto-ap.search :as search])) [yang.scheduler :as scheduler]
[mount.core :as mount]))
(defn can-user-edit-vendor? [vendor-id id] (defn can-user-edit-vendor? [vendor-id id]
(if (is-admin? id) (if (is-admin? id)
@@ -183,30 +186,27 @@
matches)) matches))
(defn search [context args _] (defn search [context args _]
(comment (let [search-query (cleanse-query (:query args))] (let [search-query (cleanse-query (:query args))]
(for [[id name] (search/search (cond-> {:q search-query} (for [[id name] (search/search (cond-> {:q search-query}
(not (is-admin? (:id context))) (assoc :hidden false)) (not (is-admin? (:id context))) (assoc :hidden false))
"vendors")] "vendor")]
{:name name {:name name
:id (Long/parseLong id)}) :id id})
)) ))
(if-let [search-query (cleanse-query (:query args))]
(let [data (if (is-admin? (:id context)) (defn rebuild-search-index []
(dc/q '[:find ?n ?i ?s (search/full-index-query
:in $ ?q (for [result (map first (dc/qseq '[:find (pull ?v [:vendor/search-terms :db/id :vendor/name :vendor/hidden])
:where [(fulltext $ :vendor/search-terms ?q) [[?i ?n _ ?s]]]] :in $
(dc/db conn) :where [?v :vendor/search-terms ]]
search-query) (dc/db conn)))]
(dc/q '[:find ?n ?i ?s {:id (:db/id result)
:in $ ?q :text (or (first (:vendor/search-terms result))
:where [(fulltext $ :vendor/search-terms ?q) [[?i ?n _ ?s]]] (:vendor/name result))
(not [?i :vendor/hidden true])] :hidden (boolean (:vendor/hidden result))})
(dc/db conn) "vendor"))
search-query))]
(->> data #_{:clj-kondo/ignore [:clojure-lsp/unused-public-var]}
(sort-by (comp - last)) (mount/defstate indexer
(partial-match-first (:query args)) :start (scheduler/every (* 5 60 1000) (heartbeat rebuild-search-index "rebuild-search-index"))
(map (fn [[n i]] :stop (scheduler/stop indexer))
{:name n
:id i}))))
[]))

View File

@@ -10,7 +10,7 @@
(org.apache.lucene.store FSDirectory))) (org.apache.lucene.store FSDirectory)))
(defn full-index-query [results index-name] (defn full-index-query [results index-name]
(let [directory (FSDirectory/open (Paths/get (java.net.URI. (str "file:///tmp/search" (:dd-env env) "/" index-name)))) (let [directory (FSDirectory/open (Paths/get (java.net.URI. (str "file:///tmp/search/" (:dd-env env) "/" index-name))))
analyzer (StandardAnalyzer.) analyzer (StandardAnalyzer.)
index-writer-config (IndexWriterConfig. analyzer) index-writer-config (IndexWriterConfig. analyzer)
index-writer (IndexWriter. directory index-writer-config)] index-writer (IndexWriter. directory index-writer-config)]
@@ -22,10 +22,7 @@
(.add (TextField. "name" text Field$Store/YES)) (.add (TextField. "name" text Field$Store/YES))
(.add (StoredField. "id" (long id))))]] (.add (StoredField. "id" (long id))))]]
(doseq [k (filter (complement #{:text :id}) (keys x))] (doseq [k (filter (complement #{:text :id}) (keys x))]
(println "K" (name k) (get x k))
(.add doc (StringField. (name k) (str (get x k)) Field$Store/YES))) (.add doc (StringField. (name k) (str (get x k)) Field$Store/YES)))
(println "adding" text)
(flush)
(.addDocument index-writer doc)) (.addDocument index-writer doc))
(finally (finally
(.close index-writer))))) (.close index-writer)))))