platform/server/elastic/src/adapter.ts

//
// Copyright © 2020, 2021 Anticrm Platform Contributors.
// Copyright © 2021 Hardcore Engineering Inc.
//
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. You may
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//
// See the License for the specific language governing permissions and
// limitations under the License.
//

import type { Class, Doc, DocumentQuery, Ref, TxResult } from '@anticrm/core'
import type { FullTextAdapter, IndexedDoc } from '@anticrm/server-core'

import { Client } from '@elastic/elasticsearch'

class ElasticAdapter implements FullTextAdapter {
  constructor (private readonly client: Client, private readonly db: string) {}

  async close (): Promise<void> {
    await this.client.close()
  }

  async search (_class: Ref<Class<Doc>>, query: DocumentQuery<Doc>, size: number | undefined): Promise<IndexedDoc[]> {
    if (query.$search === undefined) return []
    const search = query.$search.replace(/[\\/+\-=&><!()|{}^"~*&:[\]]/g, '\\$&')

    const request: any = {
      bool: {
        must: [
          {
            multi_match: {
              query: search,
              fields: [
                'content0',
                'content1',
                'content2',
                'content3',
                'content4',
                'content5',
                'content6',
                'content7',
                'content8',
                'content9',
                'attachment.content'
              ]
            }
          }
        ],
        should: [
          {
            term: {
              _class: {
                value: _class,
                case_insensitive: true
              }
            }
          }
        ]
      }
    }

    if (query.space != null) {
      request.bool.should.push({
        term: {
          space: {
            value: query.space,
            boost: 2.0,
            case_insensitive: true
          }
        }
      })
    }

    try {
      const result = await this.client.search({
        index: this.db,
        body: {
          query: request,
          size: size ?? 200
        }
      })
      const hits = result.body.hits.hits as any[]
      return hits.map((hit) => hit._source)
    } catch (err) {
      console.error(JSON.stringify(err, null, 2))
      return []
    }
  }

  async index (doc: IndexedDoc): Promise<TxResult> {
    if (doc.data === undefined) {
      await this.client.index({
        index: this.db,
        id: doc.id,
        type: '_doc',
        body: doc
      })
    } else {
      await this.client.index({
        index: this.db,
        id: doc.id,
        type: '_doc',
        pipeline: 'attachment',
        body: doc
      })
    }
    return {}
  }

  async update (id: Ref<Doc>, update: Record<string, any>): Promise<TxResult> {
    await this.client.update({
      index: this.db,
      id,
      body: {
        doc: update
      }
    })

    return {}
  }
}

/**
 * @public
 */
export async function createElasticAdapter (
  url: string,
  dbName: string
): Promise<FullTextAdapter & { close: () => Promise<void> }> {
  const client = new Client({
    node: url
  })

  return new ElasticAdapter(client, dbName)
}