import { Injectable } from '@angular/core';
import { webSocket, WebSocketSubject } from 'rxjs/webSocket';
import { Observable } from 'rxjs';

@Injectable({
  providedIn: 'root'
})
export class WebSocketService {
  private socket$: WebSocketSubject<any>;

  constructor() { }

  connect(url: string): Observable<any> {
    this.socket$ = webSocket(url);
    return this.socket$;
  }

  send(data: any): void {
    this.socket$.next(data);
  }

  close(): void {
    this.socket$.complete();
  }
}