go 实现api数据安全传输-Go服务器开发

在Go语言中实现API数据的安全传输通常涉及以下几个方面：

使用HTTPS：通过SSL/TLS加密传输数据，确保数据在网络上安全。
身份验证和授权：使用OAuth 2.0、JWT（JSON Web Tokens）等机制来保证只有经过授权的用户才能访问API。
数据加密：在发送敏感数据时，可以对其进行加密处理。

下面是一个简单的示例，演示如何在Go中创建一个安全的HTTPS API，并使用JWT进行身份验证。

1. 创建HTTPS服务器

首先，你需要生成SSL证书。可以使用OpenSSL命令生成自签名证书（仅用于测试）：

openssl req -newkey rsa:2048 -nodes -keyout server.key -x509 -days 365 -out server.crt

然后，创建一个简单的HTTPS服务器示例：

package main

import (
	"fmt"
	"net/http"
	"time"
)

func handler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "Hello, Secure World!")
}

func main() {
	http.HandleFunc("/", handler)
	
	// 启动HTTPS服务器
	err := http.ListenAndServeTLS(":443", "server.crt", "server.key", nil)
	if err != nil {
		fmt.Println("Error starting server:", err)
	}
}

2. 使用JWT进行身份验证

安装 github.com/dgrijalva/jwt-go 包：

go get github.com/dgrijalva/jwt-go

然后更新你的代码以支持JWT身份验证：

package main

import (
	"fmt"
	"net/http"
	"time"

	"github.com/dgrijalva/jwt-go"
)

var mySigningKey = []byte("secret")

// 创建JWT令牌
func CreateToken(username string) (string, error) {
	token := jwt.New(jwt.SigningMethodHS256)
	expirationTime := time.Now().Add(5 * time.Minute)
	token.Claims["exp"] = expirationTime.Unix()
	token.Claims["username"] = username

	return token.SignedString(mySigningKey)
}

// 验证JWT令牌
func ValidateToken(tokenString string) (*jwt.Token, error) {
	return jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
		return mySigningKey, nil
	})
}

// 登录处理函数，发放JWT令牌
func loginHandler(w http.ResponseWriter, r *http.Request) {
	username := r.FormValue("username")
	password := r.FormValue("password") // 实际应用中应该更安全地处理密码

	tokenString, err := CreateToken(username)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	w.Write([]byte(tokenString))
}

// 受保护的API处理函数，需要Bearer Token进行访问
func protectedHandler(w http.ResponseWriter, r *http.Request) {
	authHeader := r.Header.Get("Authorization")
	if authHeader == "" || len(authHeader) < 7 || authHeader[:6] != "Bearer" {
		http.Error(w, "Unauthorized", http.StatusUnauthorized)
		return
	}

	tokenString := authHeader[7:]
	if _, err := ValidateToken(tokenString); err != nil {
		http.Error(w, "Unauthorized", http.StatusUnauthorized)
		return
	}

	w.Write([]byte("This is a protected route"))
}

func main() {
	http.HandleFunc("/login", loginHandler)
	http.HandleFunc("/protected", protectedHandler)

	err := http.ListenAndServeTLS(":443", "server.crt", "server.key", nil)
	if err != nil {
	    fmt.Println("Error starting server:", err)
    }
}