feat: support http2

close #424
vitejs · Jun 22, 2020 · f4fd832 · f4fd832
1 parent df7f335
commit f4fd832
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/src/client/client.ts b/src/client/client.ts
@@ -50,8 +50,7 @@ console.log('[vite] connecting...')
 
 declare var __VUE_HMR_RUNTIME__: HMRRuntime
 
-const socketProtocol = location.protocol === 'https:' ? 'wss' : 'ws'
-const socket = new WebSocket(`${socketProtocol}://${location.host}`)
+const socket = new WebSocket(`ws://${location.hostname}:24678`)
 
 function warnFailedFetch(err: Error, path: string | string[]) {
   if (!err.message.match('fetch')) {
@@ -140,12 +139,9 @@ socket.addEventListener('message', async ({ data }) => {
 socket.addEventListener('close', () => {
   console.log(`[vite] server connection lost. polling for restart...`)
   setInterval(() => {
-    new WebSocket(`${socketProtocol}://${location.host}`).addEventListener(
-      'open',
-      () => {
-        location.reload()
-      }
-    )
+    new WebSocket(`ws://${location.host}`).addEventListener('open', () => {
+      location.reload()
+    })
   }, 1000)
 })
 

diff --git a/src/node/server/index.ts b/src/node/server/index.ts
@@ -116,7 +116,7 @@ function resolveServer(
   requestListener: RequestListener
 ) {
   if (https) {
-    return require('https').createServer(
+    return require('http2').createSecureServer(
       resolveHttpsConfig(httpsOption),
       requestListener
     )

diff --git a/src/node/server/serverPluginHmr.ts b/src/node/server/serverPluginHmr.ts
@@ -114,7 +114,7 @@ export const hmrPlugin: ServerPlugin = ({
   })
 
   // start a websocket server to send hmr notifications to the client
-  const wss = new WebSocket.Server({ server })
+  const wss = new WebSocket.Server({ port: 24678 })
 
   wss.on('connection', (socket) => {
     debugHmr('ws client connected')