在不使用太多条件语句的情况下使用Structs查找员工

package main

import (
    "fmt"
)

type Employee struct {
    ID   int
    Name string
}

type EmployeeDB struct {
    employees map[int]Employee
}

func (db *EmployeeDB) AddEmployee(emp Employee) {
    db.employees[emp.ID] = emp
}

func (db *EmployeeDB) FindEmployeeByID(id int) (Employee, error) {
    emp, ok := db.employees[id]
    if !ok {
        return Employee{}, fmt.Errorf("Employee with ID %d not found", id)
    }
    return emp, nil
}

func main() {
    db := EmployeeDB{
        employees: make(map[int]Employee),
    }

    // 添加员工
    emp1 := Employee{ID: 1, Name: "John"}
    emp2 := Employee{ID: 2, Name: "Jane"}
    db.AddEmployee(emp1)
    db.AddEmployee(emp2)

    // 查找员工
    emp, err := db.FindEmployeeByID(1)
    if err != nil {
        fmt.Println(err)
    } else {
        fmt.Println("Employee found:", emp)
    }
}