misskey-awawa/src/client/app/common/scripts/stream.ts

302 lines
6.7 KiB
TypeScript
Raw Normal View History

import autobind from 'autobind-decorator';
import { EventEmitter } from 'eventemitter3';
2018-10-07 16:16:32 +00:00
import ReconnectingWebsocket from 'reconnecting-websocket';
import { wsUrl } from '../../config';
import MiOS from '../../mios';
/**
* Misskey stream connection
*/
export default class Stream extends EventEmitter {
private stream: ReconnectingWebsocket;
2018-10-14 00:48:47 +00:00
public state: string;
2018-10-10 11:59:10 +00:00
private sharedConnectionPools: Pool[] = [];
private sharedConnections: SharedConnection[] = [];
private nonSharedConnections: NonSharedConnection[] = [];
constructor(os: MiOS) {
super();
this.state = 'initializing';
const user = os.store.state.i;
2019-05-07 09:48:38 +00:00
this.stream = new ReconnectingWebsocket(wsUrl + (user ? `?i=${user.token}` : ''), '', { minReconnectionDelay: 1 }); // https://github.com/pladaria/reconnecting-websocket/issues/91
this.stream.addEventListener('open', this.onOpen);
this.stream.addEventListener('close', this.onClose);
this.stream.addEventListener('message', this.onMessage);
}
2018-10-10 13:13:32 +00:00
@autobind
public useSharedConnection(channel: string): SharedConnection {
2018-10-10 11:59:10 +00:00
let pool = this.sharedConnectionPools.find(p => p.channel === channel);
2018-10-10 11:59:10 +00:00
if (pool == null) {
pool = new Pool(this, channel);
this.sharedConnectionPools.push(pool);
}
2018-10-10 11:59:10 +00:00
const connection = new SharedConnection(this, channel, pool);
this.sharedConnections.push(connection);
return connection;
}
@autobind
public removeSharedConnection(connection: SharedConnection) {
2018-10-10 11:59:10 +00:00
this.sharedConnections = this.sharedConnections.filter(c => c !== connection);
}
2018-10-11 13:17:27 +00:00
@autobind
public removeSharedConnectionPool(pool: Pool) {
this.sharedConnectionPools = this.sharedConnectionPools.filter(p => p !== pool);
}
2018-10-10 13:13:32 +00:00
@autobind
public connectToChannel(channel: string, params?: any): NonSharedConnection {
const connection = new NonSharedConnection(this, channel, params);
this.nonSharedConnections.push(connection);
return connection;
}
@autobind
public disconnectToChannel(connection: NonSharedConnection) {
2018-10-10 11:59:10 +00:00
this.nonSharedConnections = this.nonSharedConnections.filter(c => c !== connection);
}
/**
* Callback of when open connection
*/
@autobind
private onOpen() {
const isReconnect = this.state == 'reconnecting';
this.state = 'connected';
this.emit('_connected_');
// チャンネル再接続
if (isReconnect) {
for (const p of this.sharedConnectionPools)
2018-10-11 13:17:27 +00:00
p.connect();
for (const c of this.nonSharedConnections)
c.connect();
}
}
/**
* Callback of when close connection
*/
@autobind
private onClose() {
2018-10-11 13:35:34 +00:00
if (this.state == 'connected') {
this.state = 'reconnecting';
this.emit('_disconnected_');
}
}
/**
* Callback of when received a message from connection
*/
@autobind
private onMessage(message) {
const { type, body } = JSON.parse(message.data);
if (type == 'channel') {
const id = body.id;
2018-10-10 11:59:10 +00:00
let connections: Connection[];
connections = this.sharedConnections.filter(c => c.id === id);
if (connections.length === 0) {
connections = [this.nonSharedConnections.find(c => c.id === id)];
}
for (const c of connections.filter(c => c != null)) {
2018-10-10 11:59:10 +00:00
c.emit(body.type, body.body);
}
} else {
this.emit(type, body);
}
}
/**
* Send a message to connection
*/
@autobind
public send(typeOrPayload, payload?) {
const data = payload === undefined ? typeOrPayload : {
type: typeOrPayload,
body: payload
};
this.stream.send(JSON.stringify(data));
}
/**
* Close this connection
*/
@autobind
public close() {
this.stream.removeEventListener('open', this.onOpen);
this.stream.removeEventListener('message', this.onMessage);
}
}
2018-10-10 11:59:10 +00:00
class Pool {
public channel: string;
public id: string;
protected stream: Stream;
2018-10-11 12:25:55 +00:00
public users = 0;
2018-10-10 11:59:10 +00:00
private disposeTimerId: any;
private isConnected = false;
2018-10-10 11:59:10 +00:00
constructor(stream: Stream, channel: string) {
this.channel = channel;
2018-10-10 11:59:10 +00:00
this.stream = stream;
2018-10-13 10:16:47 +00:00
this.id = Math.random().toString().substr(2, 8);
2018-10-11 13:17:27 +00:00
this.stream.on('_disconnected_', this.onStreamDisconnected);
}
@autobind
private onStreamDisconnected() {
this.isConnected = false;
2018-10-10 11:59:10 +00:00
}
@autobind
public inc() {
if (this.users === 0 && !this.isConnected) {
this.connect();
}
this.users++;
// タイマー解除
if (this.disposeTimerId) {
clearTimeout(this.disposeTimerId);
this.disposeTimerId = null;
}
}
@autobind
public dec() {
this.users--;
// そのコネクションの利用者が誰もいなくなったら
if (this.users === 0) {
// また直ぐに再利用される可能性があるので、一定時間待ち、
// 新たな利用者が現れなければコネクションを切断する
this.disposeTimerId = setTimeout(() => {
this.disconnect();
}, 3000);
}
}
@autobind
public connect() {
2018-10-11 13:17:27 +00:00
if (this.isConnected) return;
2018-10-10 11:59:10 +00:00
this.isConnected = true;
this.stream.send('connect', {
channel: this.channel,
2018-10-10 11:59:10 +00:00
id: this.id
});
}
@autobind
2018-10-10 11:59:10 +00:00
private disconnect() {
2018-10-11 13:17:27 +00:00
this.stream.off('_disconnected_', this.onStreamDisconnected);
2018-10-10 11:59:10 +00:00
this.stream.send('disconnect', { id: this.id });
2018-10-11 13:17:27 +00:00
this.stream.removeSharedConnectionPool(this);
2018-10-10 11:59:10 +00:00
}
}
abstract class Connection extends EventEmitter {
public channel: string;
protected stream: Stream;
public abstract id: string;
constructor(stream: Stream, channel: string) {
super();
this.stream = stream;
this.channel = channel;
}
@autobind
public send(id: string, typeOrPayload, payload?) {
2018-10-08 16:50:49 +00:00
const type = payload === undefined ? typeOrPayload.type : typeOrPayload;
const body = payload === undefined ? typeOrPayload.body : payload;
2018-10-09 18:24:09 +00:00
this.stream.send('ch', {
2018-10-10 11:59:10 +00:00
id: id,
2018-10-08 16:50:49 +00:00
type: type,
body: body
});
}
2018-10-09 06:13:06 +00:00
public abstract dispose(): void;
}
class SharedConnection extends Connection {
2018-10-10 11:59:10 +00:00
private pool: Pool;
2018-10-10 11:59:10 +00:00
public get id(): string {
return this.pool.id;
}
constructor(stream: Stream, channel: string, pool: Pool) {
super(stream, channel);
2018-10-10 11:59:10 +00:00
this.pool = pool;
this.pool.inc();
}
@autobind
2018-10-10 11:59:10 +00:00
public send(typeOrPayload, payload?) {
super.send(this.pool.id, typeOrPayload, payload);
}
@autobind
public dispose() {
2018-10-10 11:59:10 +00:00
this.pool.dec();
this.removeAllListeners();
this.stream.removeSharedConnection(this);
}
}
class NonSharedConnection extends Connection {
2018-10-10 11:59:10 +00:00
public id: string;
protected params: any;
constructor(stream: Stream, channel: string, params?: any) {
2018-10-10 11:59:10 +00:00
super(stream, channel);
this.params = params;
2018-10-13 10:16:47 +00:00
this.id = Math.random().toString().substr(2, 8);
2018-10-10 11:59:10 +00:00
this.connect();
}
@autobind
public connect() {
this.stream.send('connect', {
channel: this.channel,
id: this.id,
params: this.params
});
}
@autobind
public send(typeOrPayload, payload?) {
super.send(this.id, typeOrPayload, payload);
}
@autobind
public dispose() {
this.removeAllListeners();
this.stream.send('disconnect', { id: this.id });
this.stream.disconnectToChannel(this);
}
}