Go 标准库之 GoRequests 介绍与基本使用 - ToB企服应用市场:ToB评测及商务社交产业平台

go get -u github.com/levigross/grequests

复制代码

import "github.com/levigross/grequests"

复制代码

func Get() {
resp, err := grequests.Get("http://127.0.0.1:8080/book/", nil)
if err != nil {
log.Fatalln("Unable to make request: ", err)
}
if !resp.Ok {
log.Fatalln("请求超时!")
}
// 解析响应的JSON数据
var data []map[string]interface{}
if err := resp.JSON(&data); err != nil {
log.Fatalln("Unable to parse JSON response: ", err)
}
fmt.Println(data)
}

复制代码

func Post() {
postData := map[string]string{
"id": "1",
"name": "Go入门到进阶",
}
geq := &grequests.RequestOptions{
JSON: postData,
}
resp, err := grequests.Post("http://127.0.0.1:8080/book/create", geq)
if err != nil {
log.Fatalln("Unable to make request: ", err)
}
fmt.Println(resp.String())
}

复制代码

func UploadFile() {
// 允许您通过指定磁盘上的位置来创建FileUpload结构片
// 打开要上传的文件
file, err := os.Open("./go.mod")
if err != nil {
log.Fatalln("Unable to open file: ", err)
}
defer file.Close()
// 创建FileUpload结构片
ro := &grequests.RequestOptions{
Files: []grequests.FileUpload{{
FileName: "1.txt", // 上传后的文件名称
FieldName: "file", // 上传文件对应字段
FileContents: file, // 使用文件内容作为FileContents
}},
}
// 发送POST请求
resp, err := grequests.Post("http://127.0.0.1:8080/book/upload/", ro)
if err != nil {
log.Fatalln("Unable to make request: ", err)
}
fmt.Println(resp.String())
}

复制代码

func Proxy() {
// 代理服务器
const proxyServer = "http-pro.xxx.com:9010"
// 代理隧道验证信息
const proxyUser = "xxxxxxxxx"
const proxyPass = "xxxxxxxxx"
// 初始化代理URL
proxyUrl, _ := url.Parse("http://" + proxyUser + ":" + proxyPass + "@" + proxyServer)
// 创建请求选项
ro := &grequests.RequestOptions{
Proxies: map[string]*url.URL{
"http": proxyUrl,
},
Headers: map[string]string{
"user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36",
},
}
// 发起GET请求
resp, err := grequests.Get("http://www.example.com", ro)
if err != nil {
fmt.Println("Error:", err)
return
}
// 打印响应状态码
fmt.Println("Status code:", resp.StatusCode)
// 打印响应体
fmt.Println("Response:", resp.String())
}

复制代码

session := grequests.Session{
RequestOptions: &grequests.RequestOptions{
Headers: map[string]string{
"authority": "mp3.haoge500.com",
"referer": "https://www.zz123.com/",
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36",
},
},
}

复制代码

package main
import (
"encoding/json"
"github.com/gin-gonic/gin"
"net/http"
"os"
)
type Book struct {
ID string `json:"id"`
Name string `json:"name"`
}
type BookHandler struct {
}
func (b *BookHandler) RegisterRoutes(server *gin.Engine) {
bg := server.Group("/book")
bg.POST("/upload", b.Upload)
bg.POST("/create", b.Create)
bg.GET("/", b.GetAllBooks) // 查询书籍
}
func (b *BookHandler) Upload(ctx *gin.Context) {
// 从请求中获取文件
file, err := ctx.FormFile("file")
if err != nil {
ctx.JSON(http.StatusBadRequest, gin.H{"error": "无法获取上传的文件"})
return
}
// 将文件保存到服务器
// 注意：这里需要确保保存文件的目录存在，并且服务器有写入权限
savePath := "./uploads/" + file.Filename
if err := ctx.SaveUploadedFile(file, savePath); err != nil {
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "文件保存失败"})
return
}
ctx.JSON(http.StatusOK, gin.H{"message": "文件上传成功"})
}
func (b *BookHandler) Create(ctx *gin.Context) {
var req Book
if err := ctx.Bind(&req); err != nil {
return
}
// 将新的书籍数据保存到data.json文件中
if err := addBookToFile(&req); err != nil {
ctx.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to save book data"})
return
}
ctx.JSON(http.StatusOK, gin.H{"message": "Book added successfully"})
}
func (b *BookHandler) GetAllBooks(c *gin.Context) {
// 从data.json文件中读取书籍数据
books, err := getBooksFromFile()
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read book data"})
return
}
// 获取URL查询参数中的id
id := c.Query("id")
// 如果提供了ID，查找具有匹配ID的书籍列表
if id != "" {
// 查找具有匹配ID的书籍
var foundBooks []Book
for _, book := range books {
if book.ID == id {
foundBooks = append(foundBooks, book)
}
}
// 如果找到了匹配的书籍，返回这些书籍
if len(foundBooks) > 0 {
c.JSON(http.StatusOK, foundBooks)
return
}
// 如果没有找到匹配的书籍，返回404
c.JSON(http.StatusNotFound, gin.H{"error": "Books not found"})
return
}
// 如果没有提供ID，返回所有书籍
c.JSON(http.StatusOK, books)
}
func addBookToFile(book *Book) error {
// 读取现有的data.json文件内容
var books []Book
data, err := os.ReadFile("data.json")
if err != nil && !os.IsNotExist(err) {
return err
}
// 如果文件存在，解析现有的书籍数据
if err == nil {
if err := json.Unmarshal(data, &books); err != nil {
return err
}
}
// 将新的书籍添加到数组中
books = append(books, *book)
// 将更新后的书籍数组序列化为JSON
newData, err := json.MarshalIndent(books, "", " ")
if err != nil {
return err
}
// 将序列化的JSON数据写入data.json文件
if err := os.WriteFile("data.json", newData, 0644); err != nil {
return err
}
return nil
}
func getBooksFromFile() ([]Book, error) {
// 读取data.json文件内容
data, err := os.ReadFile("data.json")
if err != nil {
return nil, err
}
// 解析JSON数据到书籍数组中
var books []Book
if err := json.Unmarshal(data, &books); err != nil {
return nil, err
}
return books, nil
}
func InitWebServer(bookHandler *BookHandler) *gin.Engine {
server := gin.Default()
bookHandler.RegisterRoutes(server)
return server
}
func main() {
// 确保上传目录存在
os.MkdirAll("./uploads", 0755)
bookHandler := &BookHandler{}
server := InitWebServer(bookHandler)
server.Run(":8080") // 在8080端口启动服务器
}

复制代码