前一篇说了聊天室客户端的实现，这一篇就接着上篇讲服务端，与很多聊天室不同，我在服务端里加入了管理员，管理员与普通用户的功能相比多了屏蔽发言，以及踢出群聊的功能
数据的接收与发送
和客户端的没什么不同
func recv(recvData chan string, conn net.Conn) {
for {
buf := make([]byte, 1024)
n, err := conn.Read(buf)
if err != nil {
connection <- conn
return
}
recvData <- string(buf[:n])
}
}
/**
* 数据发送
* sentData 接收数据Channel
* conn TCP连接对象
**/
func sent(sentData chan string, conn net.Conn) {
for {
data := <-sentData
_, err := conn.Write([]byte(data))
if err != nil {
connection <- conn
return
}
}
}
客户端的连接管理
func connManager(connection chan net.Conn) {
for {
conn := <-connection
username := ipToUname[conn.RemoteAddr().String()]
notesInfo(messages, "用户:"+username+"已退出聊天室！")
conn.Close()
delete(conns, username)
delete(ipToUname, conn.RemoteAddr().String())
}
}
数据解析与封装
func dataSent(conns *map[string]clientInfo, messages chan string) {
for {
msg := <-messages
fmt.Println(msg)
data := strings.Split(msg, "-") // 聊天数据分析:
length := len(data)
if length == 2 { // 管理员单个用户发送控制命令
(*conns)[data[0]].sentData <- data[1]
} else if length == 3 { // 用户列表
(*conns)[data[1]].sentData <- data[2]
} else if length == 4 { // 向单个用户发送数据
msg = data[1] + " say to you : " + data[3]
(*conns)[data[2]].sentData <- msg
} else {
// 群发
for _, value := range *conns {
value.sentData <- msg
}
}
}
}
管理员注册处理
/**
* 获取全部已注册用户的数据,用于登录注册时验证
*/
func getAllUser(filename string) map[string]userData {
buf := make([]byte, dataSize)
udata := make(map[string]userData)
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, 0766) //打开保存用户的文件
defer file.Close() //保证文件关闭
errorExit(err, "getAllUser")
n, _ := file.Read(buf)
json.Unmarshal(buf[:n], &udata) //将流文件转成json并赋值给udata这个map
return udata
}
/**
* 添加新注册用户数据
*/
func insertNewUser(filename, username, password string) {
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, 0766)
defer file.Close()
checkError(err, "insertNewUser")
uData[username] = userData{password, 1}
data, _ := json.MarshalIndent(uData, "", " ")
file.WriteString(string(data))
}
func regProcess(username, password string) (status bool, info string) {
if len(uData) == maxReg {
return false, "注册人数已满！"
}
if _, ok := uData[username]; ok {
return false, "用户已存在，请更换注册名!"
} else {
insertNewUser(dataFileName, username, password)
return true, success
}
}
管理员登录处理
func logProcess(username, password string) (status bool, info string) {
if len(conns) == maxLog {
return false, "当前登录人数已满,请稍后登录"
}
if user, ok := uData[username]; !ok {
return false, "用户名或密码错误!"
} else {
if user.Password == password {
if _, ok := conns[username]; ok {
return false, "用户已登录！"
} else {
return true, success
}
} else {
return false, "用户名或密码错误!"
}
}
}
用户进入聊天室资格认证
func userAuth(conn *net.Conn, recvData, sentData chan string) {
for {
data := strings.Split(<-recvData, "-") // 等待用户发送登录或注册数据
flag, username, password := data[0], data[1], data[2]
if flag == "Reg" {
_, info := regProcess(username, password)
sentData <- info
} else {
status, info := logProcess(username, password)
sentData <- info
if status == true {
messages <- ("用户:" + username + "进入聊天室") // 用户登录成功，发送系统通知
conns[username] = clientInfo{*conn, sentData} // 记录用户登录状态信息
ipToUname[(*conn).RemoteAddr().String()] = username // 记录IP与用户名对应信息
messages <- "List-" + username + "-" + userList() // 向用户发送已在线用户列表
go dataRec(&conns, username, recvData, messages) // 开启服务器对该客户端数据接收线程
break
}
}
}
}
服务端的tcp连接创建
func createTCP(tcpaddr string) *net.TCPListener {
tcpAddr, err := net.ResolveTCPAddr("tcp4", tcpaddr)
errorExit(err, "ResolveTCPAddr")
l, err := net.ListenTCP("tcp", tcpAddr)
errorExit(err, "DialTCP")
return l
}
管理员的登录注册
func intoManager() {
fmt.Println("请输入将要进行 *** 作：1、管理员注册 2、管理员登录")
var input string
LOOP:
{
fmt.Scanln(&input)
switch input {
case "1":
adminReg()
case "2":
adminLog()
default:
goto LOOP
}
}
admimManager(messages)
}
/**
* 管理员登录程序
**/
func adminLog() {
for {
var adminname, password string
fmt.Println("请输入管理员用户名：")
fmt.Scanln(&adminname)
fmt.Println("请输入管理员密码：")
fmt.Scanln(&password)
if pwd, ok := adminList[adminname]; !ok {
fmt.Println("用户名或者密码错误")
} else {
if pwd != password {
fmt.Println("用户名或者密码错误！")
} else {
fmt.Println("登录成功！")
break
}
}
}
}
/**
* 管理员注册程序
**/
func adminReg() {
var adminname, password string
fmt.Println("请输入管理员用户名：")
fmt.Scanln(&adminname)
fmt.Println("请输入管理员密码：")
fmt.Scanln(&password)
adminList[adminname] = password //将注册的管理员姓名密码保存到adminList中，单次启动有效
fmt.Println("注册成功！请登录")
adminLog() //跳转到登录
}
管理员一系列功能实现
func admimManager(messages chan string) {
for {
var input, objUser string
fmt.Scanln(&input)
switch input {
case "/to":
fmt.Println(userList())
fmt.Println("请输入聊天对象：")
fmt.Scanln(&objUser)
if _, ok := conns[objUser]; !ok {
fmt.Println("不存在此用户!")
} else {
fmt.Println("请输入消息：")
fmt.Scanln(&input)
notesInfo(messages, "To-Manager-"+objUser+"-"+input)
}
case "/all":
fmt.Println("请输入消息：")
fmt.Scanln(&input)
notesInfo(messages, "Manager say : "+input)
case "/shield":
fmt.Println(userList())
fmt.Println("请输入屏蔽用户名：")
fmt.Scanln(&objUser)
notesInfo(messages, objUser+"-/shield")
notesInfo(messages, "用户："+objUser+"已被管理员禁言！")
case "/remove":
fmt.Println(userList())
fmt.Println("请输入踢出用户名：")
fmt.Scanln(&objUser)
notesInfo(messages, "用户："+objUser+"已被管理员踢出聊天室！")
if _, ok := conns[objUser]; !ok {
fmt.Println("不存在此用户!")
} else {
conns[objUser].conn.Close() // 删除该用户的连接
delete(conns, objUser) // 从已登录的列表中删除该用户
}
}
}
}
服务端启动程序
func StartServer(port string) {
l := createTCP(":" + port)
fmt.Println("服务端启动成功，正在监听端口！")
go dataSent(&conns, messages) // 启动服务器广播线程
go intoManager() // 启动管理模块
go connManager(connection) // 启动连接管理线程
for {
conn, err := l.Accept()
if checkError(err, "Accept") == false {
continue
}
fmt.Println("客户端:", conn.RemoteAddr().String(), "连接服务器成功！")
var recvData = make(chan string)
var sentData = make(chan string)
go recv(recvData, conn) // 开启对客户端的接受数据线程
go sent(sentData, conn) // 开启对客户端的发送数据线程
go userAuth(&conn, recvData, sentData) // 用户资格认证
}
}
服务端主程序
func main() {
if len(os.Args) == 2 {
uData = getAllUser(dataFileName) // 用户登录、注册数据初始化
StartServer(os.Args[1]) // 启动客户端
} else {
fmt.Println("输入错误！")
}
}
用到的标准库
import (
"encoding/json"
"fmt"
"net"
"os"
"strings"
)
总结
终于写完了，啊啊啊！
源码地址
https://gitee.com/qian-xiawudi/test1
欢迎分享，转载请注明来源：内存溢出
微信扫一扫
支付宝扫一扫
评论列表（0条）