forked from yanyiwu/nodejieba
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
50 lines (43 loc) · 1.53 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
49
50
var binary = require('@mapbox/node-pre-gyp');
var path = require('path');
var binding_path = binary.find(path.resolve(path.join(__dirname,'./package.json')));
var nodejieba = require(binding_path);
var isDictLoaded = false;
var exports ={
DEFAULT_DICT: __dirname + "/dict/jieba.dict.utf8",
DEFAULT_HMM_DICT: __dirname + "/dict/hmm_model.utf8",
DEFAULT_USER_DICT: __dirname + "/dict/user.dict.utf8",
DEFAULT_IDF_DICT: __dirname + "/dict/idf.utf8",
DEFAULT_STOP_WORD_DICT: __dirname + "/dict/stop_words.utf8",
load: function (dictJson) {
if (!dictJson) {
dictJson = {};
}
dict = dictJson.dict || exports.DEFAULT_DICT;
hmmDict = dictJson.hmmDict || exports.DEFAULT_HMM_DICT;
userDict = dictJson.userDict || exports.DEFAULT_USER_DICT;
idfDict = dictJson.idfDict || exports.DEFAULT_IDF_DICT;
stopWordDict = dictJson.stopWordDict || exports.DEFAULT_STOP_WORD_DICT;
isDictLoaded = true;
return nodejieba.load(dict, hmmDict, userDict, idfDict, stopWordDict);
}
};
function wrapWithDictLoad(functName) {
var someFunct = nodejieba[functName];
exports[functName] = function () {
if (!isDictLoaded) {
exports.load();
}
return someFunct.apply(this, arguments);
}
}
wrapWithDictLoad("cut");
wrapWithDictLoad("cutAll");
wrapWithDictLoad("cutHMM");
wrapWithDictLoad("cutForSearch");
wrapWithDictLoad("cutSmall");
wrapWithDictLoad("tag");
wrapWithDictLoad("extract");
wrapWithDictLoad("textRankExtract");
wrapWithDictLoad("insertWord");
module.exports = exports;