1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
import * as rp from 'request-promise';
import { Database } from './database';
import { thisExpression } from 'babel-types';
export class Search {
public static Instance = new Search();
private url = 'http://localhost:8983/solr/';
public async updateDocument(document: any) {
try {
const res = await rp.post(this.url + "dash/update", {
headers: { 'content-type': 'application/json' },
body: JSON.stringify([document])
});
return res;
} catch (e) {
// console.warn("Search error: " + e + document);
}
}
public async updateDocuments(documents: any[]) {
try {
const res = await rp.post(this.url + "dash/update", {
headers: { 'content-type': 'application/json' },
body: JSON.stringify(documents)
});
return res;
} catch (e) {
// console.warn("Search error: ", e, documents);
}
}
public async search(query: string, start: number = 0, rows: number = 10) {
try {
const searchResults = JSON.parse(await rp.get(this.url + "dash/select", {
qs: {
q: query,
fl: "id",
start,
rows,
}
}));
const { docs, numFound } = searchResults.response;
const ids = docs.map((field: any) => field.id);
return { ids, numFound };
} catch {
return { ids: [], numFound: -1 };
}
}
public async clear() {
try {
return await rp.post(this.url + "dash/update", {
body: {
delete: {
query: "*:*"
}
},
json: true
});
} catch { }
}
public deleteDocuments(docs: string[]) {
const promises: rp.RequestPromise[] = [];
const nToDelete = 1000;
let index = 0;
while (index < docs.length) {
const count = Math.min(docs.length - index, nToDelete);
const deleteIds = docs.slice(index, index + count);
index += count;
promises.push(rp.post(this.url + "dash/update", {
body: {
delete: {
query: deleteIds.map(id => `id:"${id}"`).join(" ")
}
},
json: true
}));
}
return Promise.all(promises);
}
}
|
