63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package test
|
||
|
||
import (
|
||
"fmt"
|
||
"golang.org/x/net/websocket"
|
||
"log"
|
||
"net/http"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
func echoHandler(ws *websocket.Conn) {
|
||
msg := make([]byte, 512)
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
fmt.Println(err) //这里的err其实就是panic传入的内容，55
|
||
}
|
||
}()
|
||
for {
|
||
n, err := ws.Read(msg)
|
||
if err != nil {
|
||
log.Println(err.Error())
|
||
return
|
||
}
|
||
fmt.Printf("Receive: %s\n", msg[:n])
|
||
|
||
send_msg := "[" + string(msg[:n]) + "]"
|
||
m, err := ws.Write([]byte(send_msg))
|
||
if err != nil {
|
||
log.Println(err.Error())
|
||
return
|
||
}
|
||
fmt.Printf("Send: %s\n", msg[:m])
|
||
}
|
||
}
|
||
|
||
var staticfs = http.FileServer(http.Dir("E:\\"))
|
||
|
||
//这里可以自行定义安全策略
|
||
func static(w http.ResponseWriter, r *http.Request) {
|
||
//fmt.Printf("访问静态文件:%s\n", r.URL.Path)
|
||
old := r.URL.Path
|
||
r.URL.Path = strings.Replace(old, "/file/", "", 1)
|
||
staticfs.ServeHTTP(w, r)
|
||
}
|
||
|
||
func TestMain(t *testing.T) {
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
fmt.Println(err) //这里的err其实就是panic传入的内容，55
|
||
}
|
||
}()
|
||
|
||
http.Handle("/echo", websocket.Handler(echoHandler))
|
||
http.HandleFunc("/file/", static)
|
||
|
||
err := http.ListenAndServe("localhost:8080", nil)
|
||
|
||
if err != nil {
|
||
panic("ListenAndServe: " + err.Error())
|
||
}
|
||
}
|