init

server/logic/gateway_logic_gen.go

@@ -0,0 +1,25 @@
+package logic
+
+import (
+	"context"
+	"fusen-gateway/gen/go/service"
+
+	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
+	"google.golang.org/grpc"
+)
+
+func AutoRegisterHandler(ctx context.Context, mux *runtime.ServeMux, opts ...grpc.DialOption) error {
+	var err error
+
+	err = service.RegisterAuthHandlerClientNacos(ctx, mux, opts...)
+	if err != nil {
+		return err
+	}
+
+	err = service.RegisterInfoHandlerClientNacos(ctx, mux, opts...)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

server/logic/gateway_response.go

@@ -0,0 +1,56 @@
+package logic
+
+import (
+	"context"
+	"encoding/json"
+	"fusen-gateway/proto/goutils/basic"
+	"net/http"
+
+	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
+	"google.golang.org/protobuf/encoding/protojson"
+	"google.golang.org/protobuf/proto"
+)
+
+type EmptyMarshaler struct {
+	runtime.JSONPb
+}
+
+func (m *EmptyMarshaler) Marshal(v interface{}) ([]byte, error) {
+	return nil, nil
+}
+
+var fsProtoJSON = &runtime.JSONPb{
+	MarshalOptions: protojson.MarshalOptions{
+		EmitUnpopulated: true,
+	},
+	UnmarshalOptions: protojson.UnmarshalOptions{
+		DiscardUnknown: true,
+	},
+}
+
+func ResponseHeaderMatcher(ctx context.Context, w http.ResponseWriter, resp proto.Message) error {
+	headers := w.Header()
+	key := http.CanonicalHeaderKey("Grpc-Metadata-" + string(basic.M_FusenResponse))
+	w.WriteHeader(200)
+
+	if location, ok := headers[key]; ok {
+		// w.Header().Set("Location", location[0])
+		// w.WriteHeader(http.StatusFound)
+		jbuf := []byte(location[0])
+		gresp := &basic.GatewayResponse{}
+		err := json.Unmarshal(jbuf, gresp)
+		if err != nil {
+			return err
+		}
+		gresp.Data = resp
+		rdata, err := fsProtoJSON.Marshal(resp)
+		if err != nil {
+			return err
+		}
+
+		w.Write(rdata)
+	}
+	// w.WriteHeader(200)
+	// w.Write([]byte("error"))
+	return nil
+}