-
Notifications
You must be signed in to change notification settings - Fork 41
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #16 from wk8/jrouge/json
Adding support for JSON serialization/deserialization
- Loading branch information
Showing
11 changed files
with
403 additions
and
59 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| # Changelog | ||
|
|
||
| # 2.1 - Dec 7th 2022 | ||
| * Added support for JSON serialization/deserialization |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,126 @@ | ||
| package orderedmap | ||
|
|
||
| import ( | ||
| "encoding" | ||
| "encoding/json" | ||
| "fmt" | ||
|
|
||
| "github.com/buger/jsonparser" | ||
| "github.com/mailru/easyjson/jwriter" | ||
| ) | ||
|
|
||
| var ( | ||
| _ json.Marshaler = &OrderedMap[int, any]{} | ||
| _ json.Unmarshaler = &OrderedMap[int, any]{} | ||
| ) | ||
|
|
||
| // MarshalJSON implements the json.Marshaler interface. | ||
| func (om *OrderedMap[K, V]) MarshalJSON() ([]byte, error) { | ||
| writer := jwriter.Writer{} | ||
| writer.RawByte('{') | ||
|
|
||
| for pair, firstIteration := om.Oldest(), true; pair != nil; pair = pair.Next() { | ||
| if firstIteration { | ||
| firstIteration = false | ||
| } else { | ||
| writer.RawByte(',') | ||
| } | ||
|
|
||
| switch key := any(pair.Key).(type) { | ||
| case string: | ||
| writer.String(key) | ||
| case encoding.TextMarshaler: | ||
| writer.RawByte('"') | ||
| writer.Raw(key.MarshalText()) | ||
| writer.RawByte('"') | ||
| case int: | ||
| writer.IntStr(key) | ||
| case int8: | ||
| writer.Int8Str(key) | ||
| case int16: | ||
| writer.Int16Str(key) | ||
| case int32: | ||
| writer.Int32Str(key) | ||
| case int64: | ||
| writer.Int64Str(key) | ||
| case uint: | ||
| writer.UintStr(key) | ||
| case uint8: | ||
| writer.Uint8Str(key) | ||
| case uint16: | ||
| writer.Uint16Str(key) | ||
| case uint32: | ||
| writer.Uint32Str(key) | ||
| case uint64: | ||
| writer.Uint64Str(key) | ||
| default: | ||
| return nil, fmt.Errorf("unsupported key type: %T", key) | ||
| } | ||
|
|
||
| writer.RawByte(':') | ||
| // the error is checked at the end of the function | ||
| writer.Raw(json.Marshal(pair.Value)) //nolint:errchkjson | ||
| } | ||
|
|
||
| writer.RawByte('}') | ||
|
|
||
| return writer.Buffer.Buf, writer.Error | ||
| } | ||
|
|
||
| // UnmarshalJSON implements the json.Unmarshaler interface. | ||
| func (om *OrderedMap[K, V]) UnmarshalJSON(data []byte) error { | ||
| return jsonparser.ObjectEach( | ||
| data, | ||
| func(keyData []byte, valueData []byte, dataType jsonparser.ValueType, offset int) error { | ||
| if dataType == jsonparser.String { | ||
| // jsonparser removes the enclosing quotes; we need to restore them to make a valid JSON | ||
| valueData = data[offset-len(valueData)-2 : offset] | ||
| } | ||
|
|
||
| var key K | ||
| var value V | ||
|
|
||
| if typedKeyPointer, ok := any(&key).(encoding.TextUnmarshaler); ok { | ||
| // pointer receiver | ||
| if err := typedKeyPointer.UnmarshalText(keyData); err != nil { | ||
| return err | ||
| } | ||
| } else { | ||
| keyAlreadyUnmarshalled := false | ||
| switch typedKey := any(key).(type) { | ||
| case string: | ||
| keyData = quoteString(keyData) | ||
| case encoding.TextUnmarshaler: | ||
| // not a pointer receiver | ||
| if err := typedKey.UnmarshalText(keyData); err != nil { | ||
| return err | ||
| } | ||
| keyAlreadyUnmarshalled = true | ||
| case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64: | ||
| default: | ||
| return fmt.Errorf("unsupported key type: %T", typedKey) | ||
| } | ||
|
|
||
| if !keyAlreadyUnmarshalled { | ||
| if err := json.Unmarshal(keyData, &key); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| } | ||
|
|
||
| if err := json.Unmarshal(valueData, &value); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| om.Set(key, value) | ||
| return nil | ||
| }) | ||
| } | ||
|
|
||
| func quoteString(data []byte) []byte { | ||
| withQuotes := make([]byte, len(data)+2) //nolint:gomnd | ||
| copy(withQuotes[1:], data) | ||
| withQuotes[0] = '"' | ||
| withQuotes[len(data)+1] = '"' | ||
| return withQuotes | ||
| } |
Oops, something went wrong.