Merge pull request #86 from ainize-team/feature/yoojin/custom_request…

…_key Get req key from optional param
ainize-team · Dec 21, 2023 · edcd0d7 · edcd0d7
2 parents af1db3a + dfc8be0
commit edcd0d7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/controllers/serviceController.ts b/src/controllers/serviceController.ts
@@ -77,12 +77,12 @@ export default class ServiceController {
     return await this.ain.getValue(creditHistoryPath) as creditHistories;
   }
 
-  async request(serviceName: string, requestData: any) : Promise<any> {
+  async request(serviceName: string, requestData: any, requestKey?: string) : Promise<any> {
     this.checkRunning(serviceName);
     const result = await new Promise(async (resolve, reject) => {
-      const requestKey = Date.now();
+      requestKey = requestKey || Date.now().toString();
       const requesterAddress = this.ain.getAddress();
-      const responsePath = Path.app(serviceName).response(requesterAddress, requestKey.toString());
+      const responsePath = Path.app(serviceName).response(requesterAddress, requestKey);
       await this.handler.subscribe(responsePath, resolve);
       const requestPath = Path.app(serviceName).request(requesterAddress, requestKey);
       const requestOp = buildSetOperation("SET_VALUE", requestPath, requestData);

diff --git a/src/service.ts b/src/service.ts
@@ -79,9 +79,9 @@ export default class Service {
    * @param {string} requestData String data for request to service. 
    * @returns {string} Response data from service.
    */
-  async request(requestData: any) {
+  async request(requestData: any, requestKey?: string) {
     this.isLoggedIn();
-    return await this.serviceController.request(this.serviceName, requestData);
+    return await this.serviceController.request(this.serviceName, requestData, requestKey);
   }
 
   /**