Golang TCP服务器群聊消息

main.go:

 

package main

import (
	"fmt"
)

func main()  {
	server := NewServer("127.0.0.1",8888)
	if !server.Start() {
		fmt.Println("服务器启动失败")
		return
	}

}

 

 

server.go

增加广播函数,轮询往当前在线的设备中发送数据。

 

package main

import (
	"fmt"
	"net"
	"sync"
)

type Server struct {
	Ip string
	Port int

	users map[string]*User
	userMutex sync.RWMutex
	msgC chan string
}

func NewServer(ip string,port int) *Server  {
	server := &Server{
		Ip: ip,
		Port: port,
		users: make(map[string]*User),
		msgC: make(chan string),
	}
	return server
}

func (this *Server) handler(conn net.Conn)  {
	fmt.Println(fmt.Sprintf("%s: 连接成功",conn.RemoteAddr().String()))
	this.userMutex.Lock()
	this.users[conn.RemoteAddr().String()] = NewUser(conn,this)
	this.userMutex.Unlock()
}

func (this *Server) Start() bool {
	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d",this.Ip,this.Port))
	defer listener.Close()

	if err != nil {
		fmt.Println("net listen err:",err)
		return false
	}

	go this.listenMsg()

	for  {
		con, err := listener.Accept()
		if err != nil {
			fmt.Println("net Accept err:",err)
			continue
		}
		this.userOnline(con.RemoteAddr().String())
		go this.handler(con)
	}

	return true
}

func (this *Server) listenMsg()  {
	for  {
		msg:= <- this.msgC
		this.brodCast(msg)
	}
}

func (this *Server) brodCast(msg string )  {
	this.userMutex.Lock()
	for _, v := range this.users {
		sendMsg := "[" + v.Conn.RemoteAddr().String()+"]"+v.Name+":"+msg+"\n"
		_,err := v.Conn.Write([]byte(sendMsg))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

func (this *Server) userOnline(name string)  {
	this.userMutex.Lock()
	for _, v := range this.users {
		_,err := v.Conn.Write([]byte(fmt.Sprintf("user:%s online",name)))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

func (this *Server) userOffline(ip string)  {
	this.userMutex.Lock()
	username := this.users[ip].Name
	delete(this.users, ip)
	for _, v := range this.users {
		_,err := v.Conn.Write([]byte(fmt.Sprintf("user:%s offline",username)))
		if err != nil {
			fmt.Println("server brodCast err:",err)
		}
	}
	this.userMutex.Unlock()
}

 

 

user.go

在消息监听函数中不断的阻塞读取消息,如果数据为0设备下线。

如果数据开头是all:则判断为广播数据,轮询往当前在线的设备中发送数据。

 

package main

import (
	"fmt"
	"net"
	"runtime"
	"strings"
)

type User struct {
	Name string
	Address string
	C chan string
	Conn net.Conn
	server *Server
}

func NewUser(conn net.Conn,ser *Server) *User  {
	user := &User{
		Name: conn.RemoteAddr().String(),
		Address: conn.RemoteAddr().String(),
		C: make(chan string),
		Conn: conn,
		server: ser,
	}

	go user.listenMessage()

	return user
}

func (this *User) listenMessage()  {

	for  {
			msg :=make([]byte,4096)
			num,err :=this.Conn.Read(msg)
			if err != nil {
				fmt.Println("user Read err:",err)
			}
			if num==0 {
				fmt.Println(fmt.Sprintf("user:%s device offline.",this.Conn.RemoteAddr().String()))
				this.server.userOffline(this.Conn.RemoteAddr().String())
				_ = this.Conn.Close()
				close(this.C)
				runtime.Goexit()
			}

		msgStr := string(msg[:num])
		if len(msgStr)>4 && msgStr[:4]=="all:" {
			//fmt.Println(strings.Split(msgStr,":")[1])
			this.server.brodCast(strings.Split(msgStr,":")[1])
		}
	}
}

 

 

 

 

 

目录:

Golang 从入门到放弃

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在 Go 中,可以通过 `net.Conn` 接口来写入 TCP 数据包。如果要修改已经发送的数据包,则需要使用底层的 socket API。 下面是一个示例,演示如何使用 Go 的 `syscall` 包来修改已经发送的 TCP 数据包: ```go package main import ( "fmt" "net" "syscall" ) func main() { // 连接到服务器 conn, err := net.Dial("tcp", "localhost:8080") if err != nil { panic(err) } defer conn.Close() // 发送一些数据 message := []byte("hello") _, err = conn.Write(message) if err != nil { panic(err) } // 修改已经发送的数据 fd, err := syscall.Socket(syscall.AF_INET, syscall.SOCK_RAW, syscall.IPPROTO_TCP) if err != nil { panic(err) } defer syscall.Close(fd) // 构造 IP 数据包头部 ipHeader := []byte{ 0x45, 0x00, 0x00, 0x28, 0x00, 0x00, 0x40, 0x00, 0x40, 0x06, 0x00, 0x00, 0x7f, 0x00, 0x00, 0x01, 0x7f, 0x00, 0x00, 0x01, } // 构造 TCP 数据包头部 tcpHeader := []byte{ 0x00, 0x50, 0x00, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, } // 构造 TCP 数据 data := []byte("modified") // 将 IP 和 TCP 头部、数据拼接起来 packet := append(ipHeader, tcpHeader...) packet = append(packet, data...) // 发送数据包 err = syscall.Sendto(fd, packet, 0, &syscall.SockaddrInet4{ Port: 8080, Addr: [4]byte{127, 0, 0, 1}, }) if err != nil { panic(err) } fmt.Println("修改数据成功") } ``` 在上面的示例中,我们首先通过 `net.Dial()` 方法连接到服务器,然后发送了一些数据。接着,我们使用 `syscall.Socket()` 创建了一个原始的 TCP 套接字,利用这个套接字可以构造和发送 TCP 数据包。我们构造了 IP 和 TCP 头部,将它们与修改过的数据拼接起来,最后通过 `syscall.Sendto()` 发送数据包。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值