123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239 |
- import { Subject, AnonymousSubject } from '../../Subject';
- import { Subscriber } from '../../Subscriber';
- import { Observable } from '../../Observable';
- import { Subscription } from '../../Subscription';
- import { root } from '../../util/root';
- import { ReplaySubject } from '../../ReplaySubject';
- import { tryCatch } from '../../util/tryCatch';
- import { errorObject } from '../../util/errorObject';
- import { assign } from '../../util/assign';
- /**
- * We need this JSDoc comment for affecting ESDoc.
- * @extends {Ignored}
- * @hide true
- */
- export class WebSocketSubject extends AnonymousSubject {
- constructor(urlConfigOrSource, destination) {
- if (urlConfigOrSource instanceof Observable) {
- super(destination, urlConfigOrSource);
- }
- else {
- super();
- this.WebSocketCtor = root.WebSocket;
- this._output = new Subject();
- if (typeof urlConfigOrSource === 'string') {
- this.url = urlConfigOrSource;
- }
- else {
- // WARNING: config object could override important members here.
- assign(this, urlConfigOrSource);
- }
- if (!this.WebSocketCtor) {
- throw new Error('no WebSocket constructor can be found');
- }
- this.destination = new ReplaySubject();
- }
- }
- resultSelector(e) {
- return JSON.parse(e.data);
- }
- /**
- * Wrapper around the w3c-compatible WebSocket object provided by the browser.
- *
- * @example <caption>Wraps browser WebSocket</caption>
- *
- * let socket$ = Observable.webSocket('ws://localhost:8081');
- *
- * socket$.subscribe(
- * (msg) => console.log('message received: ' + msg),
- * (err) => console.log(err),
- * () => console.log('complete')
- * );
- *
- * socket$.next(JSON.stringify({ op: 'hello' }));
- *
- * @example <caption>Wraps WebSocket from nodejs-websocket (using node.js)</caption>
- *
- * import { w3cwebsocket } from 'websocket';
- *
- * let socket$ = Observable.webSocket({
- * url: 'ws://localhost:8081',
- * WebSocketCtor: w3cwebsocket
- * });
- *
- * socket$.subscribe(
- * (msg) => console.log('message received: ' + msg),
- * (err) => console.log(err),
- * () => console.log('complete')
- * );
- *
- * socket$.next(JSON.stringify({ op: 'hello' }));
- *
- * @param {string | WebSocketSubjectConfig} urlConfigOrSource the source of the websocket as an url or a structure defining the websocket object
- * @return {WebSocketSubject}
- * @static true
- * @name webSocket
- * @owner Observable
- */
- static create(urlConfigOrSource) {
- return new WebSocketSubject(urlConfigOrSource);
- }
- lift(operator) {
- const sock = new WebSocketSubject(this, this.destination);
- sock.operator = operator;
- return sock;
- }
- _resetState() {
- this.socket = null;
- if (!this.source) {
- this.destination = new ReplaySubject();
- }
- this._output = new Subject();
- }
- // TODO: factor this out to be a proper Operator/Subscriber implementation and eliminate closures
- multiplex(subMsg, unsubMsg, messageFilter) {
- const self = this;
- return new Observable((observer) => {
- const result = tryCatch(subMsg)();
- if (result === errorObject) {
- observer.error(errorObject.e);
- }
- else {
- self.next(result);
- }
- let subscription = self.subscribe(x => {
- const result = tryCatch(messageFilter)(x);
- if (result === errorObject) {
- observer.error(errorObject.e);
- }
- else if (result) {
- observer.next(x);
- }
- }, err => observer.error(err), () => observer.complete());
- return () => {
- const result = tryCatch(unsubMsg)();
- if (result === errorObject) {
- observer.error(errorObject.e);
- }
- else {
- self.next(result);
- }
- subscription.unsubscribe();
- };
- });
- }
- _connectSocket() {
- const { WebSocketCtor } = this;
- const observer = this._output;
- let socket = null;
- try {
- socket = this.protocol ?
- new WebSocketCtor(this.url, this.protocol) :
- new WebSocketCtor(this.url);
- this.socket = socket;
- if (this.binaryType) {
- this.socket.binaryType = this.binaryType;
- }
- }
- catch (e) {
- observer.error(e);
- return;
- }
- const subscription = new Subscription(() => {
- this.socket = null;
- if (socket && socket.readyState === 1) {
- socket.close();
- }
- });
- socket.onopen = (e) => {
- const openObserver = this.openObserver;
- if (openObserver) {
- openObserver.next(e);
- }
- const queue = this.destination;
- this.destination = Subscriber.create((x) => socket.readyState === 1 && socket.send(x), (e) => {
- const closingObserver = this.closingObserver;
- if (closingObserver) {
- closingObserver.next(undefined);
- }
- if (e && e.code) {
- socket.close(e.code, e.reason);
- }
- else {
- observer.error(new TypeError('WebSocketSubject.error must be called with an object with an error code, ' +
- 'and an optional reason: { code: number, reason: string }'));
- }
- this._resetState();
- }, () => {
- const closingObserver = this.closingObserver;
- if (closingObserver) {
- closingObserver.next(undefined);
- }
- socket.close();
- this._resetState();
- });
- if (queue && queue instanceof ReplaySubject) {
- subscription.add(queue.subscribe(this.destination));
- }
- };
- socket.onerror = (e) => {
- this._resetState();
- observer.error(e);
- };
- socket.onclose = (e) => {
- this._resetState();
- const closeObserver = this.closeObserver;
- if (closeObserver) {
- closeObserver.next(e);
- }
- if (e.wasClean) {
- observer.complete();
- }
- else {
- observer.error(e);
- }
- };
- socket.onmessage = (e) => {
- const result = tryCatch(this.resultSelector)(e);
- if (result === errorObject) {
- observer.error(errorObject.e);
- }
- else {
- observer.next(result);
- }
- };
- }
- /** @deprecated internal use only */ _subscribe(subscriber) {
- const { source } = this;
- if (source) {
- return source.subscribe(subscriber);
- }
- if (!this.socket) {
- this._connectSocket();
- }
- let subscription = new Subscription();
- subscription.add(this._output.subscribe(subscriber));
- subscription.add(() => {
- const { socket } = this;
- if (this._output.observers.length === 0) {
- if (socket && socket.readyState === 1) {
- socket.close();
- }
- this._resetState();
- }
- });
- return subscription;
- }
- unsubscribe() {
- const { source, socket } = this;
- if (socket && socket.readyState === 1) {
- socket.close();
- this._resetState();
- }
- super.unsubscribe();
- if (!source) {
- this.destination = new ReplaySubject();
- }
- }
- }
- //# sourceMappingURL=WebSocketSubject.js.map
|