forked from yanyiwu/nodejieba
-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
49 lines (41 loc) · 1.63 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
45
46
47
48
var binary = require('node-pre-gyp');
var path = require('path');
var binding_path = binary.find(path.resolve(path.join(__dirname,'./package.json')));
var nodejieba = require(binding_path);
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;