Merge pull request #460 from NYPL/main

main -> QA subject_prefix

Author: danamansana

.gitignore

@@ -44,3 +44,6 @@ build
 .elasticbeanstalk/*
 !.elasticbeanstalk/*.cfg.yml
 !.elasticbeanstalk/*.global.yml
+
+
+config/local*

lib/api-request.js

@@ -69,6 +69,10 @@ class ApiRequest {
       .some((userParam) => ApiRequest.IDENTIFIER_NUMBER_PARAMS.includes(userParam))
   }
 
+  hasSubjectPrefix () {
+    return this.params.subject_prefix
+  }
+
   static fromParams (params) {
     return new ApiRequest(params)
   }

lib/elasticsearch/elastic-query-builder.js

@@ -88,6 +88,10 @@ class ElasticQueryBuilder {
     if (this.request.hasIdentifierNumberParam()) {
       this.requireSpecificIdentifierMatch()
     }
+
+    if (this.request.hasSubjectPrefix()) {
+      this.applySubjectPrefix()
+    }
   }
 
   /**
@@ -156,6 +160,19 @@ class ElasticQueryBuilder {
     // TODO: Boost on subjectLiteral prefix and term matching
   }
 
+  applySubjectPrefix () {
+    const q = this.request.params.subject_prefix
+    this.query.addMust({
+      bool: {
+        should: [
+          prefixMatch('subjectLiteral.raw', q),
+          prefixMatch('parallelSubjectLiteral.raw', q)
+        ]
+      }
+    })
+    this.boostSubjectMatches()
+  }
+
   /**
   * Build ES query for standard_number searches
   */
@@ -328,6 +345,17 @@ class ElasticQueryBuilder {
     ])
   }
 
+  /**
+  * Boost bibs with strong subjectLiteral matches
+  **/
+  boostSubjectMatches () {
+    const q = this.request.params.subject_prefix
+    this.query.addShoulds([
+      termMatch('subjectLiteral.raw', q, 50),
+      termMatch('parallelSubjectLiteral.raw', q, 50)
+    ])
+  }
+
   /**
   * Boost bibs with strong creator/contributor matches
   */

lib/resources.js

@@ -93,6 +93,7 @@ const parseSearchParams = function (params) {
     contributor: { type: 'string' },
     title: { type: 'string' },
     subject: { type: 'string' },
+    subject_prefix: { type: 'string' },
     isbn: { type: 'string' },
     issn: { type: 'string' },
     lccn: { type: 'string' },
@@ -620,8 +621,11 @@ module.exports = function (app, _private = null) {
 
   // Conduct a search across resources:
   app.resources.search = function (params, opts, request) {
+    app.logger.debug('Unparsed params: ', params)
     params = parseSearchParams(params)
 
+    app.logger.debug('Parsed params: ', params)
+
     let body = buildElasticBody(params)
 
     // Strip unnecessary _source fields

test/elastic-query-builder.test.js

@@ -340,6 +340,50 @@ describe('ElasticQueryBuilder', () => {
       })
     })
 
+    describe('subject_prefix=', () => {
+      it('applies subject_prefix clauses to query', () => {
+        const request = new ApiRequest({ subject_prefix: 'toast' })
+        const inst = ElasticQueryBuilder.forApiRequest(request)
+
+        const query = inst.query.toJson()
+
+        expect(query.bool.must[0].bool.should.length).to.equal(2)
+        expect(query.bool.must[0].bool.should[0])
+        expect(query.bool.must[0].bool.should[0]).to.deep.equal({
+          prefix: {
+            'subjectLiteral.raw': {
+              value: 'toast',
+              boost: 1
+            }
+          }
+        })
+        expect(query.bool.must[0].bool.should[1]).to.deep.equal({
+          prefix: {
+            'parallelSubjectLiteral.raw': {
+              value: 'toast',
+              boost: 1
+            }
+          }
+        })
+        expect(query.bool.should[0]).to.deep.equal({
+          term: {
+            'subjectLiteral.raw': {
+              value: 'toast',
+              boost: 50
+            }
+          }
+        })
+        expect(query.bool.should[1]).to.deep.equal({
+          term: {
+            'parallelSubjectLiteral.raw': {
+              value: 'toast',
+              boost: 50
+            }
+          }
+        })
+      })
+    })
+
     describe('multiple adv search params', () => {
       it('applies multiple param clauses to query', () => {
         const request = new ApiRequest({