首页 >后端开发 >Golang >如何利用Go语言开发点餐系统的预定餐厅功能

如何利用Go语言开发点餐系统的预定餐厅功能

王林
王林原创
2023-11-01 17:18:57596浏览

如何利用Go语言开发点餐系统的预定餐厅功能

如何利用Go语言开发点餐系统的预定餐厅功能,需要具体代码示例

餐厅预定功能是现代点餐系统中非常重要的一部分,它可以让顾客提前预约餐厅的就餐时间,有效避免了等位的情况,提升了顾客的就餐体验。在本文中,我们将使用Go语言来开发一个简单的餐厅预定功能。

首先,我们需要创建一个数据库来存储餐厅的信息和预定订单的数据。我们可以使用MySQL作为数据库管理系统,创建一个名为"restaurant"的数据库,并在其中创建两个表分别为"restaurants"和"reservations"。

restaurants表的结构如下:

CREATE TABLE restaurants (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(255) NOT NULL,
    address VARCHAR(255) NOT NULL,
    capacity INT NOT NULL,
    available_time_start TIME NOT NULL,
    available_time_end TIME NOT NULL
);

reservations表的结构如下:

CREATE TABLE reservations (
    id INT PRIMARY KEY AUTO_INCREMENT,
    restaurant_id INT NOT NULL,
    guest_name VARCHAR(255) NOT NULL,
    reservation_time DATETIME NOT NULL,
    FOREIGN KEY (restaurant_id) REFERENCES restaurants(id)
);

接下来,我们将使用Go语言来开发点餐系统的预定餐厅功能。首先,我们需要创建一个restaurant结构体来表示餐厅的信息,包括餐厅的名称、地址、容量和可用时间段等。

type Restaurant struct {
    ID               int
    Name             string
    Address          string
    Capacity         int
    AvailableTimeStart time.Time
    AvailableTimeEnd   time.Time
}

我们还需要创建一个reservation结构体来表示预定订单的信息,包括预定的餐厅ID、顾客姓名和预定时间等。

type Reservation struct {
    ID               int
    RestaurantID     int
    GuestName        string
    ReservationTime  time.Time
}

接下来,我们需要实现一些数据库操作的函数,包括查询所有餐厅、查询可用餐厅、查询预定订单、添加预定订单等。

首先,我们创建一个db变量来存储数据库连接。

var db *sql.DB

在main函数中,我们需要打开数据库连接。

func main() {
    var err error
    db, err = sql.Open("mysql", "root:password@tcp(localhost:3306)/restaurant")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 运行HTTP服务器
    http.HandleFunc("/restaurants", getRestaurantsHandler)
    http.HandleFunc("/available-restaurants", getAvailableRestaurantsHandler)
    http.HandleFunc("/reservations", getReservationsHandler)
    http.HandleFunc("/reserve", addReservationHandler)

    log.Fatal(http.ListenAndServe(":8080", nil))
}

下面,我们来实现一些数据库操作的函数。

首先,查询所有餐厅的函数如下:

func getRestaurantsHandler(w http.ResponseWriter, r *http.Request) {
    rows, err := db.Query("SELECT * FROM restaurants")
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    var restaurants []Restaurant
    for rows.Next() {
        var restaurant Restaurant
        err := rows.Scan(&restaurant.ID, &restaurant.Name, &restaurant.Address, &restaurant.Capacity, &restaurant.AvailableTimeStart, &restaurant.AvailableTimeEnd)
        if err != nil {
            log.Fatal(err)
        }
        restaurants = append(restaurants, restaurant)
    }

    json.NewEncoder(w).Encode(restaurants)
}

接下来,查询可用餐厅的函数如下:

func getAvailableRestaurantsHandler(w http.ResponseWriter, r *http.Request) {
    currentTime := time.Now()
    rows, err := db.Query("SELECT * FROM restaurants WHERE available_time_start <= ? AND available_time_end >= ?", currentTime, currentTime)
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    var restaurants []Restaurant
    for rows.Next() {
        var restaurant Restaurant
        err := rows.Scan(&restaurant.ID, &restaurant.Name, &restaurant.Address, &restaurant.Capacity, &restaurant.AvailableTimeStart, &restaurant.AvailableTimeEnd)
        if err != nil {
            log.Fatal(err)
        }
        restaurants = append(restaurants, restaurant)
    }

    json.NewEncoder(w).Encode(restaurants)
}

然后,查询预定订单的函数如下:

func getReservationsHandler(w http.ResponseWriter, r *http.Request) {
    rows, err := db.Query("SELECT * FROM reservations")
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()

    var reservations []Reservation
    for rows.Next() {
        var reservation Reservation
        err := rows.Scan(&reservation.ID, &reservation.RestaurantID, &reservation.GuestName, &reservation.ReservationTime)
        if err != nil {
            log.Fatal(err)
        }
        reservations = append(reservations, reservation)
    }

    json.NewEncoder(w).Encode(reservations)
}

最后,添加预定订单的函数如下:

func addReservationHandler(w http.ResponseWriter, r *http.Request) {
    var reservation Reservation
    err := json.NewDecoder(r.Body).Decode(&reservation)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }

    stmt, err := db.Prepare("INSERT INTO reservations (restaurant_id, guest_name, reservation_time) VALUES (?, ?, ?)")
    if err != nil {
        log.Fatal(err)
    }
    defer stmt.Close()

    _, err = stmt.Exec(reservation.RestaurantID, reservation.GuestName, reservation.ReservationTime)
    if err != nil {
        log.Fatal(err)
    }

    w.WriteHeader(http.StatusOK)
}

通过以上步骤,我们已经完成了使用Go语言开发点餐系统的预定餐厅功能。我们可以使用Postman等工具来测试这些HTTP接口,也可以为前端开发提供相应的数据接口。当然,这只是一个简单的示例,在实际开发中还需要根据具体需求进行相应的改进和完善。

以上是如何利用Go语言开发点餐系统的预定餐厅功能的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn