最近再看Go语言web编程，go语言搭建Web服务器，既可以用go原生的net/http包，也可以用gin/fasthttp/fiber等这些Web框架。本博客使用net/http模块编写了一个简单的登录验证和文件上传的功能，在此做个简单记录。
目录
代码如下：
package main
import (
"fmt"
"html/template"
"io"
"log"
"net/http"
"os"
)
/*
go运行方式：
（1）解释运行
go run main.go
（2）编译运行
--使用默认名
go build main.go
./main
--指定可执行程序名
go build -o test main.go
./test
*/
// http://127.0.0.1:8181/login
func login(w http.ResponseWriter, r *http.Request) {
fmt.Println("method", r.Method)
if r.Method == "GET" {
t, _ := template.ParseFiles("login.html")
t.Execute(w, nil)
/*
//字符串拼装表单
html := `<html>
<head>
<title>上传文件</title>
</head>
<body>
<form enctype="multipart/form-data" action="http://localhost:8181/upload" method="post">
<input type="file" name="uploadfile" />
<input type="hidden" name="token" value="{
{.}}" />
<input type="submit" value="upload" />
</form>
</body>
</html>`
crutime := time.Now().Unix()
h := md5.New()
io.WriteString(h, strconv.FormatInt(crutime, 10))
token := fmt.Sprintf("%x", h.Sum(nil))
t := template.Must(template.New("test").Parse(html))
t.Execute(w, token)
*/
} else {
r.ParseForm()
fmt.Println("username", r.Form["username"])
fmt.Println("password", r.Form["password"])
fmt.Fprintf(w, "登录成功")
}
}
// http://127.0.0.1:8181/upload
func upload(writer http.ResponseWriter, r *http.Request) {
//表示maxMemory,调用ParseMultipart后，上传的文件存储在maxMemory大小的内存中，
//如果大小超过maxMemory，剩下部分存储在系统的临时文件中
r.ParseMultipartForm(32 << 10)
//根据input中的name="uploadfile"来获得上传的文件句柄
file, header, err := r.FormFile("uploadfile")
if err != nil {
fmt.Fprintf(writer, "上传出错")
fmt.Println(err)
return
}
defer file.Close()
/*
fmt.Printf("Uploaded File: %+v\n", header.Filename)
fmt.Printf("File Size: %+v\n", header.Size)
// 注意此处的header.Header是textproto.MIMEHeader类型 （ map[string][]string ）
fmt.Printf("MIME Type: %+v\n", header.Header.Get("Content-Type"))
// 将文件保存到服务器指定的目录（* 用来随机数的占位符）
tempFile, err := ioutil.TempFile("uploads", "*"+header.Filename)
*/
fmt.Println("handler.Filename", header.Filename)
f, err := os.OpenFile("./filedir/"+header.Filename, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
fmt.Println(err)
fmt.Fprintf(writer, "上传出错")
return
}
defer f.Close()
io.Copy(f, file)
fmt.Fprintf(writer, "上传成功")
}
func common_handle() {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello world !"))
})
http.HandleFunc("/login", login)
http.HandleFunc("/upload", upload)
}
func main1() {
common_handle()
//监听8181端口
err := http.ListenAndServe(":8181", nil)
if err != nil {
log.Fatal("err:", err)
}
}
// 声明helloHandler
type helloHandler struct{}
// 定义helloHandler
func (m11111 *helloHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello world, this is my first golang programe !"))
}
func welcome(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Welcome to golang family !"))
}
func main2() {
a := helloHandler{}
//使用http.Handle
http.Handle("/hello", &a)
http.Handle("/welcome", http.HandlerFunc(welcome))
common_handle()
server := http.Server{
Addr: "127.0.0.1:8181",
Handler: nil, // 对应DefaultServeMux路由
}
server.ListenAndServe()
}
func main() {
main2()
}
login.html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8"/>
<title>欢迎进入首页</title>
</head>
<body>
<h3>登录测试</h3>
<hr/>
<form action="http://localhost:8181/login" method="post">
<table border=0 title="测试">
<tr>
<td>用户名：</td>
<td><input type="text" name="username"></td>
</tr>
<tr>
<td>密码：</td>
<td><input type="password" name="password"></td>
</tr>
<tr>
<td colspan=2>
<input type="reset" />
<input type="submit" value="登录" />
</td>
</tr>
</table>
</form>
<br>
<h3>文件上传测试</h3>
<hr/>
<form action="http://localhost:8181/upload" method="post" enctype="multipart/form-data">
<input type="file" name="uploadfile"/>
<input type="submit" value="upload">
</form>
</body>
</html>1.文件目录结构
2.编译运行
3.用户登录
http://127.0.0.1:8181/login
4.文件上传
5.mime/multipart模拟form表单上传文件
使用mime/multipart包，可以将multipart/form-data数据解析为一组文件和表单字段，或者使用multipart.Writer将文件和表单字段写入HTTP请求体中。
以下例子中首先打开要上传的文件，然后创建一个multipart.Writer，用于构造multipart/form-data格式的请求体。我们使用CreateFormFile方法创建一个multipart.Part，用于表示文件字段，将文件内容复制到该Part中。我们还使用WriteField方法添加其他表单字段。然后，我们关闭multipart.Writer，以便写入Content-Type和boundary，并使用NewRequest方法创建一个HTTP请求。我们将Content-Type设置为multipart/form-data，并使用默认的HTTP客户端发送请求。最后，我们读取并处理响应。
test.go
package main
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"mime/multipart"
"net/http"
"os"
"path/filepath"
)
func main() {
// 需要上传的文件路径
filePath := "image_2023_06_29T11_46_39_023Z.png"
// 打开要上传的文件
file, err := os.Open(filePath)
if err != nil {
fmt.Println("Failed to open file:", err)
return
}
defer file.Close()
// 创建multipart.Writer，用于构造multipart/form-data格式的请求体
var requestBody bytes.Buffer
multipartWriter := multipart.NewWriter(&requestBody)
// 创建一个multipart.Part，用于表示文件字段
part, err := multipartWriter.CreateFormFile("uploadfile", filepath.Base(filePath))
if err != nil {
fmt.Println("Failed to create form file:", err)
return
}
// 将文件内容复制到multipart.Part中
_, err = io.Copy(part, file)
if err != nil {
fmt.Println("Failed to copy file content:", err)
return
}
// 添加其他表单字段
multipartWriter.WriteField("title", "My file")
// 关闭multipart.Writer，以便写入Content-Type和boundary
err = multipartWriter.Close()
if err != nil {
fmt.Println("Failed to close multipart writer:", err)
return
}
// 创建HTTP请求
req, err := http.NewRequest("POST", "http://127.0.0.1:8181/upload", &requestBody)
if err != nil {
fmt.Println("Failed to create request:", err)
return
}
// 设置Content-Type为multipart/form-data
req.Header.Set("Content-Type", multipartWriter.FormDataContentType())
// 发送HTTP请求
client := http.DefaultClient
resp, err := client.Do(req)
if err != nil {
fmt.Println("Failed to send request:", err)
return
}
defer resp.Body.Close()
// 处理响应
respBody, err := ioutil.ReadAll(resp.Body)
if err != nil {
fmt.Println("Failed to read response:", err)
return
}
fmt.Println("Response:", string(respBody))
}
编译执行：
go build test.go
./test
运行结果展示：
