首页 > 其他分享 >即时通讯系统 -- V1.0客户端全功能实现

即时通讯系统 -- V1.0客户端全功能实现

时间:2023-03-09 20:44:31浏览次数:60  
标签:chatMsg err -- fmt 全功能 V1.0 client Println conn

实现了客户端的菜单功能,分为

  1. 公聊模式

  2. 私聊模式

  3. 更新用户名

  4. 退出
    并实现了这四个功能,本次的简易即时聊天系统到此结束

  5. 公聊模式

func (client *Client) PublicChat() {
	//发给服务器
	for {
		var chatMsg string
		fmt.Println(">>>>>>请输入聊天内容, exit退出。")
		fmt.Scanln(&chatMsg)
		if chatMsg == "exit" {
			break
		}
		if len(chatMsg) != 0 {
			sendMsg := chatMsg + "\n"
			_, err := client.conn.Write([]byte(sendMsg))
			if err != nil {
				fmt.Println("conn.Write error:", err)
				break
			}
		}
	}
}
  1. 私聊模式
// 查询在线用户
func (client *Client) SelectUsers() {
	sendMsg := "who\n"
	_, err := client.conn.Write([]byte(sendMsg))
	if err != nil {
		fmt.Println("conn Write error:", err)
		return
	}
}

// 私聊模式
func (client *Client) PrivateChat() {
	for {
		var remoteName string
		client.SelectUsers()
		fmt.Println(">>>>>请选择聊天对象, exit退出")
		fmt.Scanln(&remoteName)
		if remoteName == "exit" {
			break
		}
		for {
			var chatMsg string
			fmt.Println(">>>>>请输入消息内容, exit退出")
			fmt.Scanln(&chatMsg)
			if chatMsg == "exit" {
				break
			}
			if len(chatMsg) != 0 {
				sendMsg := "to|" + remoteName + "|" + chatMsg + "\n\n"
				_, err := client.conn.Write([]byte(sendMsg))
				if err != nil {
					fmt.Println("conn.Write error:", err)
					break
				}
			}
		}
	}
}
  1. 更新用户名
func (client *Client) UpdateName() bool {
	fmt.Println(">>>>>请输入用户名:")
	fmt.Scanln(&client.Name)

	sendMsg := "rename|" + client.Name + "\n"
	_, err := client.conn.Write([]byte(sendMsg))
	if err != nil {
		fmt.Println("conn.Write err:", err)
		return false
	}
	return true
}

客户端完整代码

package main

import (
	"flag"
	"fmt"
	"io"
	"net"
	"os"
)

type Client struct {
	ServerIp   string
	ServerPort int
	Name       string
	conn       net.Conn
	flag       int //当前客户端模式
}

func NewClient(serverIp string, serverPort int) *Client {
	//常见客户端对象
	client := &Client{
		ServerIp:   serverIp,
		ServerPort: serverPort,
		flag:       99,
	}

	//连接server
	conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", serverIp, serverPort))
	if err != nil {
		fmt.Println("net.Dial error:", err)
		return nil
	}

	client.conn = conn
	//返回对象
	return client
}

var serverIp string
var serverPort int

func init() {
	//./client -ip 127.0.0.1 -port 8888
	flag.StringVar(&serverIp, "ip", "127.0.0.1", "设置服务器IP地址(默认是127.0.0.1)")
	flag.IntVar(&serverPort, "port", 8888, "设置服务区端口(默认是8888)")
}

func (client *Client) menu() bool {
	var flag int
	fmt.Println("1.公聊模式")
	fmt.Println("2.私聊模式")
	fmt.Println("3.更新用户名")
	fmt.Println("0.退出")

	fmt.Scanln(&flag)
	if flag >= 0 && flag <= 3 {
		client.flag = flag
	} else {
		fmt.Println(">>>>>请输入合法范围的数<<<<<")
		return false
	}
	return true
}

// 查询在线用户
func (client *Client) SelectUsers() {
	sendMsg := "who\n"
	_, err := client.conn.Write([]byte(sendMsg))
	if err != nil {
		fmt.Println("conn Write error:", err)
		return
	}
}

// 私聊模式
func (client *Client) PrivateChat() {
	for {
		var remoteName string
		client.SelectUsers()
		fmt.Println(">>>>>请选择聊天对象, exit退出")
		fmt.Scanln(&remoteName)
		if remoteName == "exit" {
			break
		}
		for {
			var chatMsg string
			fmt.Println(">>>>>请输入消息内容, exit退出")
			fmt.Scanln(&chatMsg)
			if chatMsg == "exit" {
				break
			}
			if len(chatMsg) != 0 {
				sendMsg := "to|" + remoteName + "|" + chatMsg + "\n\n"
				_, err := client.conn.Write([]byte(sendMsg))
				if err != nil {
					fmt.Println("conn.Write error:", err)
					break
				}
			}
		}
	}
}
func (client *Client) PublicChat() {
	//提示用户输入消息

	//发给服务器
	for {
		//消息不为空
		var chatMsg string
		fmt.Println(">>>>>>请输入聊天内容, exit退出。")
		fmt.Scanln(&chatMsg)
		if chatMsg == "exit" {
			break
		}
		if len(chatMsg) != 0 {
			sendMsg := chatMsg + "\n"
			_, err := client.conn.Write([]byte(sendMsg))
			if err != nil {
				fmt.Println("conn.Write error:", err)
				break
			}
		}
	}
}
func (client *Client) UpdateName() bool {
	fmt.Println(">>>>>请输入用户名:")
	fmt.Scanln(&client.Name)

	sendMsg := "rename|" + client.Name + "\n"
	_, err := client.conn.Write([]byte(sendMsg))
	if err != nil {
		fmt.Println("conn.Write err:", err)
		return false
	}
	return true
}
func (client *Client) Run() {
	for client.flag != 0 {
		for client.menu() != true {
		}
		//根据不同的flag处理不同业务
		switch client.flag {
		case 1:
			client.PublicChat()
		case 2:
			client.PrivateChat()
		case 3:
			client.UpdateName()
		}
	}
}

// 处理 server 回应的消息,直接显示到标准输出
func (client *Client) DealResponse() {
	io.Copy(os.Stdout, client.conn)
	/*
		等价于
		for {
			buf := make()
			client.conn.Read(buf)
			fmt.Println(buf)
		}
	*/
}
func main() {
	//命令行解析
	flag.Parse()
	client := NewClient(serverIp, serverPort)
	if client == nil {
		fmt.Println(">>>>> 连接服务器失败...")
		return
	}
	fmt.Println(">>>>> 连接服务器成功...")

	//单独开启一个 goruntine,处理 server 发回的消息
	go client.DealResponse()
	client.Run()
}

总结:

  1. user 与 server 打交道的功能都是在服务器实现的,在 user.go 和 server.go 中,客户端主要实现一些逻辑并调用他们的功能,比如将 chatMsg 修改为 to|xxx|xxx 的格式,再调用 user.go 中的接口
  2. 要开一个 goruntine 来处理 server 发回的内容,用
    io.Copy(os.Stdout, client.conn)
    更方便
  3. 本次即时通讯系统的开发(抄代码)让我第一次完整的体会到开发的过程,与写算法题相比,虽然不需要冥思苦想一些很精妙的解法,但是统筹考虑,合理设计的难度更加高,尤其是涉及到并发与加锁的时候,还需要积累更多的经验

标签:chatMsg,err,--,fmt,全功能,V1.0,client,Println,conn
From: https://www.cnblogs.com/hzy717zsy/p/17201338.html

相关文章

  • 2019GPLT
    2019GPLT7-26翻了从左到右扫描输入的句子:如果句子中有超过3个连续的6,则将这串连续的6替换成9;但如果有超过9个连续的6,则将这串连续的6替换成27。其他内容不......
  • Opengl入门基础(七)-数学基础
    opengl除了基础的模型构建和贴图,还需要进行位置变换,缩放等功能,之前定义的顶点坐标(0,0.5,0),纹理坐标(0,0.5)实际上是一个向量,而对多个坐标构成的模型进行变换、缩放实际上是向量与......
  • ifc4x3 IfcAdvancedBrep
    1语义定义高级B-rep是一种边界表示模型,其中所有面、边和顶点都被显式表示。它是一个具有显式拓扑和基本或自由几何体的实体。B-rep的面类型为IfcAdvancedFace。高级B-rep......
  • rsa公钥和私钥到底哪个才是用来加密,哪个用来解密?
    本文转自:91博客;原文地址:http://www.9191boke.com/138589019.html 公钥和私钥在一些银行系统、第三方支付系统SDK中经常会遇到,刚接触公钥私钥的朋友们估计很难区分两者......
  • 2. 两数相加
    思路:所给的链表是逆序的,要求返回的也是逆序的,这正是模拟人在做加法运算时候的进位习惯,从低位到高位运算。只需要新建一个列表存储结果链表,设定一个变量做为进位标识。/*......
  • ubuntu 遇坑合集
    使用wget无法下载http3链接的文件解决方案wget--no-check-certificatehttps://packages.microsoft.com/config/ubuntu/18.04/packages-microsoft-prod.deb-Opackages-......
  • Fetch封装
    Fetch使用Get=(url)=>{fetch(url)//默认是GET.then(response=>response.json())//把数据解析成json格式,然后取出.then(result=>{t......
  • CSRF和SSRF
    CSRF又称跨站点请求伪造(有很多前提)比如:写一个index.html其中的script去请求一个地址,如果此时访问了攻击者的服务器则会触发修改的数据包,访问了写的地址,修改了用户的信......
  • 明解数据库------数据库存储演变史
    一、文件系统存储计算机刚开始出现的时候,那时候没有硬盘,只有内存,数据不会进行存储,一般只用于科技计算,计算完输出结果后,程序就撤出内存了。后来随着技术发展,有了硬盘、文......
  • 明解Redis(一)概述
    一、Redis简介Redis全称为RemoteDictionaryServer(远程词典服务),开源(BSD许可)的,使用C语言开发的。Redis是内存中的数据结构存储系统,它可以用作数据库、缓存和消息中间......