go语言gin框架中间件详解

package main
 
import (
	"fmt"
	"GitHub.com/gin-gonic/gin"
	"time"
)
 
var (
	limiter = NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次
)
 
// NewLimiter 创建限流器
func NewLimiter(limit int, duration time.Duration) *Limiter {
	return &Limiter{
		limit:     limit,
		duration:  duration,
		timestamps: make(map[string][]int64),
	}
}
 
// Limiter 限流器
type Limiter struct {
	limit     int               // 限制的请求数量
	duration  time.Duration     // 时间窗口
	timestamps map[string][]int64 // 请求的时间戳
}
 
// Middleware 限流中间件
func (l *Limiter) Middleware(c *gin.Context) {
	ip := c.ClientIP() // 获取客户端IP地址
 
	// 检查请求时间戳切片是否存在
	if _, ok := l.timestamps[ip]; !ok {
		l.timestamps[ip] = make([]int64, 0)
	}
 
	now := time.Now().Unix() // 当前时间戳
 
	// 移除过期的请求时间戳
	for i := 0; i < len(l.timestamps[ip]); i++ {
		if l.timestamps[ip][i] < now-int64(l.duration.Seconds()) {
			l.timestamps[ip] = append(l.timestamps[ip][:i], l.timestamps[ip][i+1:]...)
			i--
		}
	}
 
	// 检查请求数量是否超过限制
	if len(l.timestamps[ip]) >= l.limit {
		c.JSON(429, gin.H{
			"message": "Too Many Requests",
		})
		c.Abort()
		return
	}
 
	// 添加当前请求时间戳到切片
	l.timestamps[ip] = append(l.timestamps[ip], now)
 
	// 继续处理请求
	c.Next()
}
 
func main() {
	r := gin.Default()
 
	// 使用限流中间件
	r.Use(limiter.Middleware)
 
	r.GET("/", func(c *gin.Context) {
		c.jsON(200, gin.H{
			"message": "Hello World",
		})
	})
 
	r.Run(":8080")
}

package main
 
import (
	"github.com/gin-gonic/gin"
)
 
func main() {
	// 创建Gin引擎
	r := gin.Default()
 
	// 使用CORS中间件
	r.Use(corsMiddleware())
 
	// 定义路由处理函数
	r.GET("/hello", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "Hello, world!",
		})
	})
 
	// 启动Web服务器
	r.Run(":8080")
}
 
// corsMiddleware 返回CORS中间件处理函数
func corsMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 允许所有的跨域请求
		c.Header("Access-Control-Allow-Origin", "*")
		c.Header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
		c.Header("Access-Control-Allow-Headers", "Content-Type, Authorization")
		c.Header("Access-Control-Max-Age", "86400") // 预检请求缓存时间，单位为秒
 
		// 处理预检请求
		if c.Request.Method == "OPTIONS" {
			c.AbortWithStatus(200)
			return
		}
 
		// 继续处理其他请求
		c.Next()
	}
}

import (
	"github.com/gin-gonic/gin"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/mysql" // 导入数据库驱动
)
 
func main() {
	// 连接数据库
	db, err := gorm.Open("mysql", "user:passWord@/dbname?charset=utf8&parseTime=True&loc=Local")
	if err != nil {
		panic("连接数据库失败: " + err.Error())
	}
	defer db.Close()
 
	// 注册Gin路由
	router := gin.Default()
	// 将db作为中间件传递给路由处理函数
	router.Use(func(c *gin.Context) {
		c.Set("db", db)
		c.Next()
	})
	// 在路由处理函数中可以通过c.MustGet("db").(*gorm.DB)获取到db对象，然后进行数据库操作
 
	// ... 定义其他路由和处理函数
 
	router.Run(":8080")
}

import (
	"github.com/gin-gonic/gin"
	"github.com/go-xorm/xorm"
	_ "github.com/go-sql-driver/mysql" // 导入数据库驱动
)
 
func main() {
	// 连接数据库
	engine, err := xorm.NewEngine("mysql", "user:password@/dbname?charset=utf8")
	if err != nil {
		panic("连接数据库失败: " + err.Error())
	}
	defer engine.Close()
 
	// 注册Gin路由
	router := gin.Default()
	// 将engine作为中间件传递给路由处理函数
	router.Use(func(c *gin.Context) {
		c.Set("engine", engine)
		c.Next()
	})
	// 在路由处理函数中可以通过c.MustGet("engine").(*xorm.Engine)获取到engine对象，然后进行数据库操作
 
	// ... 定义其他路由和处理函数
 
	router.Run(":8080")
}

package main
 
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/go-redis/redis/v8"
)
 
func main() {
	// 创建Gin引擎
	router := gin.Default()
 
	// 创建Redis客户端
	redisClient := redis.NewClient(&redis.Options{
		Addr:     "localhost:6379", // Redis服务器地址
		Password: "",               // Redis密码
		DB:       0,                // Redis数据库编号
	})
 
	// 使用Redis中间件
	router.Use(func(c *gin.Context) {
		// 在Gin的上下文中设置Redis客户端
		c.Set("redis", redisClient)
 
		// 继续处理后续的请求
		c.Next()
	})
 
	// 定义路由和处理函数
	router.GET("/get/:key", func(c *gin.Context) {
		// 从上下文中获取Redis客户端
		redisClient := c.MustGet("redis").(*redis.Client)
 
		// 从URL参数中获取键名
		key := c.Param("key")
 
		// 使用Redis客户端进行GET操作
		val, err := redisClient.Get(c, key).Result()
		if err == redis.Nil {
			c.JSON(200, gin.H{
				"result": fmt.Sprintf("Key '%s' not found", key),
			})
		} else if err != nil {
			c.JSON(500, gin.H{
				"error": err.Error(),
			})
		} else {
			c.JSON(200, gin.H{
				"result": val,
			})
		}
	})
 
	// 启动Web服务器
	router.Run(":8080")
}

package main

import (
    "fmt"
    "net/Http"

    "github.com/gin-gonic/gin"
    "github.com/olivere/elastic/v7"
)

func main() {
    // 创建Gin引擎
    r := gin.Default()

    // 添加ES中间件
    r.Use(elasticsearchMiddleware())

    // 定义路由
    r.GET("/", func(c *gin.Context) {
        // 从上下文中获取ES客户端
        esClient := c.MustGet("esClient").(*elastic.Client)

        // 使用ES客户端进行查询
        // 这里只是一个示例，具体的ES查询操作可以根据实际需求进行修改
        _, err := esClient.Ping().Do(c.Request.Context())
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to ping Elasticsearch"})
            return
        }

        c.JSON(http.StatusOK, gin.H{"message": "Hello from Gin with Elasticsearch middleware!"})
    })

    // 启动Gin应用
    r.Run(":8080")
}

// ElasticSearchMiddleware 是用于处理ES请求的中间件
func ElasticSearchMiddleware() gin.HandlerFunc {
    // 创建ES客户端
    client, err := elastic.NewClient(elastic.SetURL("http://localhost:9200"))
    if err != nil {
        panic(err)
    }

    // 返回Gin中间件处理函数
    return func(c *gin.Context) {
        // 将ES客户端添加到上下文中
        c.Set("esClient", client)

        // 继续处理下一个中间件或路由处理函数
        c.Next()
    }
}

package main
 
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/streadway/amqp"
)
 
// RabbitMQ中间件
func RabbitMQMiddleware(conn *amqp.Connection, queueName string) gin.HandlerFunc {
	return func(c *gin.Context) {
		// 创建RabbitMQ通道
		ch, err := conn.Channel()
		if err != nil {
			c.AbortWithError(500, err)
			return
		}
		defer ch.Close()
 
		// 操作RabbitMQ队列
		_, err = ch.QueueDeclare(queueName, false, false, false, false, nil)
		if err != nil {
			c.AbortWithError(500, err)
			return
		}
 
		// 将RabbitMQ连接和通道作为上下文信息传递给下一个处理器
		c.Set("rabbitmq_conn", conn)
		c.Set("rabbitmq_ch", ch)
 
		// 继续处理下一个中间件或请求处理函数
		c.Next()
	}
}
 
func main() {
	// 创建RabbitMQ连接
	conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
	if err != nil {
		fmt.Println("连接RabbitMQ失败:", err)
		return
	}
	defer conn.Close()
 
	// 创建Gin引擎
	r := gin.Default()
 
	// 添加RabbitMQ中间件
	r.Use(RabbitMQMiddleware(conn, "my_queue"))
 
	// 定义路由和处理函数
	r.GET("/", func(c *gin.Context) {
		// 从上下文中获取RabbitMQ连接和通道
		conn := c.MustGet("rabbitmq_conn").(*amqp.Connection)
		ch := c.MustGet("rabbitmq_ch").(*amqp.Channel)
 
		// 在处理函数中使用RabbitMQ连接和通道进行操作
		// ...
 
		c.JSON(200, gin.H{
			"message": "Hello RabbitMQ!",
		})
	})
 
	// 启动Gin服务
	r.Run(":8080")
}

package main
 
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/nats-io/nats.go"
	"log"
	"time"
)
 
func main() {
	// 连接到NATS服务器
	nc, err := nats.Connect(nats.DefaultURL)
	if err != nil {
		log.Fatal(err)
	}
	defer nc.Close()
 
	// 创建Gin实例
	router := gin.Default()
 
	// 注册NATS中间件
	router.Use(NATSMiddleware(nc))
 
	// 定义路由处理函数
	router.GET("/ping", func(c *gin.Context) {
		// 从NATS发布一条消息
		err := nc.Publish("ping", []byte("Hello from NATS!"))
		if err != nil {
			log.Println(err)
			c.JSON(500, gin.H{"error": "Failed to publish message to NATS"})
			return
		}
		c.JSON(200, gin.H{"message": "pong"})
	})
 
	// 启动HTTP服务器
	router.Run(":8080")
}
 
// NATSMiddleware 是一个自定义的Gin中间件，用于处理NATS连接
func NATSMiddleware(nc *nats.Conn) gin.HandlerFunc {
	return func(c *gin.Context) {
		// 将NATS连接对象绑定到Gin的上下文中
		c.Set("nats", nc)
		c.Next()
	}
}

func PaginationMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 在这里实现分页逻辑
	}
}

func PaginationMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 从查询参数中获取页码和每页数量，默认值为1和10
		page := c.DefaultQuery("page", "1")
		pageSize := c.DefaultQuery("pageSize", "10")
 
		// 将页码和每页数量转换为整数
		pageInt, err := strconv.Atoi(page)
		if err != nil {
			pageInt = 1
		}
		pageSizeInt, err := strconv.Atoi(pageSize)
		if err != nil {
			pageSizeInt = 10
		}
 
		// 计算起始偏移量
		offset := (pageInt - 1) * pageSizeInt
 
		// 设置分页参数到Context中，供后续处理函数使用
		c.Set("page", pageInt)
		c.Set("pageSize", pageSizeInt)
		c.Set("offset", offset)
 
		// 调用后续处理函数
		c.Next()
	}
}

// 使用PaginationMiddleware作为中间件
router.GET("/users", PaginationMiddleware(), func(c *gin.Context) {
	// 从Context中获取分页参数
	page := c.MustGet("page").(int)
	pageSize := c.MustGet("pageSize").(int)
	offset := c.MustGet("offset").(int)
 
	// 根据分页参数进行查询
	// 例如，从数据库查询用户列表，并应用分页参数
 
	// 返回分页结果
	c.JSON(http.StatusOK, gin.H{
		"page":     page,
		"pageSize": pageSize,
		"offset":   offset,
		// 其他返回的数据
	})
})

package main
 
import (
	"fmt"
	"io"
	"mime"
	"net/http"
	"os"
	"path/filepath"
 
	"github.com/gin-gonic/gin"
)
 
func main() {
	r := gin.Default()
 
	// 处理上传文件的API接口
	r.POST("/upload", func(c *gin.Context) {
		// 获取上传的文件
		file, err := c.FormFile("file")
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}
 
		// 判断文件类型
		ext := filepath.Ext(file.Filename)
		mimeType := mime.TypeByExtension(ext)
		if mimeType == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "无效的文件类型"})
			return
		}
 
		// 判断文件大小
		if file.Size > 10*1024*1024 { // 限制文件大小为10MB
			c.JSON(http.StatusBadRequest, gin.H{"error": "文件大小超过限制"})
			return
		}
 
		// 将上传的文件保存到本地
		dst := fmt.Sprintf("./uploads/%s", file.Filename)
		err = c.SaveUploadedFile(file, dst)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
 
		c.JSON(http.StatusOK, gin.H{"message": "文件上传成功"})
	})
 
	r.Run(":8080")
}