GO网络编程（三）：海量用户通信系统1：登录功能初步

一、准备工作

需求分析
1)用户注册
2)用户登录
3)显示在线用户列表
4)群聊(广播)
5)点对点聊天
6)离线留言
主界面

首先，在项目根目录下初始化mod，然后按照如下结构设计目录：

海量用户通信系统/
├── go.mod
├── client/
│   ├── main.go
│   └── login.go
└── server/

二、main

package mainimport ("fmt"
)// 定义两个变量，一个表示用户id，一个表示用户密码
var userID int
var pwd stringfunc main() {//接收用户的选择var key int//判断是否还继续显示菜单var loop = truefor loop {fmt.Println("------------------欢迎登录多人聊天系统")fmt.Println("\t\t\t 1 登录聊天室")fmt.Println("\t\t\t 2 注册用户")fmt.Println("\t\t\t 3 退出系统")fmt.Println("\t\t\t 请选择（1-3）：")fmt.Scanln(&key)switch key {case 1:fmt.Println("登录聊天室")loop = falsecase 2:fmt.Println("注册用户")loop = falsecase 3:fmt.Println("退出系统")loop = falsedefault:fmt.Println("你的输入有误，请重新输入")}}//根据用户输入显示新的提示信息if key == 1 {//说明用户要登录fmt.Printf("请输入用户的id号：")fmt.Scanf("%d\n", &userID)fmt.Printf("请输入用户的密码：")fmt.Scanf("%s\n", &pwd)//先把登录的函数，写到另外一个文件err := login(userID, pwd)if err != nil {fmt.Println("登录失败！")} else {fmt.Println("登录成功！")}} else if key == 2 {fmt.Println("进行用户注册的逻辑...")}
}

特别注意：scanf中要加上换行符，否则其余scanf会接收换行符，如果用scanln，换行符处理更麻烦，所以推荐用scanf且加上\n的形式。

三、login

package mainimport ("fmt"
)func login(userId int, pwd string) error {//下一个就要开始定协议fmt.Printf("userId=%d pwd=%s\n", userId, pwd)return nil
}

四、测试login

特别注意：login和main是同包，所以login的函数名开头无需大写，main无需导包即可使用，但如果不导包，编译器是不会编译调用函数所在文件的！所以运行main前必须先编译login，也可以在client下用“go run .”（注意有一个点）来编译运行当前目录下所有文件。如果你用的是vscode+code runner，且仍想通过运行按钮来执行main，那么你需要打开settings，打开settings.json，找到"code-runner.executorMap"中的go选项，将原来的值“go run”改为“go run ./client”，运行server目录的情况同理。
运行结果如下：

                    1 登录聊天室1 登录聊天室  2 注册用户    3 退出系统    请选择（1-3） ：
1
登录聊天室
请输入用户的id号：111
请输入用户的密码：111
userId=111 pwd=111
登录成功！