Golang项目之函数版学生管理系统

前言

学习第3天，基于函数的方式编写小项目：学生管理系统

代码

package main

import (
	"fmt"
	"os"
)
/*
	函数版学生管理系统
	写一个系统能够查看所有的学生/新增学生/删除学生
*/
var (
	allStudent map[int64]*student // 变量声明
)

type student struct{
	id int64
	name string
}

// newStudent是一个student类型的构造函数
func newStudent(id int64,name string)*student{
	return &student{
		id: id,
		name: name,
	}
}

func showAllStudent(){
	if len(allStudent) != 0{
		// 如果map不为空,把所有学生都打印出来
		for k,v := range allStudent{
			fmt.Printf("学号：%d 姓名：%s\n",k,v.name)
		}
	} else{
		// 否则，提示学生信息为空
		fmt.Println("msg：学生信息为空")
	}


}

func addStudent(){
	// 想allStudent中添加一个新的学生
	// 1.创建一个新学生
	// 1.1 获取用户输入
	var (
		id int64
		name string
	)
	fmt.Print("请输入学生的学号:")
	fmt.Scanln(&id)
	fmt.Print("请输入学生的姓名:")
	fmt.Scanln(&name)

	// 1.2 判断学生是否存在
	if _,ok := allStudent[id];! ok{
		// 学生不存在，则添加
		// 1.3造学生（调用student的构造函数）
		newStu := newStudent(id,name)
		// 2.追加到allStudent这个map中
		allStudent[id] = newStu
	}else{
		// 学生存在，则提示
		fmt.Println("msg:添加的学生已存在")
	}

}

func deleteStudent(){
	// 1.请用户输入要删除的学生的序号
	var (
		deleteId int64
	)
	fmt.Print("请输入学生的学号:")
	fmt.Scanln(&deleteId)
	// 2.判断学生是否存在
	if _,ok := allStudent[deleteId];!ok{
		fmt.Println("msg：学生不存在")
	}else{
		// 2.去allStudent这个map中根据学号删除对应的键值对
		delete(allStudent,deleteId)
	}
}


func main(){
	allStudent = make(map[int64]*student, 48)  // 初始化（开辟内存空间）


	for {
		// 1.打印菜单
		fmt.Println("欢迎光临学生管理系统！")
		fmt.Println(`
			1.查看所有学生
			2.新增学生
			3.删除学生
			4.退出
		`)
		fmt.Print("请输入编号：")
		// 2.等待用户选择下一步要做什么
		var choice int
		fmt.Scanln(&choice)
		fmt.Printf("你选择了> %d <这个选项！\n",choice)
		// 3.执行对应的函数
		switch choice {
		case 1:
			fmt.Println("------学生信息-----")
			showAllStudent()
			fmt.Println("--------end-------")
		case 2:
			addStudent()
		case 3:
			deleteStudent()
		case 4:
			os.Exit(1)
		default:
			fmt.Println("输入编号不存在")
		}
	}

}