GOLANG和websocket 写的简易聊天室

一键远控手机电脑

已于 2022-10-21 11:22:02 修改

阅读量701

点赞数

分类专栏： Go 文章标签： golang websocket 开发语言

于 2022-10-20 16:09:08 首次发布

本文链接：https://blog.csdn.net/shelutai/article/details/127429235

版权

Go 专栏收录该内容

48 篇文章 2 订阅

订阅专栏

main.go

package main

import (
	"fmt"
	"net/http"
	"time"

	"golang.org/x/net/websocket"
)

type Connect struct {
	wc *websocket.Conn
	sc chan []byte
}

var H = make(map[*Connect]bool, 2)

func (c *Connect) writer() {
	for msg := range c.sc {

		fmt.Printf("%#v\n", c.sc)

		c.wc.Write(msg)

	}
}

func (c *Connect) reader() {
	for range time.NewTicker(time.Second).C {

		buf := make([]byte, 1500)
		n, err := c.wc.Read(buf)
		if err != nil {
			fmt.Println(err)
		}

		fmt.Println(string(buf[:n]))
		if len(string(buf[:n])) < 1 {
			continue
		}

		for conn := range H {
			// fmt.Printf("%#v\n", conn)
			conn.sc <- buf[:n]
		}

	}
}

func msg(wc *websocket.Conn) {

	c := &Connect{
		wc: wc,
		sc: make(chan []byte),
	}

	// H = Hub{r: make(chan *Connect, 100)}

	if c.wc != nil {
		H[c] = true
	}

	go c.writer()
	c.reader()
}

func main() {

	http.Handle("/", http.FileServer(http.Dir("./")))

	http.Handle("/msg", websocket.Handler(msg))

	fmt.Println("start server at 8081")

	http.ListenAndServe(":8081", nil)
}

index.html

<html>
    <head>
        <title>socket test</title>
    </head>

    <script>
        const socket = new WebSocket(`ws://${window.location.host}/msg`);

        socket.onmessage = e => {
            document.getElementById("msg").innerHTML += e.data + "<br />";
        }
        
        function submit() {
            let inputmsg  = document.getElementById("name").value +"说：" +document.getElementById("input").value;

            console.log(inputmsg)

            socket.send(inputmsg)
        }
    </script>

    <body>
        <div> Name:<input type="text" id="name" value="小明" /> </div>
        <div> Massage:<input type="text" id="input" /> <input type="submit" onclick="submit()"> </div>
        Content:

        <div id="msg"></div>
        
    </body>
</html>