go/json-iterator
1
0
Fork 0
mirror of https://github.com/json-iterator/go.git synced 2025-06-06 22:36:25 +02:00
Code Issues Releases Activity

use extension to implement EscapeHtml config option

Browse Source
This commit is contained in:
Tao Wen 2018-02-17 22:55:35 +08:00
parent ef3038593b
commit 2074f25bd3
2 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
feature_config.go
View File

@ -74,8 +74,9 @@ func (cfg Config) Froze() API {
if cfg.MarshalFloatWith6Digits { if cfg.MarshalFloatWith6Digits {
api.marshalFloatWith6Digits() api.marshalFloatWith6Digits()
} }
encoderExtension := EncoderExtension{}
if cfg.EscapeHTML { if cfg.EscapeHTML {
api.escapeHTML() api.escapeHTML(encoderExtension)
} }
if cfg.UseNumber { if cfg.UseNumber {
api.useNumber() api.useNumber()
@ -83,6 +84,9 @@ func (cfg Config) Froze() API {
if cfg.ValidateJsonRawMessage { if cfg.ValidateJsonRawMessage {
api.validateJsonRawMessage() api.validateJsonRawMessage()
} }
if len(encoderExtension) > 0 {
api.extensions = append(api.extensions, encoderExtension)
}
api.configBeforeFrozen = cfg api.configBeforeFrozen = cfg
return api return api
} }
@ -178,8 +182,8 @@ func (encoder *htmlEscapedStringEncoder) IsEmpty(ptr unsafe.Pointer) bool {
return *((*string)(ptr)) == "" return *((*string)(ptr)) == ""
} }
func (cfg *frozenConfig) escapeHTML() { func (cfg *frozenConfig) escapeHTML(encoderExtension EncoderExtension) {
cfg.addEncoderToCache(reflect.TypeOf((*string)(nil)).Elem(), &htmlEscapedStringEncoder{}) encoderExtension[reflect.TypeOf((*string)(nil)).Elem()] = &htmlEscapedStringEncoder{}
} }
func (cfg *frozenConfig) cleanDecoders() { func (cfg *frozenConfig) cleanDecoders() {

26
feature_reflect_extension.go
View File

@ -80,6 +80,32 @@ func (extension *DummyExtension) DecorateEncoder(typ reflect.Type, encoder ValEn
return encoder return encoder
} }
type EncoderExtension map[reflect.Type]ValEncoder
// UpdateStructDescriptor No-op
func (extension EncoderExtension) UpdateStructDescriptor(structDescriptor *StructDescriptor) {
}
// CreateDecoder No-op
func (extension EncoderExtension) CreateDecoder(typ reflect.Type) ValDecoder {
return nil
}
// CreateEncoder get encoder from map
func (extension EncoderExtension) CreateEncoder(typ reflect.Type) ValEncoder {
return extension[typ]
}
// DecorateDecoder No-op
func (extension EncoderExtension) DecorateDecoder(typ reflect.Type, decoder ValDecoder) ValDecoder {
return decoder
}
// DecorateEncoder No-op
func (extension EncoderExtension) DecorateEncoder(typ reflect.Type, encoder ValEncoder) ValEncoder {
return encoder
}
type funcDecoder struct { type funcDecoder struct {
fun DecoderFunc fun DecoderFunc
} }