evaengine 0.11.2 → 1.0.1

package/src/services/rest_client.js

@@ -0,0 +1,112 @@
+import constitute from 'constitute';
+import HttpClient from './http_client.js';
+import Namespace from './namespace.js';
+import { RestServiceLogicException, RestServiceIOException } from '../exceptions/index.js';
+import ServiceInterface from './interface.js';
+
+class RestClient extends ServiceInterface {
+  /**
+   * @param {HttpClient} client
+   * @param {Namespace} ns
+   */
+  constructor(client, ns) {
+    super();
+    this.client = client;
+    this.ns = ns;
+    this.baseUrl = null;
+  }
+
+    setBaseUrl(baseUrl) {
+    this.baseUrl = baseUrl;
+    return this;
+  }
+
+  getBaseUrl() {
+    return this.baseUrl;
+  }
+
+  getProto() {
+    return this.client.getProto();
+  }
+
+  getInstance() {
+    return this.client.getInstance();
+  }
+
+  populateTrace(params) {
+    const {
+      traceId, spanId, sampled
+    } = this.ns.get('tracer') || {};
+    if (!traceId || !spanId) {
+      return params;
+    }
+
+    if (!params.headers) {
+      Object.assign(params, { headers: {} });
+    }
+    Object.assign(params.headers, {
+      'X-B3-TraceId': traceId,
+      'X-B3-SpanId': spanId,
+      'X-B3-Sampled': sampled
+    });
+    return params;
+  }
+
+  saveToTracer(headers) {
+    if (!headers || Object.keys(headers).length < 1) {
+      return;
+    }
+
+    const tracer = this.ns.get('tracer');
+    if (!tracer) {
+      return;
+    }
+    Object.entries(headers).forEach(([key, value]) => {
+      if (key.startsWith('x-debug-')) {
+        tracer.debug[key] = value;
+      }
+    });
+  }
+
+  rawRequest(params) {
+    if (this.baseUrl) {
+      Object.assign(params, { url: this.baseUrl + (params.url || params.uri) });
+      if (params.uri) {
+        delete params.uri;
+      }
+    }
+    return this.client.getInstance()(this.populateTrace(params));
+  }
+
+  async request(params) {
+    if (this.baseUrl) {
+      Object.assign(params, {
+        url: this.baseUrl + (params.url || params.uri)
+      });
+      if (params.uri) {
+        delete params.uri;
+      }
+    }
+    try {
+      const { headers, body } = await this.client.getInstance()(Object.assign(
+        this.populateTrace(params),
+        {
+          json: true,
+          resolveWithFullResponse: true
+        }
+      ));
+      this.saveToTracer(headers);
+      return body;
+    } catch (e) {
+      //FIXME TypeError无法被记录?
+      const { statusCode } = e;
+      if (statusCode && statusCode >= 400 && statusCode < 500) {
+        throw new RestServiceLogicException(e);
+      }
+      throw new RestServiceIOException(e);
+    }
+  }
+}
+
+constitute.Dependencies(HttpClient, Namespace)(RestClient);
+export default RestClient;