2018-12-17 16:34:32 +00:00
|
|
|
import App, { Container } from 'next/app'
|
2018-06-23 20:17:37 +00:00
|
|
|
import React from 'react'
|
|
|
|
import io from 'socket.io-client'
|
|
|
|
|
|
|
|
class MyApp extends App {
|
|
|
|
static async getInitialProps ({ Component, ctx }) {
|
|
|
|
let pageProps = {}
|
|
|
|
|
|
|
|
if (Component.getInitialProps) {
|
|
|
|
pageProps = await Component.getInitialProps(ctx)
|
|
|
|
}
|
|
|
|
|
|
|
|
return { pageProps }
|
|
|
|
}
|
|
|
|
state = {
|
|
|
|
socket: null
|
|
|
|
}
|
|
|
|
componentDidMount () {
|
|
|
|
// connect to WS server and listen event
|
|
|
|
const socket = io()
|
|
|
|
this.setState({ socket })
|
|
|
|
}
|
|
|
|
|
|
|
|
// close socket connection
|
|
|
|
componentWillUnmount () {
|
|
|
|
this.state.socket.close()
|
|
|
|
}
|
|
|
|
|
|
|
|
render () {
|
2018-12-17 16:34:32 +00:00
|
|
|
const { Component, pageProps } = this.props
|
2018-06-23 20:17:37 +00:00
|
|
|
return (
|
|
|
|
<Container>
|
|
|
|
<Component {...pageProps} socket={this.state.socket} />
|
|
|
|
</Container>
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
export default MyApp
|