零基础入门，打造属于你自己的Web框架

后端

2023-04-14 21:57:50

从零开始打造自己的Web框架

准备踏上构建你自己的Web框架之旅了吗？这篇分步指南将带你领略整个过程，从设置开发环境到部署你的杰作。

1. 准备工作

1.1 安装必要的工具

首先，你需要装备好以下工具：

Go语言： 前往官方网站下载并安装Go语言，确保版本为1.18或更高。
Goland： 下载并安装Goland IDE，确保版本为2022.2.5或更高。

1.2 创建项目

使用Goland创建一个新的Go项目，项目名称可以随意选择，如"my-web-framework"。

2. 构建Web框架

2.1 处理HTTP请求和响应

使用Go语言的net/http包设置一个HTTP服务器来处理请求和响应。

package main

import (
	"fmt"
	"net/http"
)

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Hello, World!")
	})

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

2.2 使用路由

借助httprouter包，可以将请求映射到特定的处理函数。

package main

import (
	"github.com/julienschmidt/httprouter"
	"net/http"
)

func main() {
	router := httprouter.New()

	router.GET("/", func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
		fmt.Fprintf(w, "Hello, World!")
	})

	router.ServeHTTP(http.ListenAndServe(":8080", nil))
}

2.3 使用中间件

negroni包允许你在请求和响应之前或之后执行代码，用于验证或日志记录等目的。

package main

import (
	"github.com/codegangsta/negroni"
	"github.com/julienschmidt/httprouter"
	"net/http"
)

func main() {
	router := httprouter.New()

	n := negroni.New(negroni.HandlerFunc(func(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
		// 验证代码
		next(rw, r)
	}))

	router.GET("/", n.ServeHTTP(func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
		fmt.Fprintf(w, "Hello, World!")
	}))

	router.ServeHTTP(http.ListenAndServe(":8080", nil))
}

2.4 使用模板引擎

html/template包提供模板引擎，用于将数据呈现为HTML页面。

package main

import (
	"github.com/julienschmidt/httprouter"
	"html/template"
	"net/http"
)

func main() {
	router := httprouter.New()

	tmpl, err := template.ParseFiles("templates/index.html")
	if err != nil {
		panic(err)
	}

	router.GET("/", func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
		tmpl.Execute(w, nil)
	})

	router.ServeHTTP(http.ListenAndServe(":8080", nil))
}

2.5 连接数据库

通过database/sql包连接到数据库。

package main

import (
	"database/sql"
	"fmt"

	_ "github.com/go-sql-driver/mysql" // 导入MySQL驱动
)

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

	rows, err := db.Query("SELECT * FROM users")
	if err != nil {
		panic(err)
	}

	for rows.Next() {
		var id int
		var name string
		err = rows.Scan(&id, &name)
		if err != nil {
			panic(err)
		}

		fmt.Println(id, name)
	}
}

2.6 实现持久化

使用gorm包实现数据持久化。

package main

import (
	"database/sql"
	"fmt"

	_ "github.com/go-sql-driver/mysql" // 导入MySQL驱动
	"gorm.io/gorm"
)

type User struct {
	ID   int
	Name string
}

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

	gormDB, err := gorm.Open(db, &gorm.Config{})
	if err != nil {
		panic(err)
	}

	gormDB.AutoMigrate(&User{})

	user := User{Name: "John Doe"}
	gormDB.Create(&user)

	fmt.Println(user.ID, user.Name)
}