線上客服系統的源碼中Golang Gin框架實現IP白名單機制

我的客服系統有一些介面是專門給內部調用的，只允許其他內部系統來調用，不允許隨意訪問，可以使用IP白名單機制

使用 Gin 框架實現 IP 白名單機制可以使用中間件的方式實現。你可以編寫一個中間件函數，在每個請求到來時檢查它的 IP 地址是否在白名單中，如果不在，則返回錯誤信息。

例如，你可以這樣編寫中間件函數：

func IPWhiteList(whitelist []string) gin.HandlerFunc {
    return func(c *gin.Context) {
        // 獲取請求的 IP 地址
        ip := c.ClientIP()
        // 檢查 IP 地址是否在白名單中
        allowed := false
        for _, value := range whitelist {
            if value == ip {
                allowed = true
                break
            }
        }
        // 如果 IP 地址不在白名單中，則返回錯誤信息
        if !allowed {
            c.AbortWithStatusJSON(http.StatusForbidden, gin.H{"error": "IP address not allowed"})
            return
        }
        // 允許請求繼續訪問後續的處理函數
        c.Next()
    }
}

然後，你可以將這個中間件函數添加到你的 Gin 路由器中，以便攔截所有請求併進行 IP 白名單檢查。

例如，你可以這樣使用中間件函數：

// 創建 Gin 路由器
router := gin.Default()

// 定義 IP 白名單
whitelist := []string{"127.0.0.1", "192.168.0.100"}

// 使用中間件函數檢查 IP 白名單
router.Use(IPWhiteList(whitelist))

如果不想攔截所有的，你也可以像我的這樣，只給某幾個介面加

//IP白名單
func SystemIpWhite(c *gin.Context) {
    ip := c.ClientIP()

        //白名單數據源
    blackList := models.FindConfig("SystemWhiteList")
    strings.ReplaceAll(blackList, "\r\n", "\n")
    list := strings.Split(blackList, "\n")
    exist := false
    for _, word := range list {
        word = strings.Trim(word, " ")
        if word == "" {
            continue
        }
        if ip == word {
            exist = true
            break
        }
    }

    if !exist {
        log.Println("ip whitelist forbidden", ip)
        c.String(403, "403 forbidden")
        c.Abort()
        return
    }
}

        //內部介面，白名單驗證來源IP,傳遞賬號，生成token
        otherGroup.GET("/internalToken", middleware.SystemIpWhite, controller.InternalToken)

唯一線上客服系統

https://gofly.v1kf.com

十年開發經驗程式員，離職全心創業中，歷時三年開發出的產品《唯一客服系統》