Go Web (1)
一、网络通信与 Socket 编程
(一)Socket 编程基础
Socket 是网络通信的核心,它允许程序之间通过网络进行数据交换。在 Go 中,可以使用标准库 net
来实现 Socket 编程。
(二)Socket 服务端与客户端示例
- 服务端代码 :
go
package main
import (
"net"
"log"
)
func handleConnection(conn net.Conn) {
defer conn.Close()
for {
buf := make([]byte, 1024)
n, err := conn.Read(buf)
if err != nil {
log.Println("Error reading:", err.Error())
return
}
data := buf[:n]
log.Printf("Received: %s", data)
reply := "Server received your message: " + string(data)
conn.Write([]byte(reply))
}
}
func main() {
listener, err := net.Listen("tcp", ":8080")
if err != nil {
log.Fatal("Error starting server:", err)
}
defer listener.Close()
log.Println("Server listening on :8080")
for {
conn, err := listener.Accept()
if err != nil {
log.Println("Error accepting connection:", err.Error())
continue
}
go handleConnection(conn)
}
}
- 客户端代码 :
go
package main
import (
"net"
"log"
)
func main() {
conn, err := net.Dial("tcp", "localhost:8080")
if err != nil {
log.Fatal("Error connecting:", err)
}
defer conn.Close()
log.Println("Connected to server")
message := "Hello from client!"
_, err = conn.Write([]byte(message))
if err != nil {
log.Fatal("Error sending message:", err)
}
buffer := make([]byte, 1024)
n, err := conn.Read(buffer)
if err != nil {
log.Fatal("Error reading response:", err)
}
log.Printf("Server response: %s", buffer[:n])
}
(三)点对点通信
点对点通信是两个网络节点之间的直接通信。通过 Socket 编程,可以实现设备之间的点对点消息传递,常用于即时通讯等场景。
二、MySQL 数据库操作
(一)配置数据库
- 安装 MySQL :在本地或服务器上安装 MySQL 数据库,并创建数据库和表。
- Go 中连接 MySQL :使用
database/sql
包和mysql
驱动来连接 MySQL。
go
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql"
)
func main() {
db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/dbname")
if err != nil {
panic(err.Error())
}
defer db.Close()
err = db.Ping()
if err != nil {
panic(err.Error())
}
fmt.Println("Connected to database successfully!")
}
(二)数据库增删改查操作
- 插入数据 :
go
func insertData(db *sql.DB) {
sqlStatement := `INSERT INTO users (name, email) VALUES (?, ?)`
_, err := db.Exec(sqlStatement, "John Doe", "[email protected]")
if err != nil {
panic(err.Error())
}
}
- 查询数据 :
go
func queryData(db *sql.DB) {
rows, err := db.Query("SELECT id, name, email FROM users")
if err != nil {
panic(err.Error())
}
defer rows.Close()
for rows.Next() {
var id int
var name, email string
err := rows.Scan(&id, &name, &email)
if err != nil {
panic(err.Error())
}
fmt.Printf("ID: %d, Name: %s, Email: %s\n", id, name, email)
}
}
- 更新数据 :
go
func updateData(db *sql.DB) {
sqlStatement := `UPDATE users SET email = ? WHERE id = ?`
_, err := db.Exec(sqlStatement, "[email protected]", 1)
if err != nil {
panic(err.Error())
}
}
- 删除数据 :
go
func deleteData(db *sql.DB) {
sqlStatement := `DELETE FROM users WHERE id = ?`
_, err := db.Exec(sqlStatement, 1)
if err != nil {
panic(err.Error())
}
}
三、Go Web 开发实例
(一)单控制器实例
go
package main
import (
"net/http"
"fmt"
)
func handler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Hello, World!")
}
func main() {
http.HandleFunc("/", handler)
http.ListenAndServe(":8080", nil)
}
(二)多控制器实例
go
package main
import (
"net/http"
"fmt"
)
func homeHandler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Home Page")
}
func aboutHandler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "About Page")
}
func main() {
http.HandleFunc("/", homeHandler)
http.HandleFunc("/about", aboutHandler)
http.ListenAndServe(":8080", nil)
}
四、获取请求头和请求参数
(一)获取请求头
go
package main
import (
"net/http"
"fmt"
)
func headerHandler(w http.ResponseWriter, r *http.Request) {
headers := r.Header
fmt.Fprintf(w, "Request Headers:\n")
for key, value := range headers {
fmt.Fprintf(w, "%s: %s\n", key, value)
}
}
func main() {
http.HandleFunc("/headers", headerHandler)
http.ListenAndServe(":8080", nil)
}
(二)获取请求参数
go
package main
import (
"net/http"
"fmt"
)
func paramsHandler(w http.ResponseWriter, r *http.Request) {
name := r.URL.Query().Get("name")
age := r.URL.Query().Get("age")
fmt.Fprintf(w, "Name: %s, Age: %s", name, age)
}
func main() {
http.HandleFunc("/params", paramsHandler)
http.ListenAndServe(":8080", nil)
}
五、Cookie 操作
(一)设置 Cookie
go
package main
import (
"net/http"
"fmt"
)
func setCookieHandler(w http.ResponseWriter, r *http.Request) {
cookie := http.Cookie{Name: "username", Value: "john_doe", Path: "/"}
http.SetCookie(w, &cookie)
fmt.Fprintf(w, "Cookie set")
}
func main() {
http.HandleFunc("/setcookie", setCookieHandler)
http.ListenAndServe(":8080", nil)
}
(二)获取 Cookie
go
package main
import (
"net/http"
"fmt"
)
func getCookieHandler(w http.ResponseWriter, r *http.Request) {
cookie, err := r.Cookie("username")
if err != nil {
fmt.Fprintf(w, "Cookie not found")
return
}
fmt.Fprintf(w, "Cookie value: %s", cookie.Value)
}
func main() {
http.HandleFunc("/getcookie", getCookieHandler)
http.ListenAndServe(":8080", nil)
}
六、Restful
使用 Go 框架构建 Restful API
以 Go 的 net/http
包为例,构建一个简单的用户管理 Restful API。
go
package main
import (
"net/http"
"fmt"
"encoding/json"
"strconv"
)
type User struct {
ID int `json:"id"`
Name string `json:"name"`
Email string `json:"email"`
}
var users = []User{
{ID: 1, Name: "John Doe", Email: "[email protected]"},
{ID: 2, Name: "Jane Smith", Email: "[email protected]"},
}
func getUsers(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(users)
}
func getUser(w http.ResponseWriter, r *http.Request) {
idStr := r.URL.Query().Get("id")
id, err := strconv.Atoi(idStr)
if err != nil {
http.Error(w, "Invalid ID", http.StatusBadRequest)
return
}
for _, user := range users {
if user.ID == id {
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(user)
return
}
}
http.Error(w, "User not found", http.StatusNotFound)
}
func createUser(w http.ResponseWriter, r *http.Request) {
var newUser User
err := json.NewDecoder(r.Body).Decode(&newUser)
if err != nil {
http.Error(w, "Invalid request body", http.StatusBadRequest)
return
}
users = append(users, newUser)
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(newUser)
}
func updateUser(w http.ResponseWriter, r *http.Request) {
idStr := r.URL.Query().Get("id")
id, err := strconv.Atoi(idStr)
if err != nil {
http.Error(w, "Invalid ID", http.StatusBadRequest)
return
}
for i, user := range users {
if user.ID == id {
var updatedUser User
err := json.NewDecoder(r.Body).Decode(&updatedUser)
if err != nil {
http.Error(w, "Invalid request body", http.StatusBadRequest)
return
}
users[i] = updatedUser
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(updatedUser)
return
}
}
http.Error(w, "User not found", http.StatusNotFound)
}
func deleteUser(w http.ResponseWriter, r *http.Request) {
idStr := r.URL.Query().Get("id")
id, err := strconv.Atoi(idStr)
if err != nil {
http.Error(w, "Invalid ID", http.StatusBadRequest)
return
}
for i, user := range users {
if user.ID == id {
users = append(users[:i], users[i+1:]...)
fmt.Fprintf(w, "User deleted successfully")
return
}
}
http.Error(w, "User not found", http.StatusNotFound)
}
func main() {
http.HandleFunc("/users", getUsers)
http.HandleFunc("/user", getUser)
http.HandleFunc("/createuser", createUser)
http.HandleFunc("/updateuser", updateUser)
http.HandleFunc("/deleteuser", deleteUser)
http.ListenAndServe(":8080", nil)
}