HikariTish

1
func InitDB() {
2
    dsn := "user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
3
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
4
    if err != nil {
5
        log.Fatal(err)
6
    }
7

8
    sqlDB, err := db.DB()
9
    if err != nil {
10
        log.Fatal(err)
11
    }
12

13
    // 设置连接池参数
14
    sqlDB.SetMaxIdleConns(10)           // 最大空闲连接数
15
    sqlDB.SetMaxOpenConns(100)          // 最大打开连接数
16
    sqlDB.SetConnMaxLifetime(time.Hour) // 连接最大生命周期
17
}

1
import "github.com/gin-contrib/gzip"
2

3
func main() {
4
    r := gin.Default()
5

6
    // 使用 Gzip 中间件
7
    r.Use(gzip.Gzip(gzip.DefaultCompression))
8

9
    r.GET("/data", func(c *gin.Context) {
10
        c.JSON(http.StatusOK, gin.H{
11
            "message": "Large data response",
12
            "data":    strings.Repeat("x", 10000),
13
        })
14
    })
15

16
    r.Run(":8080")
17
}

1
import (
2
    "github.com/patrickmn/go-cache"
3
    "time"
4
)
5

6
var cacheStore *cache.Cache
7

8
func InitCache() {
9
    // 创建缓存（默认过期时间 5 分钟，清理间隔 10 分钟）
10
    cacheStore = cache.New(5*time.Minute, 10*time.Minute)
11
}
12

13
func CacheMiddleware(duration time.Duration) gin.HandlerFunc {
14
    return func(c *gin.Context) {
15
        // 生成缓存键
16
        key := c.Request.URL.Path + "?" + c.Request.URL.RawQuery
17

18
        // 尝试从缓存获取
19
        if cached, found := cacheStore.Get(key); found {
20
            c.JSON(http.StatusOK, cached)
21
            c.Abort()
22
            return
23
        }
24

25
        // 创建响应写入器
26
        writer := &responseWriter{
27
            ResponseWriter: c.Writer,
28
            body:           &bytes.Buffer{},
29
        }
30
        c.Writer = writer
31

32
        c.Next()
33

34
        // 缓存响应
35
        if c.Writer.Status() == http.StatusOK {
36
            cacheStore.Set(key, writer.body.String(), duration)
37
        }
38
    }
39
}
40

41
type responseWriter struct {
42
    gin.ResponseWriter
43
    body *bytes.Buffer
44
}
45

46
func (w *responseWriter) Write(b []byte) (int, error) {
47
    w.body.Write(b)
48
    return w.ResponseWriter.Write(b)
49
}

1
func main() {
2
    // 使用 gin.New() 而不是 gin.Default()
3
    r := gin.New()
4

5
    // 只添加必要的中间件
6
    r.Use(gin.Recovery())
7

8
    // 使用路由分组
9
    api := r.Group("/api/v1")
10
    {
11
        users := api.Group("/users")
12
        {
13
            users.GET("", getUsers)
14
            users.POST("", createUser)
15
            users.GET("/:id", getUser)
16
            users.PUT("/:id", updateUser)
17
            users.DELETE("/:id", deleteUser)
18
        }
19
    }
20

21
    r.Run(":8080")
22
}