clause-sdk-node/index.js at master · chatopera/clause-sdk-node · GitHub

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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/**
 * Intent Service Client
 */
const debug = require("debug")("chatopera:clause");
const thrift = require("thrift");
const Serving = require("./lib/gen-nodejs/Serving");
const ttypes = require("./lib/gen-nodejs/server_types");
const Q = require("q");

const METHODS = [
  // 自定义字典
  "postCustomDict",
  "putCustomDict",
  "getCustomDicts",
  "getCustomDict",
  "delCustomDict",
  // 系统字典管理
  //   "postSysDict", // 不开放到Superbrain
  //   "putSysDict",  // 不开放到Superbrain
  "getSysDicts",
  "getSysDict",
  "refSysDict",
  "unrefSysDict",
  // 词条管理
  "putDictWord",
  "getDictWords",
  "delDictWord",
  "hasDictWord",
  // 获得所有自定义词典列表和被引用的系统列表
  "myDicts",
  // 获得被引用的系统词典列表
  "mySysdicts",
  // 意图管理
  "postIntent",
  "putIntent",
  "getIntents",
  "getIntent",
  "delIntent",
  // 意图说法管理
  "postUtter",
  "putUtter",
  "getUtters",
  "getUtter",
  "delUtter",
  // 意图槽位管理
  "postSlot",
  "putSlot",
  "getSlots",
  "getSlot",
  "delSlot",
  // 对话管理
  "train",
  "status",
  "chat",
  "version",
  "prover",
  "devver",
  "online",
  "offline",
  "putSession",
  "getSession"
];

function Client() {
  this._client = null;
  this.resolved = false;
}

Client.prototype.connect = function(ip, port) {
  let deferred = Q.defer();

  var _conn = thrift.createConnection(ip, port, {
    transport: thrift.TFramedTransport,
    protocol: thrift.TBinaryProtocol,
    max_attempts: 1000,
    retry_max_delay: 2000,
    connect_timeout: 600000
  });

  this._client = thrift.createClient(Serving, _conn);

  _conn.on("error", err => {
    console.error("clause: Thrift Connection error", err);
    if (!this.resolved) {
      deferred.reject(err);
    }
  });

  _conn.on("connect", () => {
    debug("thrift clause connection is established successfully.");
    this.resolved = true;
    deferred.resolve();
  });

  for (let x of METHODS) {
    Client.prototype[x] = Q.nbind(this._client[x], this._client);
  }

  return deferred.promise;
};

exports = module.exports = Client;