微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

golang实现的私聊服务

一个简单的golang实现的私聊,由服务器端决定两两匹配
服务器端 chat_server.go

package main


import (
    "fmt"
    "io"
    "log"
    "net"
)


const listenAddr = "localhost:4000"

func main() {
    l,err := net.Listen("tcp",listenAddr)
    if err != nil {
        log.Fatal(err)
    }
    for {
        c,err := l.Accept()
        if err != nil {
        log.Fatal(err)
    }
    //go io.copy(c,c)
    go match(c)
    }
}


var partner = make(chan io.ReadWriteCloser)


func match(c io.ReadWriteCloser) {
    fmt.Fprint(c,"Waiting for a partner...")
    select {
    case partner <- c:
        // Now handled by the other goroutine
    case p := <-partner:
    chat(p,c)
    }
}


/*
func chat(a,b io.ReadWriteCloser) {
    fmt.Fprintln(a,"Found one! Say hi.")
    fmt.Fprintln(b,"Found one! Say hi.")
    go io.copy(a,b)
    io.copy(b,a)
}
*/


func chat(a,"Found one! Say hi.")
    errc := make(chan error,1)
    go cp(a,b,errc)
    go cp(b,a,errc)
    if err := <-errc; err != nil {
        log.Println(err)
    }
    a.Close()
    b.Close()
}

func cp(w io.Writer,r io.Reader,errc chan<- error)        {
    _,err := io.copy(w,r)
    errc <- err
}

客户端chat_client.go:

package main
import (
    "fmt"
    "log"
    "net"
)

func main() {
    l,err := net.Dial("tcp","localhost:4000")
    if err != nil {
        log.Fatal(err)
    }
    go send(l)
    for {
        datas := make([]byte,255)
        n,err := l.Read(datas)
        if err != nil {
            log.Fatal(err)
        }
    fmt.Println(string(datas[0:n]))
    }   
}


func send(l net.Conn){
    sms := make([]byte,255)
    for{
        //fmt.Print("请输入要发送的消息:")
        _,err := fmt.Scan(&sms)
        if err != nil {
            fmt.Println("数据输入异常:",err.Error())
        }

        l.Write(sms)
    }
}

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐