package main

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"

	"github.com/go-kit/kit/transport/http"
)

func main() {
	http.Handle("/your-endpoint", http.NewServer(
		yourEndpoint,
		decodeRequest,
		encodeResponse,
	))
}

// 請求和響應類型
type YourRequest struct {
	// 定義你的請求結(jié)構
	// ...
}

type YourResponse struct {
	// 定義你的響應結(jié)構
	// ...
}

// 你的端點邏輯
func yourEndpoint(ctx context.Context, request interface{}) (interface{}, error) {
	// 獲取原始請求體
	rawBody, ok := request.(json.RawMessage)
	if !ok {
		return nil, errors.New("無法訪問原始請求體")
	}

	// 根據(jù)需要處理原始請求體
	fmt.Println("原始請求體:", string(rawBody))

	// 你的實際端點邏輯在這里
	// ...

	// 返回響應（示例響應）
	return YourResponse{Message: "請求成功處理"}, nil
}

// 請求解碼器以獲取原始請求體
func decodeRequest(_ context.Context, r *http.Request) (interface{}, error) {
	// 讀取原始請求體
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		return nil, err
	}

	// 將原始請求體作為json.RawMessage返回
	return json.RawMessage(body), nil
}

// 響應編碼器
func encodeResponse(_ context.Context, w http.ResponseWriter, response interface{}) error {
	return json.NewEncoder(w).Encode(response)
}

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"

	"github.com/go-kit/kit/endpoint"
	"github.com/go-kit/kit/log"
	"github.com/go-kit/kit/transport/http"
)

// 表示請求負載的結(jié)構體
type Request struct {
	Message string `json:"message"`
}

// 表示響應負載的結(jié)構體
type Response struct {
	Result string `json:"result"`
}

func main() {
	// 創(chuàng)建一個簡單的Go Kit服務
	var svc MyService
	endpoint := makeUppercaseEndpoint(&svc)

	// 創(chuàng)建一個Go Kit HTTP傳輸
	httpHandler := http.NewServer(
		endpoint,
		decodeRequest,
		encodeResponse,
	)

	// 啟動HTTP服務器
	http.ListenAndServe(":8080", httpHandler)
}

// MyService是一個只有一個方法的簡單服務
type MyService struct{}

// Uppercase是MyService上的一個方法
func (MyService) Uppercase(ctx context.Context, message string) (string, error) {
	return fmt.Sprintf("接收到消息：%s", message), nil
}

// makeUppercaseEndpoint是創(chuàng)建Uppercase方法的Go Kit端點的輔助函數(shù)
func makeUppercaseEndpoint(svc MyService) endpoint.Endpoint {
	return func(ctx context.Context, request interface{}) (interface{}, error) {
		req := request.(Request)
		result, err := svc.Uppercase(ctx, req.Message)
		return Response{Result: result}, err
	}
}

// decodeRequest是解碼傳入JSON請求的輔助函數(shù)
func decodeRequest(_ context.Context, r *http.Request) (interface{}, error) {
	var request Request
	if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
		return nil, err
	}
	return request, nil
}

// encodeResponse是編碼傳出JSON響應的輔助函數(shù)
func encodeResponse(_ context.Context, w http.ResponseWriter, response interface{}) error {
	return json.NewEncoder(w).Encode(response)
}