go写的简易聊天

学习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 !")

}

默认client发送需要另一个client的ip#content的格式,这个格式自定义,过几天用swoole实现以下这个,对比以下如果查看ip list 格式是ip*list这个也可以自定义

注释很详细,逻辑很清晰,给自己一个赞