@zilliz/milvus2-sdk-node 2.2.10 → 2.2.12

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (125) hide show
  1. package/README.md +3 -3
  2. package/dist/milvus/MilvusClient.d.ts +0 -4
  3. package/dist/milvus/MilvusClient.js +20 -48
  4. package/dist/milvus/MilvusClient.js.map +1 -1
  5. package/dist/milvus/OrmClient.d.ts +19 -0
  6. package/dist/milvus/OrmClient.js +77 -0
  7. package/dist/milvus/OrmClient.js.map +1 -0
  8. package/dist/milvus/const/ErrorReason.d.ts +36 -31
  9. package/dist/milvus/const/ErrorReason.js +39 -32
  10. package/dist/milvus/const/ErrorReason.js.map +1 -1
  11. package/dist/milvus/const/Milvus.d.ts +2 -9
  12. package/dist/milvus/const/Milvus.js +16 -33
  13. package/dist/milvus/const/Milvus.js.map +1 -1
  14. package/dist/milvus/const/client.d.ts +5 -0
  15. package/dist/milvus/const/client.js +10 -0
  16. package/dist/milvus/const/client.js.map +1 -0
  17. package/dist/milvus/const/defaults.d.ts +11 -0
  18. package/dist/milvus/const/defaults.js +16 -0
  19. package/dist/milvus/const/defaults.js.map +1 -0
  20. package/dist/milvus/const/highLevel.d.ts +23 -0
  21. package/dist/milvus/const/highLevel.js +26 -0
  22. package/dist/milvus/const/highLevel.js.map +1 -0
  23. package/dist/milvus/const/index.d.ts +4 -1
  24. package/dist/milvus/const/index.js +4 -1
  25. package/dist/milvus/const/index.js.map +1 -1
  26. package/dist/milvus/const/limits.d.ts +2 -0
  27. package/dist/milvus/const/limits.js +7 -0
  28. package/dist/milvus/const/limits.js.map +1 -0
  29. package/dist/milvus/grpc/BaseClient.d.ts +16 -7
  30. package/dist/milvus/grpc/BaseClient.js +37 -18
  31. package/dist/milvus/grpc/BaseClient.js.map +1 -1
  32. package/dist/milvus/grpc/Collection.d.ts +32 -3
  33. package/dist/milvus/grpc/Collection.js +242 -416
  34. package/dist/milvus/grpc/Collection.js.map +1 -1
  35. package/dist/milvus/grpc/Data.d.ts +68 -8
  36. package/dist/milvus/grpc/Data.js +539 -566
  37. package/dist/milvus/grpc/Data.js.map +1 -1
  38. package/dist/milvus/grpc/Database.d.ts +69 -0
  39. package/dist/milvus/grpc/Database.js +98 -0
  40. package/dist/milvus/grpc/Database.js.map +1 -0
  41. package/dist/milvus/grpc/GrpcClient.d.ts +18 -8
  42. package/dist/milvus/grpc/GrpcClient.js +87 -156
  43. package/dist/milvus/grpc/GrpcClient.js.map +1 -1
  44. package/dist/milvus/grpc/MilvusIndex.d.ts +1 -1
  45. package/dist/milvus/grpc/MilvusIndex.js +46 -142
  46. package/dist/milvus/grpc/MilvusIndex.js.map +1 -1
  47. package/dist/milvus/grpc/Partition.d.ts +1 -1
  48. package/dist/milvus/grpc/Partition.js +53 -157
  49. package/dist/milvus/grpc/Partition.js.map +1 -1
  50. package/dist/milvus/grpc/Resource.d.ts +2 -2
  51. package/dist/milvus/grpc/Resource.js +64 -170
  52. package/dist/milvus/grpc/Resource.js.map +1 -1
  53. package/dist/milvus/grpc/User.d.ts +1 -1
  54. package/dist/milvus/grpc/User.js +193 -381
  55. package/dist/milvus/grpc/User.js.map +1 -1
  56. package/dist/milvus/index.d.ts +1 -1
  57. package/dist/milvus/index.js +1 -1
  58. package/dist/milvus/index.js.map +1 -1
  59. package/dist/milvus/orm/Collection.d.ts +45 -0
  60. package/dist/milvus/orm/Collection.js +172 -0
  61. package/dist/milvus/orm/Collection.js.map +1 -0
  62. package/dist/milvus/orm/index.d.ts +2 -0
  63. package/dist/milvus/{types.js → orm/index.js} +3 -11
  64. package/dist/milvus/orm/index.js.map +1 -0
  65. package/dist/milvus/orm/utils.d.ts +27 -0
  66. package/dist/milvus/orm/utils.js +27 -0
  67. package/dist/milvus/orm/utils.js.map +1 -0
  68. package/dist/milvus/types/Client.d.ts +12 -1
  69. package/dist/milvus/types/Collection.d.ts +18 -4
  70. package/dist/milvus/types/Data.d.ts +41 -5
  71. package/dist/milvus/types/Database.d.ts +12 -0
  72. package/dist/milvus/types/Database.js +3 -0
  73. package/dist/milvus/types/Database.js.map +1 -0
  74. package/dist/milvus/types/HighLevel.d.ts +13 -0
  75. package/dist/milvus/types/HighLevel.js +3 -0
  76. package/dist/milvus/types/HighLevel.js.map +1 -0
  77. package/dist/milvus/types/MilvusIndex.d.ts +1 -1
  78. package/dist/milvus/types/Resource.d.ts +1 -1
  79. package/dist/milvus/types/{Index.d.ts → index.d.ts} +2 -0
  80. package/dist/milvus/types/{Index.js → index.js} +3 -1
  81. package/dist/milvus/types/index.js.map +1 -0
  82. package/dist/{utils → milvus/utils}/Blob.js +4 -4
  83. package/dist/milvus/utils/Blob.js.map +1 -0
  84. package/dist/{utils → milvus/utils}/Format.d.ts +66 -2
  85. package/dist/milvus/utils/Format.js +416 -0
  86. package/dist/milvus/utils/Format.js.map +1 -0
  87. package/dist/{utils → milvus/utils}/Function.d.ts +1 -1
  88. package/dist/{utils → milvus/utils}/Function.js +7 -7
  89. package/dist/milvus/utils/Function.js.map +1 -0
  90. package/dist/{utils → milvus/utils}/Grpc.d.ts +9 -8
  91. package/dist/milvus/utils/Grpc.js +190 -0
  92. package/dist/milvus/utils/Grpc.js.map +1 -0
  93. package/dist/{utils → milvus/utils}/Validate.d.ts +7 -1
  94. package/dist/milvus/utils/Validate.js +140 -0
  95. package/dist/milvus/utils/Validate.js.map +1 -0
  96. package/dist/{utils → milvus/utils}/index.d.ts +2 -0
  97. package/dist/{utils → milvus/utils}/index.js +2 -0
  98. package/dist/milvus/utils/index.js.map +1 -0
  99. package/dist/milvus/utils/logger.d.ts +1 -0
  100. package/dist/milvus/utils/logger.js +18 -0
  101. package/dist/milvus/utils/logger.js.map +1 -0
  102. package/dist/proto/proto/common.proto +38 -14
  103. package/dist/proto/proto/google/protobuf/descriptor.proto +8 -4
  104. package/dist/proto/proto/milvus.proto +86 -83
  105. package/dist/proto/proto/schema.proto +5 -0
  106. package/dist/sdk.json +1 -1
  107. package/package.json +14 -11
  108. package/dist/milvus/types/Index.js.map +0 -1
  109. package/dist/milvus/types.d.ts +0 -9
  110. package/dist/milvus/types.js.map +0 -1
  111. package/dist/proto/proto/feder.proto +0 -40
  112. package/dist/proto/proto/msg.proto +0 -107
  113. package/dist/utils/Blob.js.map +0 -1
  114. package/dist/utils/Format.js +0 -252
  115. package/dist/utils/Format.js.map +0 -1
  116. package/dist/utils/Function.js.map +0 -1
  117. package/dist/utils/Grpc.js +0 -170
  118. package/dist/utils/Grpc.js.map +0 -1
  119. package/dist/utils/Validate.js +0 -125
  120. package/dist/utils/Validate.js.map +0 -1
  121. package/dist/utils/index.js.map +0 -1
  122. package/dist/utils/test.d.ts +0 -29
  123. package/dist/utils/test.js +0 -112
  124. package/dist/utils/test.js.map +0 -1
  125. /package/dist/{utils → milvus/utils}/Blob.d.ts +0 -0
@@ -1,19 +1,4 @@
1
1
  "use strict";
2
- var __extends = (this && this.__extends) || (function () {
3
- var extendStatics = function (d, b) {
4
- extendStatics = Object.setPrototypeOf ||
5
- ({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
6
- function (d, b) { for (var p in b) if (Object.prototype.hasOwnProperty.call(b, p)) d[p] = b[p]; };
7
- return extendStatics(d, b);
8
- };
9
- return function (d, b) {
10
- if (typeof b !== "function" && b !== null)
11
- throw new TypeError("Class extends value " + String(b) + " is not a constructor or null");
12
- extendStatics(d, b);
13
- function __() { this.constructor = d; }
14
- d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
15
- };
16
- })();
17
2
  var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
18
3
  function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
19
4
  return new (P || (P = Promise))(function (resolve, reject) {
@@ -23,43 +8,11 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
23
8
  step((generator = generator.apply(thisArg, _arguments || [])).next());
24
9
  });
25
10
  };
26
- var __generator = (this && this.__generator) || function (thisArg, body) {
27
- var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
28
- return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
29
- function verb(n) { return function (v) { return step([n, v]); }; }
30
- function step(op) {
31
- if (f) throw new TypeError("Generator is already executing.");
32
- while (g && (g = 0, op[0] && (_ = 0)), _) try {
33
- if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
34
- if (y = 0, t) op = [op[0] & 2, t.value];
35
- switch (op[0]) {
36
- case 0: case 1: t = op; break;
37
- case 4: _.label++; return { value: op[1], done: false };
38
- case 5: _.label++; y = op[1]; op = [0]; continue;
39
- case 7: op = _.ops.pop(); _.trys.pop(); continue;
40
- default:
41
- if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
42
- if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
43
- if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
44
- if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
45
- if (t[2]) _.ops.pop();
46
- _.trys.pop(); continue;
47
- }
48
- op = body.call(thisArg, _);
49
- } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
50
- if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
51
- }
52
- };
53
11
  Object.defineProperty(exports, "__esModule", { value: true });
54
12
  exports.User = void 0;
55
- var utils_1 = require("../../utils");
56
- var Resource_1 = require("./Resource");
57
- var __1 = require("..");
58
- var User = /** @class */ (function (_super) {
59
- __extends(User, _super);
60
- function User() {
61
- return _super !== null && _super.apply(this, arguments) || this;
62
- }
13
+ const Resource_1 = require("./Resource");
14
+ const __1 = require("../");
15
+ class User extends Resource_1.Resource {
63
16
  /**
64
17
  * Create user in milvus
65
18
  *
@@ -85,27 +38,19 @@ var User = /** @class */ (function (_super) {
85
38
  * });
86
39
  * ```
87
40
  */
88
- User.prototype.createUser = function (data) {
89
- return __awaiter(this, void 0, void 0, function () {
90
- var encryptedPassword, promise;
91
- return __generator(this, function (_a) {
92
- switch (_a.label) {
93
- case 0:
94
- if (data.username === undefined || data.password === undefined) {
95
- throw new Error(__1.ERROR_REASONS.USERNAME_PWD_ARE_REQUIRED);
96
- }
97
- encryptedPassword = (0, utils_1.stringToBase64)(data.password);
98
- return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateCredential', {
99
- username: data.username,
100
- password: encryptedPassword,
101
- }, data.timeout || this.timeout)];
102
- case 1:
103
- promise = _a.sent();
104
- return [2 /*return*/, promise];
105
- }
106
- });
41
+ createUser(data) {
42
+ return __awaiter(this, void 0, void 0, function* () {
43
+ if (data.username === undefined || data.password === undefined) {
44
+ throw new Error(__1.ERROR_REASONS.USERNAME_PWD_ARE_REQUIRED);
45
+ }
46
+ const encryptedPassword = (0, __1.stringToBase64)(data.password);
47
+ const promise = yield (0, __1.promisify)(this.client, 'CreateCredential', {
48
+ username: data.username,
49
+ password: encryptedPassword,
50
+ }, data.timeout || this.timeout);
51
+ return promise;
107
52
  });
108
- };
53
+ }
109
54
  /**
110
55
  * Update user in milvus
111
56
  *
@@ -132,31 +77,23 @@ var User = /** @class */ (function (_super) {
132
77
  * });
133
78
  * ```
134
79
  */
135
- User.prototype.updateUser = function (data) {
136
- return __awaiter(this, void 0, void 0, function () {
137
- var encryptedOldPwd, encryptedNewPwd, promise;
138
- return __generator(this, function (_a) {
139
- switch (_a.label) {
140
- case 0:
141
- if (data.username === undefined ||
142
- data.newPassword === undefined ||
143
- data.oldPassword === undefined) {
144
- throw new Error(__1.ERROR_REASONS.USERNAME_PWD_ARE_REQUIRED);
145
- }
146
- encryptedOldPwd = (0, utils_1.stringToBase64)(data.oldPassword);
147
- encryptedNewPwd = (0, utils_1.stringToBase64)(data.newPassword);
148
- return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'UpdateCredential', {
149
- username: data.username,
150
- oldPassword: encryptedOldPwd,
151
- newPassword: encryptedNewPwd,
152
- }, data.timeout || this.timeout)];
153
- case 1:
154
- promise = _a.sent();
155
- return [2 /*return*/, promise];
156
- }
157
- });
80
+ updateUser(data) {
81
+ return __awaiter(this, void 0, void 0, function* () {
82
+ if (data.username === undefined ||
83
+ data.newPassword === undefined ||
84
+ data.oldPassword === undefined) {
85
+ throw new Error(__1.ERROR_REASONS.USERNAME_PWD_ARE_REQUIRED);
86
+ }
87
+ const encryptedOldPwd = (0, __1.stringToBase64)(data.oldPassword);
88
+ const encryptedNewPwd = (0, __1.stringToBase64)(data.newPassword);
89
+ const promise = yield (0, __1.promisify)(this.client, 'UpdateCredential', {
90
+ username: data.username,
91
+ oldPassword: encryptedOldPwd,
92
+ newPassword: encryptedNewPwd,
93
+ }, data.timeout || this.timeout);
94
+ return promise;
158
95
  });
159
- };
96
+ }
160
97
  /**
161
98
  * Delete user in milvus
162
99
  *
@@ -180,25 +117,17 @@ var User = /** @class */ (function (_super) {
180
117
  * });
181
118
  * ```
182
119
  */
183
- User.prototype.deleteUser = function (data) {
184
- return __awaiter(this, void 0, void 0, function () {
185
- var promise;
186
- return __generator(this, function (_a) {
187
- switch (_a.label) {
188
- case 0:
189
- if (!data.username) {
190
- throw new Error(__1.ERROR_REASONS.USERNAME_IS_REQUIRED);
191
- }
192
- return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DeleteCredential', {
193
- username: data.username,
194
- }, data.timeout || this.timeout)];
195
- case 1:
196
- promise = _a.sent();
197
- return [2 /*return*/, promise];
198
- }
199
- });
120
+ deleteUser(data) {
121
+ return __awaiter(this, void 0, void 0, function* () {
122
+ if (!data.username) {
123
+ throw new Error(__1.ERROR_REASONS.USERNAME_IS_REQUIRED);
124
+ }
125
+ const promise = yield (0, __1.promisify)(this.client, 'DeleteCredential', {
126
+ username: data.username,
127
+ }, data.timeout || this.timeout);
128
+ return promise;
200
129
  });
201
- };
130
+ }
202
131
  /**
203
132
  * List user in milvus
204
133
  *
@@ -219,19 +148,12 @@ var User = /** @class */ (function (_super) {
219
148
  * milvusClient.listUsers();
220
149
  * ```
221
150
  */
222
- User.prototype.listUsers = function (data) {
223
- return __awaiter(this, void 0, void 0, function () {
224
- var promise;
225
- return __generator(this, function (_a) {
226
- switch (_a.label) {
227
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'ListCredUsers', {}, (data === null || data === void 0 ? void 0 : data.timeout) || this.timeout)];
228
- case 1:
229
- promise = _a.sent();
230
- return [2 /*return*/, promise];
231
- }
232
- });
151
+ listUsers(data) {
152
+ return __awaiter(this, void 0, void 0, function* () {
153
+ const promise = yield (0, __1.promisify)(this.client, 'ListCredUsers', {}, (data === null || data === void 0 ? void 0 : data.timeout) || this.timeout);
154
+ return promise;
233
155
  });
234
- };
156
+ }
235
157
  /**
236
158
  * Create user role
237
159
  *
@@ -253,21 +175,14 @@ var User = /** @class */ (function (_super) {
253
175
  * milvusClient.createRole({roleName: 'myrole'});
254
176
  * ```
255
177
  */
256
- User.prototype.createRole = function (data) {
257
- return __awaiter(this, void 0, void 0, function () {
258
- var promise;
259
- return __generator(this, function (_a) {
260
- switch (_a.label) {
261
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'CreateRole', {
262
- entity: { name: data.roleName },
263
- }, data.timeout || this.timeout)];
264
- case 1:
265
- promise = _a.sent();
266
- return [2 /*return*/, promise];
267
- }
268
- });
178
+ createRole(data) {
179
+ return __awaiter(this, void 0, void 0, function* () {
180
+ const promise = yield (0, __1.promisify)(this.client, 'CreateRole', {
181
+ entity: { name: data.roleName },
182
+ }, data.timeout || this.timeout);
183
+ return promise;
269
184
  });
270
- };
185
+ }
271
186
  /**
272
187
  * Drop user role
273
188
  *
@@ -289,21 +204,14 @@ var User = /** @class */ (function (_super) {
289
204
  * milvusClient.dropRole({roleName: 'myrole'});
290
205
  * ```
291
206
  */
292
- User.prototype.dropRole = function (data) {
293
- return __awaiter(this, void 0, void 0, function () {
294
- var promise;
295
- return __generator(this, function (_a) {
296
- switch (_a.label) {
297
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'DropRole', {
298
- role_name: data.roleName,
299
- }, data.timeout || this.timeout)];
300
- case 1:
301
- promise = _a.sent();
302
- return [2 /*return*/, promise];
303
- }
304
- });
207
+ dropRole(data) {
208
+ return __awaiter(this, void 0, void 0, function* () {
209
+ const promise = yield (0, __1.promisify)(this.client, 'DropRole', {
210
+ role_name: data.roleName,
211
+ }, data.timeout || this.timeout);
212
+ return promise;
305
213
  });
306
- };
214
+ }
307
215
  /**
308
216
  * add user to role
309
217
  *
@@ -326,23 +234,16 @@ var User = /** @class */ (function (_super) {
326
234
  * milvusClient.addUserToRole({username: 'my', roleName: 'myrole'});
327
235
  * ```
328
236
  */
329
- User.prototype.addUserToRole = function (data) {
330
- return __awaiter(this, void 0, void 0, function () {
331
- var promise;
332
- return __generator(this, function (_a) {
333
- switch (_a.label) {
334
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'OperateUserRole', {
335
- username: data.username,
336
- role_name: data.roleName,
337
- type: __1.OperateUserRoleType.AddUserToRole,
338
- }, data.timeout || this.timeout)];
339
- case 1:
340
- promise = _a.sent();
341
- return [2 /*return*/, promise];
342
- }
343
- });
237
+ addUserToRole(data) {
238
+ return __awaiter(this, void 0, void 0, function* () {
239
+ const promise = yield (0, __1.promisify)(this.client, 'OperateUserRole', {
240
+ username: data.username,
241
+ role_name: data.roleName,
242
+ type: __1.OperateUserRoleType.AddUserToRole,
243
+ }, data.timeout || this.timeout);
244
+ return promise;
344
245
  });
345
- };
246
+ }
346
247
  /**
347
248
  * remove user from role
348
249
  *
@@ -365,23 +266,16 @@ var User = /** @class */ (function (_super) {
365
266
  * milvusClient.removeUserFromRole({username: 'my', roleName: 'myrole'});
366
267
  * ```
367
268
  */
368
- User.prototype.removeUserFromRole = function (data) {
369
- return __awaiter(this, void 0, void 0, function () {
370
- var promise;
371
- return __generator(this, function (_a) {
372
- switch (_a.label) {
373
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'OperateUserRole', {
374
- username: data.username,
375
- role_name: data.roleName,
376
- type: __1.OperateUserRoleType.RemoveUserFromRole,
377
- }, data.timeout || this.timeout)];
378
- case 1:
379
- promise = _a.sent();
380
- return [2 /*return*/, promise];
381
- }
382
- });
269
+ removeUserFromRole(data) {
270
+ return __awaiter(this, void 0, void 0, function* () {
271
+ const promise = yield (0, __1.promisify)(this.client, 'OperateUserRole', {
272
+ username: data.username,
273
+ role_name: data.roleName,
274
+ type: __1.OperateUserRoleType.RemoveUserFromRole,
275
+ }, data.timeout || this.timeout);
276
+ return promise;
383
277
  });
384
- };
278
+ }
385
279
  /**
386
280
  * gets all users that belong to a specified role
387
281
  *
@@ -404,22 +298,15 @@ var User = /** @class */ (function (_super) {
404
298
  * milvusClient.selectRole({roleName: 'myrole'});
405
299
  * ```
406
300
  */
407
- User.prototype.selectRole = function (data) {
408
- return __awaiter(this, void 0, void 0, function () {
409
- var promise;
410
- return __generator(this, function (_a) {
411
- switch (_a.label) {
412
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'SelectRole', {
413
- role: { name: data.roleName },
414
- include_user_info: data.includeUserInfo || true,
415
- }, data.timeout || this.timeout)];
416
- case 1:
417
- promise = _a.sent();
418
- return [2 /*return*/, promise];
419
- }
420
- });
301
+ selectRole(data) {
302
+ return __awaiter(this, void 0, void 0, function* () {
303
+ const promise = yield (0, __1.promisify)(this.client, 'SelectRole', {
304
+ role: { name: data.roleName },
305
+ include_user_info: data.includeUserInfo || true,
306
+ }, data.timeout || this.timeout);
307
+ return promise;
421
308
  });
422
- };
309
+ }
423
310
  /**
424
311
  * list all roles
425
312
  *
@@ -440,19 +327,12 @@ var User = /** @class */ (function (_super) {
440
327
  * milvusClient.listRoles();
441
328
  * ```
442
329
  */
443
- User.prototype.listRoles = function (data) {
444
- return __awaiter(this, void 0, void 0, function () {
445
- var promise;
446
- return __generator(this, function (_a) {
447
- switch (_a.label) {
448
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'SelectRole', {}, (data === null || data === void 0 ? void 0 : data.timeout) || this.timeout)];
449
- case 1:
450
- promise = _a.sent();
451
- return [2 /*return*/, promise];
452
- }
453
- });
330
+ listRoles(data) {
331
+ return __awaiter(this, void 0, void 0, function* () {
332
+ const promise = yield (0, __1.promisify)(this.client, 'SelectRole', {}, (data === null || data === void 0 ? void 0 : data.timeout) || this.timeout);
333
+ return promise;
454
334
  });
455
- };
335
+ }
456
336
  /**
457
337
  * gets all users that belong to a specified role
458
338
  *
@@ -475,22 +355,15 @@ var User = /** @class */ (function (_super) {
475
355
  * milvusClient.selectUser({username: 'name'});
476
356
  * ```
477
357
  */
478
- User.prototype.selectUser = function (data) {
479
- return __awaiter(this, void 0, void 0, function () {
480
- var promise;
481
- return __generator(this, function (_a) {
482
- switch (_a.label) {
483
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'SelectUser', {
484
- user: { name: data.username },
485
- include_role_info: data.includeRoleInfo || true,
486
- }, data.timeout || this.timeout)];
487
- case 1:
488
- promise = _a.sent();
489
- return [2 /*return*/, promise];
490
- }
491
- });
358
+ selectUser(data) {
359
+ return __awaiter(this, void 0, void 0, function* () {
360
+ const promise = yield (0, __1.promisify)(this.client, 'SelectUser', {
361
+ user: { name: data.username },
362
+ include_role_info: data.includeRoleInfo || true,
363
+ }, data.timeout || this.timeout);
364
+ return promise;
492
365
  });
493
- };
366
+ }
494
367
  /**
495
368
  * grant privileges to a role
496
369
  *
@@ -520,29 +393,22 @@ var User = /** @class */ (function (_super) {
520
393
  * });
521
394
  * ```
522
395
  */
523
- User.prototype.grantRolePrivilege = function (data) {
524
- return __awaiter(this, void 0, void 0, function () {
525
- var promise;
526
- return __generator(this, function (_a) {
527
- switch (_a.label) {
528
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'OperatePrivilege', {
529
- entity: {
530
- role: { name: data.roleName },
531
- object: { name: data.object },
532
- object_name: data.objectName,
533
- grantor: {
534
- privilege: { name: data.privilegeName },
535
- },
536
- },
537
- type: __1.OperatePrivilegeType.Grant,
538
- }, data.timeout || this.timeout)];
539
- case 1:
540
- promise = _a.sent();
541
- return [2 /*return*/, promise];
542
- }
543
- });
396
+ grantRolePrivilege(data) {
397
+ return __awaiter(this, void 0, void 0, function* () {
398
+ const promise = yield (0, __1.promisify)(this.client, 'OperatePrivilege', {
399
+ entity: {
400
+ role: { name: data.roleName },
401
+ object: { name: data.object },
402
+ object_name: data.objectName,
403
+ grantor: {
404
+ privilege: { name: data.privilegeName },
405
+ },
406
+ },
407
+ type: __1.OperatePrivilegeType.Grant,
408
+ }, data.timeout || this.timeout);
409
+ return promise;
544
410
  });
545
- };
411
+ }
546
412
  /**
547
413
  * revoke privileges to a role
548
414
  *
@@ -572,29 +438,22 @@ var User = /** @class */ (function (_super) {
572
438
  * });
573
439
  * ```
574
440
  */
575
- User.prototype.revokeRolePrivilege = function (data) {
576
- return __awaiter(this, void 0, void 0, function () {
577
- var promise;
578
- return __generator(this, function (_a) {
579
- switch (_a.label) {
580
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'OperatePrivilege', {
581
- entity: {
582
- role: { name: data.roleName },
583
- object: { name: data.object },
584
- object_name: data.objectName,
585
- grantor: {
586
- privilege: { name: data.privilegeName },
587
- },
588
- },
589
- type: __1.OperatePrivilegeType.Revoke,
590
- }, data.timeout || this.timeout)];
591
- case 1:
592
- promise = _a.sent();
593
- return [2 /*return*/, promise];
594
- }
595
- });
441
+ revokeRolePrivilege(data) {
442
+ return __awaiter(this, void 0, void 0, function* () {
443
+ const promise = yield (0, __1.promisify)(this.client, 'OperatePrivilege', {
444
+ entity: {
445
+ role: { name: data.roleName },
446
+ object: { name: data.object },
447
+ object_name: data.objectName,
448
+ grantor: {
449
+ privilege: { name: data.privilegeName },
450
+ },
451
+ },
452
+ type: __1.OperatePrivilegeType.Revoke,
453
+ }, data.timeout || this.timeout);
454
+ return promise;
596
455
  });
597
- };
456
+ }
598
457
  /**
599
458
  * revoke all roles priviledges
600
459
  * @param data
@@ -615,65 +474,40 @@ var User = /** @class */ (function (_super) {
615
474
  * ```
616
475
  */
617
476
  /* istanbul ignore next */
618
- User.prototype.dropAllRoles = function (data) {
619
- return __awaiter(this, void 0, void 0, function () {
620
- var res, promises, i, r, grants, j, entity, _a, _b;
621
- return __generator(this, function (_c) {
622
- switch (_c.label) {
623
- case 0: return [4 /*yield*/, this.listRoles({ timeout: data === null || data === void 0 ? void 0 : data.timeout })];
624
- case 1:
625
- res = _c.sent();
626
- promises = [];
627
- i = 0;
628
- _c.label = 2;
629
- case 2:
630
- if (!(i < res.results.length)) return [3 /*break*/, 10];
631
- r = res.results[i];
632
- return [4 /*yield*/, this.listGrants({
633
- roleName: r.role.name,
634
- })];
635
- case 3:
636
- grants = _c.sent();
637
- j = 0;
638
- _c.label = 4;
639
- case 4:
640
- if (!(j < grants.entities.length)) return [3 /*break*/, 7];
641
- entity = grants.entities[j];
642
- // revoke grant
643
- return [4 /*yield*/, this.revokeRolePrivilege({
644
- roleName: entity.role.name,
645
- object: entity.object.name,
646
- objectName: entity.object_name,
647
- privilegeName: entity.grantor.privilege.name,
648
- timeout: data === null || data === void 0 ? void 0 : data.timeout,
649
- })];
650
- case 5:
651
- // revoke grant
652
- _c.sent();
653
- _c.label = 6;
654
- case 6:
655
- j++;
656
- return [3 /*break*/, 4];
657
- case 7:
658
- _b = (_a = promises).push;
659
- // drop the role
660
- return [4 /*yield*/, this.dropRole({
661
- roleName: r.role.name,
662
- timeout: data === null || data === void 0 ? void 0 : data.timeout,
663
- })];
664
- case 8:
665
- _b.apply(_a, [
666
- // drop the role
667
- _c.sent()]);
668
- _c.label = 9;
669
- case 9:
670
- i++;
671
- return [3 /*break*/, 2];
672
- case 10: return [2 /*return*/, promises];
477
+ dropAllRoles(data) {
478
+ return __awaiter(this, void 0, void 0, function* () {
479
+ // find all roles
480
+ const res = yield this.listRoles({ timeout: data === null || data === void 0 ? void 0 : data.timeout });
481
+ const promises = [];
482
+ // iterate through roles
483
+ for (let i = 0; i < res.results.length; i++) {
484
+ const r = res.results[i];
485
+ // get all grants that specific to the role
486
+ const grants = yield this.listGrants({
487
+ roleName: r.role.name,
488
+ });
489
+ // iterate throught these grant
490
+ for (let j = 0; j < grants.entities.length; j++) {
491
+ const entity = grants.entities[j];
492
+ // revoke grant
493
+ yield this.revokeRolePrivilege({
494
+ roleName: entity.role.name,
495
+ object: entity.object.name,
496
+ objectName: entity.object_name,
497
+ privilegeName: entity.grantor.privilege.name,
498
+ timeout: data === null || data === void 0 ? void 0 : data.timeout,
499
+ });
673
500
  }
674
- });
501
+ promises.push(
502
+ // drop the role
503
+ yield this.dropRole({
504
+ roleName: r.role.name,
505
+ timeout: data === null || data === void 0 ? void 0 : data.timeout,
506
+ }));
507
+ }
508
+ return promises;
675
509
  });
676
- };
510
+ }
677
511
  /**
678
512
  * select a grant
679
513
  * @param data
@@ -702,28 +536,21 @@ var User = /** @class */ (function (_super) {
702
536
  * });
703
537
  * ```
704
538
  */
705
- User.prototype.selectGrant = function (data) {
706
- return __awaiter(this, void 0, void 0, function () {
707
- var promise;
708
- return __generator(this, function (_a) {
709
- switch (_a.label) {
710
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'SelectGrant', {
711
- entity: {
712
- role: { name: data.roleName },
713
- object: { name: data.object },
714
- object_name: data.objectName,
715
- grantor: {
716
- privilege: { name: data.privilegeName },
717
- },
718
- },
719
- }, data.timeout || this.timeout)];
720
- case 1:
721
- promise = _a.sent();
722
- return [2 /*return*/, promise];
723
- }
724
- });
539
+ selectGrant(data) {
540
+ return __awaiter(this, void 0, void 0, function* () {
541
+ const promise = yield (0, __1.promisify)(this.client, 'SelectGrant', {
542
+ entity: {
543
+ role: { name: data.roleName },
544
+ object: { name: data.object },
545
+ object_name: data.objectName,
546
+ grantor: {
547
+ privilege: { name: data.privilegeName },
548
+ },
549
+ },
550
+ }, data.timeout || this.timeout);
551
+ return promise;
725
552
  });
726
- };
553
+ }
727
554
  /**
728
555
  * list all grants for a role
729
556
  * @param data
@@ -746,23 +573,16 @@ var User = /** @class */ (function (_super) {
746
573
  * });
747
574
  * ```
748
575
  */
749
- User.prototype.listGrants = function (data) {
750
- return __awaiter(this, void 0, void 0, function () {
751
- var promise;
752
- return __generator(this, function (_a) {
753
- switch (_a.label) {
754
- case 0: return [4 /*yield*/, (0, utils_1.promisify)(this.client, 'SelectGrant', {
755
- entity: {
756
- role: { name: data.roleName },
757
- },
758
- }, data.timeout || this.timeout)];
759
- case 1:
760
- promise = _a.sent();
761
- return [2 /*return*/, promise];
762
- }
763
- });
576
+ listGrants(data) {
577
+ return __awaiter(this, void 0, void 0, function* () {
578
+ const promise = yield (0, __1.promisify)(this.client, 'SelectGrant', {
579
+ entity: {
580
+ role: { name: data.roleName },
581
+ },
582
+ }, data.timeout || this.timeout);
583
+ return promise;
764
584
  });
765
- };
585
+ }
766
586
  /**
767
587
  * check if the role is existing
768
588
  * @param data
@@ -785,23 +605,15 @@ var User = /** @class */ (function (_super) {
785
605
  * });
786
606
  * ```
787
607
  */
788
- User.prototype.hasRole = function (data) {
789
- return __awaiter(this, void 0, void 0, function () {
790
- var result;
791
- return __generator(this, function (_a) {
792
- switch (_a.label) {
793
- case 0: return [4 /*yield*/, this.listRoles()];
794
- case 1:
795
- result = _a.sent();
796
- return [2 /*return*/, {
797
- status: result.status,
798
- hasRole: result.results.map(function (r) { return r.role.name; }).includes(data.roleName),
799
- }];
800
- }
801
- });
608
+ hasRole(data) {
609
+ return __awaiter(this, void 0, void 0, function* () {
610
+ const result = yield this.listRoles();
611
+ return {
612
+ status: result.status,
613
+ hasRole: result.results.map(r => r.role.name).includes(data.roleName),
614
+ };
802
615
  });
803
- };
804
- return User;
805
- }(Resource_1.Resource));
616
+ }
617
+ }
806
618
  exports.User = User;
807
619
  //# sourceMappingURL=User.js.map