Golang实现并发聊天室

package mainimport ( "fmt" "net")type client struct { name string addr string pipe chan string}var Message = make(chan string)var onlinemap map[string]client = make(map[string]client)...
摘要由CSDN通过智能技术生成
package main

import (
	"fmt"
	"net"
)

type client struct {
	name string
	addr string
	pipe chan string
}

var Message = make(chan string)
var onlinemap map[string]client = make(map[string]client)

func clientToServer(client client, conn net.Conn) {
	for m := range client.pipe {
		conn.Write([]byte(m + "\n"))
	}
}

func sendMessage(name string, addr string, message string) string {
	return name + ":" + message
}

func broadcast() {
	for {
		msg := <-Message
		for _, s := range onlinemap {
			s.pipe <- msg
		}
	}
}

func handelRequest(conn net.Conn) {
	//defer conn.Close()
	addr := conn.RemoteAddr().String()
	fmt.Println(addr + "enter")
	client := client{addr, addr, make(chan string)}
	go clientToServer(client, conn)
	onlinemap[addr] = client
	Message <- sendMessage(client.name, addr, "login") //上线提示
	go func() {
		buf := make([]byte, 512)
		for {
			//fmt.Println("Please Enter Your Message:")
			n, _ := conn.Read(buf)
			if n == 0 {
				fmt.Println(client.name + "leave")
				return
			} else {
				Message <- sendMessage(client.name, addr, string(buf[:n-1]))
			}
		}
	}()
}
func main() {
	fmt.Println("Server start")
	ln, _ := net.Listen("tcp", "0.0.0.0:8888")
	defer ln.Close()
	go broadcast()
	for {
		conn, err := ln.Accept()
		if err != nil {
			fmt.Println("Accept error")
			continue
		}
		go handelRequest(conn)
	}
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值