GO语言学习——方法版学生管理系统

方法版学生管理系统

main.go文件

package main

import (
	"fmt"
	"os"
)

// 学生管理系统

var smr studentMgr // 声明一个全局的变量学生管理对象:smr

// 菜单函数
func showMenu() {
	fmt.Println("------------welcome sms!------------")
	fmt.Println(`
	1. 查看所有学生
	2. 添加学生
	3. 修改学生
	4. 删除学生
	5. 退出`)
}

func main() {
	smr = studentMgr{ // 修改的是全局的那个变量
		allStudent: make(map[int64]student, 100),
	}
	for {
		showMenu()
		// 等待用户输入选项
		fmt.Print("请输入序号:")
		var choice int
		fmt.Scanln(&choice)
		fmt.Println("您输入的是:", choice)
		switch choice {
		case 1:
			smr.showStudent()
		case 2:
			smr.addStudent()
		case 3:
			smr.editStudent()
		case 4:
			smr.deleteStudent()
		case 5:
			os.Exit(1)
		default:
			fmt.Println("请输入正确的序号!")
		}
	}
}

student_mgr.go文件

package main

import "fmt"

// 有一个物件
// 1. 它保存了一些数据  ---> 结构体的字段
// 2. 它有4个功能  ---> 结构体的方法

type student struct {
	id   int64
	name string
}

type studentMgr struct {
	allStudent map[int64]student
}

// 查看学生
func (s studentMgr) showStudent() {
	// 从s.allStudent这个map中把所有的学生逐个取出来
	for _, stu := range s.allStudent { // stu是具体每一个学生
		fmt.Printf("学号:%d 姓名:%s\n", stu.id, stu.name)
	}
	fmt.Println("查看学生完成!")
}

// 增加学生
func (s studentMgr) addStudent() {
	// 1. 根据用户输入的内容创建一个新的学生
	var (
		stuID   int64
		stuName string
	)
	// 获取用户输入
	fmt.Print("请输入您要添加的学号:")
	fmt.Scanln(&stuID)
	fmt.Print("请输入您要添加的姓名:")
	fmt.Scanln(&stuName)
	// 根据用户输入创建结构体对象
	newStu := student{
		id:   stuID,
		name: stuName,
	}
	// 2. 把新的学生放到s.allStudent这个map中
	s.allStudent[newStu.id] = newStu
	fmt.Println("添加学生完成!")
}

// 修改学生
func (s studentMgr) editStudent() {
	// 1. 获取用户输入的学号
	var stuID int64
	fmt.Print("请输入您要修改的学号:")
	fmt.Scanln(&stuID)
	// 2. 展示该学号对应的学生信息，如果没有提示查无此人
	stuObj, ok := s.allStudent[stuID]
	if !ok {
		fmt.Println("查无此人!")
		return
	}
	fmt.Printf("您要修改的学生信息如下:学号:%d 姓名:%s\n", stuObj.id, stuObj.name)
	// 3. 请输入修改后的学生姓名
	fmt.Print("请输入学生的新名字:")
	var newName string
	fmt.Scanln(&newName)
	// 4. 更新学生的姓名
	stuObj.name = newName
	s.allStudent[stuID] = stuObj // 更新map中的学生
	fmt.Println("修改学生完成!")
}

// 删除学生
func (s studentMgr) deleteStudent() {
	// 1. 请用户输入要删除的学生id
	var stuID int64
	fmt.Print("请输入要删除的学生的学号:")
	fmt.Scanln(&stuID)
	// 2. 去map中找有没有这个学生，如果没有打印“查无此人”提示
	_, ok := s.allStudent[stuID]
	if !ok {
		fmt.Println("查无此人!")
		return
	}
	// 3. 有的话就删除 如何从map中删除键值对
	delete(s.allStudent, stuID)
	fmt.Println("删除学生完成!")
}