NodeScirptDemo/sdk.js at main · Core-Claw/NodeScirptDemo · 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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
const grpc = require('@grpc/grpc-js');
const messages = require('./sdk_pb');
const services = require('./sdk_grpc_pb');
const { Empty } = require('google-protobuf/google/protobuf/empty_pb.js');

const address = "127.0.0.1:20086";

const _parameterClient = new services.ParameterClient(
    address,
    grpc.credentials.createInsecure()
);
const _resultClient = new services.ResultClient(
    address,
    grpc.credentials.createInsecure()
);
const _logClient = new services.LogClient(
    address,
    grpc.credentials.createInsecure()
);

// 通用 gRPC 响应处理函数
function handleGrpcResponse(err, response, resolve, reject) {
  if (err) {
    console.error('gRPC call failed:', err);
    reject(err);
    return;
  }
  const result = {
    code: response.getCode(),
    message: response.getMessage()
  };
  if (result.code !== 0) {
    console.error('gRPC call failed with code:', result.code);
    reject(new Error(result.message));
    return;
  }
  resolve(result);
}

exports.parameter = {
  getInputJSONString: function () {
    return new Promise((resolve, reject) => {
      _parameterClient.getInputJSONString(new Empty(), (err, response) => {
        if (err) {
          console.error('call getInputJSONString failed:', err);
          reject(err);
          return;
        }
        const result = {
          code: response.getCode(),
          jsonString: response.getJsonstring()
        }
        if (result.code !== 0) {
          console.error('getInputJSONString code:', result.code);
          reject(new Error(`Code: ${result.code}`));
          return;
        }
        resolve(result.jsonString);
      });
    });
  },

  getInputJSONObject: async function () {
    try {
      const jsonString = await this.getInputJSONString();
      return jsonString ? JSON.parse(jsonString) : {};
    } catch (err) {
      throw err;
    }
  }
};

exports.result = {
  setTableHeader: function (headers) {
    return new Promise((resolve, reject) => {
      // 创建 TableHeader 消息对象
      const tableHeaders = new messages.TableHeader();

      const headersList = headers.map(header => {
        // 使用 TableHeaderItem 来创建实例
        const headerMessage = new messages.TableHeaderItem(); // 修改为 TableHeaderItem
        headerMessage.setLabel(header.label);   // 设置 label
        headerMessage.setKey(header.key);       // 设置 key
        headerMessage.setFormat(header.format); // 设置 format
        return headerMessage;  // 返回这个消息对象
      });

      // 设置 headers list
      tableHeaders.setHeadersList(headersList);

      // 调用 gRPC 方法
      _resultClient.setTableHeader(tableHeaders, (err, response) => {
        handleGrpcResponse(err, response, resolve, reject);
      });
    });
  },

  pushData: function (obj) {
    return new Promise((resolve, reject) => {
      let jsonString = "";
      try {
        jsonString = JSON.stringify(obj);
      } catch (err) {
        console.error("pushData JSON.stringify failed, obj:", obj);
        reject(err);
        return;
      }

      const data = new messages.Data();
      data.setJsonstring(jsonString);
      _resultClient.pushData(data, (err, response) => {
        handleGrpcResponse(err, response, resolve, reject);
      });
    });
  }
};

exports.log = {
  logMessage: function (level, logString) {
    return new Promise((resolve, reject) => {
      const logBody = new messages.LogBody();
      logBody.setLog(logString);
      _logClient[level](logBody, (err, response) => {
        if (err) {
          console.error(`call log.${level} failed:`, err);
          reject(err);
          return;
        }
        const result = {
          code: response.getCode(),
          message: response.getMessage()
        };
        if (result.code !== 0) {
          console.error(`log.${level} code:`, result.code);
          reject(new Error(result.message));
          return;
        }
        resolve(result);
      });
    });
  },

  debug: function (logString) {
    return this.logMessage('debug', logString);
  },

  info: function (logString) {
    return this.logMessage('info', logString);
  },

  warn: function (logString) {
    return this.logMessage('warn', logString);
  },

  error: function (logString) {
    return this.logMessage('error', logString);
  }
};