forked from yanyiwu/nodejieba
-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
45 lines (38 loc) · 1.52 KB
/
index.js
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
var nodejieba = require( __dirname + "/build/Release/nodejieba.node");
nodejieba.DEFAULT_DICT = __dirname + "/dict/jieba.dict.utf8",
nodejieba.DEFAULT_HMM_DICT = __dirname + "/dict/hmm_model.utf8",
nodejieba.DEFAULT_USER_DICT = __dirname + "/dict/user.dict.utf8";
nodejieba.DEFAULT_IDF_DICT = __dirname + "/dict/idf.utf8";
nodejieba.DEFAULT_STOP_WORD_DICT = __dirname + "/dict/stop_words.utf8";
var isDictLoaded = false;
var someFunct = nodejieba.load;
nodejieba.load = function (dictJson) {
if (!dictJson) {
dictJson = {}
}
dict = dictJson.dict || nodejieba.DEFAULT_DICT;
hmmDict = dictJson.hmmDict || nodejieba.DEFAULT_HMM_DICT;
userDict = dictJson.userDict || nodejieba.DEFAULT_USER_DICT;
idfDict = dictJson.idfDict || nodejieba.DEFAULT_IDF_DICT;
stopWordDict = dictJson.stopWordDict || nodejieba.DEFAULT_STOP_WORD_DICT;
isDictLoaded = true;
return someFunct.call(this, dict, hmmDict, userDict, idfDict, stopWordDict);
}
function wrapWithDictLoad(obj, functName) {
var someFunct = obj[functName];
obj[functName] = function () {
if (!isDictLoaded) {
nodejieba.load();
}
return someFunct.apply(this, arguments);
}
}
wrapWithDictLoad(nodejieba, "cut");
wrapWithDictLoad(nodejieba, "cutAll");
wrapWithDictLoad(nodejieba, "cutHMM");
wrapWithDictLoad(nodejieba, "cutForSearch");
wrapWithDictLoad(nodejieba, "cutSmall");
wrapWithDictLoad(nodejieba, "tag");
wrapWithDictLoad(nodejieba, "extract");
wrapWithDictLoad(nodejieba, "insertWord");
module.exports = nodejieba;