go 实现多人聊天系统——显示客户端登录菜单

一 功能

能够正确的显示客户端的菜单。

二 界面

三 代码

package main

import (
   "fmt"
   "os"
)

// 定义两个变量，一个表示用户 id, 一个表示用户密码
var userId int
var userPwd string

func main() {
   // 接收用户的选择
   var key int
   // 判断是否还继续显示菜单
   // var loop = true
   for true {
      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.Scanf("%d\n", &key)
      switch key {
      case 1:
         fmt.Println("登陆聊天室")
         loop = false
      case 2:
         fmt.Println("注册用户")
         loop = false
      case 3:
         fmt.Println("退出系统")
         os.Exit(0)
      default:
         fmt.Println("你的输入有误，请重新输入")
      }


   }
   //更加用户的输入，显示新的提示信息
   if key == 1 {
      // 说明用户要登陆
      fmt.Println("请输入用户的id")
      fmt.Scanf("%d\n", &userId)
      fmt.Println("请输入用户的密码")
      fmt.Scanf("%s\n", &userPwd)
      // 先把登陆的函数，写到另外一个文件， 比如 login.go
      err := login(userId, userPwd)
      if err != nil {
          fmt.Println("登录失败")
      } else {
          fmt.Println("登录成功")
      }
    } else if key == 2 {
        fmt.Println("进行用户注册的逻辑....")
    }
}

四 测试

----------------欢迎登陆多人聊天系统------------

             1 登陆聊天室

             2 注册用户

             3 退出系统

             请选择(1-3):

1

登陆聊天室

请输入用户的id

100

请输入用户的密码

200