Golang 搭建Tcp服务器

富竹 · · 2916 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

使用Go搭建Tcp服务器,数据库使用MongoDB,代码复制修改参数即可使用。

package main

import (
    "fmt"
    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
    "net"
)

type Message struct {
    _Id    bson.ObjectId
    Client string
    Msg    string
}

func main() {
    session, coll := ConnectMongo()
    defer session.Close()

    lner, err := net.Listen("tcp", "192.168.124.65:8080")
    if err != nil {
        fmt.Println("Listener create error: ", err)
        return
    }
    fmt.Println("Waiting for client...")
    for {
        conn, err := lner.Accept()
        if err != nil {
            fmt.Println("Accept error: ", err)
            return
        }
        go handleConnection(coll, conn)
    }
}

func handleConnection(coll *mgo.Collection, conn net.Conn) {
    clientAddr := conn.RemoteAddr().String()
    fmt.Println("Connection success. Client address: ", clientAddr)
    defer conn.Close()

    for {
        buffer := make([]byte, 1024)
        recvLen, err := conn.Read(buffer)
        if err != nil {
            fmt.Println("Read error: ", err, clientAddr)
            return
        }
        strBuffer := string(buffer[:recvLen])
        fmt.Println("Client message: ", strBuffer)
        InsertMongo(coll, clientAddr, strBuffer)
    }
}

func ConnectMongo() (*mgo.Session, *mgo.Collection) {
    session, err := mgo.Dial("192.168.8.10:27017")
    if err != nil {
        panic(err)
    }

    session.SetMode(mgo.Monotonic, true)
    coll := session.DB("tcp_server").C("tcp_data")
    return session, coll
}

func InsertMongo(coll *mgo.Collection, client, msg string) {
    insertMsg := Message{
        Client: client,
        Msg:    msg,
    }

    err := coll.Insert(&insertMsg)
    if err != nil {
        fmt.Println("Insert mongodb error: ", err)
    }
}

有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:富竹

查看原文:Golang 搭建Tcp服务器

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2916 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传