Improve stream processing

2019-08-25 19:30:22 +01:00
parent 6128d18ee0
commit 36623bfe50
3 changed files with 41 additions and 28 deletions
--- a/codec/grpc/grpc.go
+++ b/codec/grpc/grpc.go
@@ -94,6 +94,17 @@ func (c *Codec) Write(m *codec.Message, b interface{}) error {
 		m.Header[":status"] = "200"
 		m.Header["grpc-status"] = "0"
 		//		m.Header["grpc-message"] = ""
+	case codec.Error:
+		m.Header["Trailer"] = "grpc-status, grpc-message"
+		// micro end of stream
+		if m.Error == "EOS" {
+			m.Header["grpc-status"] = "0"
+		} else {
+			m.Header["grpc-message"] = m.Error
+			m.Header["grpc-status"] = "13"
+		}
+
+		return nil
 	}

 	// marshal content