Added method OriginalString

protocol.go

@@ -49,13 +49,16 @@
 	doc string
 
 	hash string
+
+	originalString string
 }
 
 // NewProtocol creates a protocol instance.
 func NewProtocol(
 	name, namepsace string,
 	types []NamedSchema,
 	messages map[string]*Message,
+	protocolString string,
 	opts ...ProtocolOption,
 ) (*Protocol, error) {
 	var cfg protocolConfig
@@ -69,11 +72,12 @@
 	}
 
 	p := &Protocol{
-		name:       n,
-		properties: newProperties(cfg.props, protocolReserved),
-		types:      types,
-		messages:   messages,
-		doc:        cfg.doc,
+		name:           n,
+		properties:     newProperties(cfg.props, protocolReserved),
+		types:          types,
+		messages:       messages,
+		doc:            cfg.doc,
+		originalString: protocolString,
 	}
 
 	b := md5.Sum([]byte(p.String()))
@@ -97,6 +101,11 @@
 	return p.hash
 }
 
+// OriginalString returns the original string of the protocol.
+func (p *Protocol) OriginalString() string {
+	return p.originalString
+}
+
 // Types returns the types of the protocol.
 func (p *Protocol) Types() []NamedSchema {
 	return p.types
@@ -231,7 +240,7 @@
 	}
 
 	seen := seenCache{}
-	return parseProtocol(m, seen, cache)
+	return parseProtocol(m, seen, cache, protocol)
 }
 
 type protocol struct {
@@ -243,7 +252,7 @@
 	Props     map[string]any            `mapstructure:",remain"`
 }
 
-func parseProtocol(m map[string]any, seen seenCache, cache *SchemaCache) (*Protocol, error) {
+func parseProtocol(m map[string]any, seen seenCache, cache *SchemaCache, protocolString string) (*Protocol, error) {
 	var (
 		p    protocol
 		meta mapstructure.Metadata
@@ -279,7 +288,7 @@
 		}
 	}
 
-	return NewProtocol(p.Protocol, p.Namespace, types, messages, WithProtoDoc(p.Doc), WithProtoProps(p.Props))
+	return NewProtocol(p.Protocol, p.Namespace, types, messages, protocolString, WithProtoDoc(p.Doc), WithProtoProps(p.Props))
 }
 
 func parseProtocolTypes(namespace string, types []any, seen seenCache, cache *SchemaCache) ([]NamedSchema, error) {

protocol_test.go

@@ -21,11 +21,20 @@ func TestMustParseProtocol_PanicsOnError(t *testing.T) {
 }
 
 func TestNewProtocol_ValidatesName(t *testing.T) {
-	_, err := avro.NewProtocol("0test", "", nil, nil)
+	_, err := avro.NewProtocol("0test", "", nil, nil, "")
 
 	assert.Error(t, err)
 }
 
+func TestNewProtocol_OriginalString(t *testing.T) {
+	schema := `{"protocol":"test", "namespace": "org.hamba.avro", "messages":{"test":{"request": [{"name": "foobar", "type": "string"}]}}}`
+
+	proto, err := avro.ParseProtocol(schema)
+
+	assert.NoError(t, err)
+	assert.Equal(t, schema, proto.OriginalString())
+}
+
 func TestNewMessage(t *testing.T) {
 	field, _ := avro.NewField("test", avro.NewPrimitiveSchema(avro.String, nil))
 	fields := []*avro.Field{field}