go写的简易聊天
2019-05-13 本文已影响25人
学不会swoole不改名
学习go的小进步
server 代码
//chat server listen 8080
package main
import (
"fmt"
"log"
"net"
"os"
"strings"
)
const (
LOG_DIRECTORY = "./test.log" //记录错误日志的路径
)
var online = make(map[string]net.Conn)
var msgQyeue = make(chan string, 1000)
var quitChan = make(chan bool)
var logFile *os.File
var logger *log.Logger
func coninfo(con net.Conn) {
//make a buff 消息的缓冲区
buff := make([]byte, 1024)
//最后记得关掉 将退出连接的请求删掉
defer func(con net.Conn) {
addr := fmt.Sprintf("%s", con.RemoteAddr())
delete(online, addr)
con.Close()
}(con)
//不断的读取
for {
numBuff, err := con.Read(buff)
if err != nil {
continue
}
if numBuff != 0 {
msgQyeue <- string(buff[:numBuff]) //细节 因为我们最大1024字节,会有上一条消息的缓存 生产者
}
}
}
//消费数据
func outMsg() {
for {
select {
case msg := <-msgQyeue:
//doMsg(<-msgQyeue) 这样写client接收不到返回的消息
doMsg(msg)
case <-quitChan:
break
}
}
}
//对数据做处理 127.0.0.0:1234#你好
func doMsg(msg string) {
content := strings.Split(msg, "#") //按照#切分数据
if len(content) > 1 {
addr := content[0]
sendMsg := strings.Join(content[1:], "#") //127.0.0.1 #(你好#xxx)->连接起来
addr = strings.Trim(addr, " ") //对空格进行处理相当于phptrim
if con, ok := online[addr]; ok {
_, err := con.Write([]byte(sendMsg))
if err != nil {
fmt.Println("send msg is nil")
}
}
} else {
content := strings.Split(msg, "*") //自定义规则按照*切分数据
if strings.ToUpper(content[1]) == "LIST" {
var ips = ""
for i := range online {
ips = ips + "|" + i
}
if con, ok := online[content[0]]; ok {
_, err := con.Write([]byte(ips))
if err != nil {
fmt.Println("send msg is nil")
}
}
}
}
}
func main() {
//打开日志文件
logFile, err := os.OpenFile(LOG_DIRECTORY, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0)
if nil != err {
logger.Println(err)
os.Exit(-1)
}
defer logFile.Close()
logger = log.New(logFile, "\r\n", log.Ldate|log.Ltime|log.Llongfile)
//server 监听的端口
socketer, err := net.Listen("tcp", "localhost:8080")
if err != nil {
fmt.Println(err)
}
defer socketer.Close()
fmt.Println("wait for message……")
logger.Println("i am writing the log")
go outMsg()
for {
//accept是建立链接后进行通信的 如果不用协程效果是你说一句我说一句
con, err := socketer.Accept()
if err != nil {
fmt.Println(err)
}
//将con 映射到map里面
//con.RemoteAddr() 获取cil ip和端口
addr := fmt.Sprintf("%s", con.RemoteAddr())
online[addr] = con
for i := range online {
fmt.Printf("%s\n", i)
logger.Printf("%s\n", i)
}
//协程read
go coninfo(con)
}
}
client 代码
//chat here is cilent
package main
import (
"bufio"
"strings"
"fmt"
"net"
"os"
)
func checkErr(err error) {
if err != nil {
fmt.Println(err)
}
}
func sendMessage(con net.Conn) {
var input string //没什么暖用的定义 ,就是为了byte少了一层string的转换
//cil协程write server协程read
for {
reader := bufio.NewReader(os.Stdin) //bufio来读取标准输入
data, _, _ := reader.ReadLine() //返回你输入那一行的data(数据)
input = string(data) //感觉没什么暖用,还是写上了
if strings.ToUpper(input) == "EXIT" { //如果 输入的是EXIT就退出server,退出前关闭close这是一个好习惯
con.Close()
break
}
_, err := con.Write(data) //通过write来写数据 server通过read来读数据
if err != nil {
con.Close() //同上
break
}
}
}
func main() {
con, err := net.Dial("tcp", "localhost:8080") //连接server的ip和server的端口
checkErr(err)
defer con.Close() //好习惯 nbb
go sendMessage(con) //协程写入
// checkErr(err)
buf := make([]byte, 1024)
for {
numOfBytes, err := con.Read(buf) //localhost:8080读取数据
if err != nil {
fmt.Println("您已经退出,感谢您的使用")
os.Exit(0)
}
// if string(buf[:numOfBytes]) == "你好" {
// fmt.Println("recv meg is:" + string("你好大傻逼"))
// return
// }
fmt.Println("recv meg is:" + string(buf[:numOfBytes]))
}
fmt.Println("message is end !")
}
