-
Notifications
You must be signed in to change notification settings - Fork 41
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
19ae34d
commit a55bb46
Showing
9 changed files
with
216 additions
and
47 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,99 @@ | ||
| /** | ||
| * Welcome to Cloudflare Workers! This is your first worker. | ||
| * | ||
| * - Run "npm run dev" in your terminal to start a development server | ||
| * - Open a browser tab at http://localhost:8787/ to see your worker in action | ||
| * - Run "npm run deploy" to publish your worker | ||
| * | ||
| * Learn more at https://developers.cloudflare.com/workers/ | ||
| */ | ||
|
|
||
| export default { | ||
| async fetch(request, env, ctx) { | ||
| // 创建一个新的 URL 对象 | ||
| let url = new URL(request.url); | ||
| let path = url.pathname.substring(1); | ||
| let isSecure = url.protocol.startsWith("https"); | ||
|
|
||
| if (path === "ws") { | ||
| return handleWSRequest(request); | ||
| } else if (path === "locations") { | ||
| let targetUrl = `http${isSecure ? 's' : ''}://speed.cloudflare.com/locations`; | ||
| let cfRequest = new Request(targetUrl, request); | ||
| let response = await fetch(cfRequest); | ||
| return response; | ||
| } else { | ||
| return handleSpeedTestRequest(request, path, isSecure); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| async function handleSpeedTestRequest(request, path, isSecure) { | ||
| let bytes; | ||
| if (!path) { | ||
| // 路径为空，将 bytes 赋值为 100MB | ||
| bytes = 100000000; | ||
| } else { | ||
| // 其他路径，进行正常的处理 | ||
| const regex = /^(\d+)([a-z]?)$/i; | ||
| const match = path.match(regex); | ||
| if (!match) { | ||
| // 路径格式不正确，返回错误 | ||
| return new Response("路径格式不正确", { | ||
| status: 400, | ||
| }); | ||
| } | ||
|
|
||
| const bytesStr = match[1]; | ||
| const unit = match[2].toLowerCase(); | ||
|
|
||
| // 转换单位 | ||
| bytes = parseInt(bytesStr, 10); | ||
| if (unit === "k") { | ||
| bytes *= 1000; | ||
| } else if (unit === "m") { | ||
| bytes *= 1000000; | ||
| } else if (unit === "g") { | ||
| bytes *= 1000000000; | ||
| } | ||
| } | ||
|
|
||
| let targetUrl = `http${isSecure ? 's' : ''}://speed.cloudflare.com/__down?bytes=${bytes}`; | ||
| let cfRequest = new Request(targetUrl, request); | ||
| let response = await fetch(cfRequest); | ||
|
|
||
| // 将测试结果反馈给用户 | ||
| return response; | ||
| } | ||
|
|
||
| async function handleWSRequest(request) { | ||
| const upgradeHeader = request.headers.get('Upgrade'); | ||
| if (!upgradeHeader || upgradeHeader !== 'websocket') { | ||
| return new Response('Expected Upgrade: websocket', { status: 426 }); | ||
| } | ||
|
|
||
| /** @type {import("@cloudflare/workers-types").WebSocket[]} */ | ||
| // @ts-ignore | ||
| const webSocketPair = new WebSocketPair(); | ||
| const [client, webSocket] = Object.values(webSocketPair); | ||
|
|
||
| handleSession(webSocket); | ||
|
|
||
| return new Response(null, { | ||
| status: 101, | ||
| // @ts-ignore | ||
| webSocket: client, | ||
| }); | ||
| } | ||
|
|
||
| async function handleSession(websocket) { | ||
| websocket.accept() | ||
| websocket.addEventListener("message", async message => { | ||
| console.log(message) | ||
| }) | ||
|
|
||
| websocket.addEventListener("close", async evt => { | ||
| // Handle when a client closes the WebSocket connection | ||
| console.log(evt) | ||
| }) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters