mirror of https://github.com/jackc/pgx.git
128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package pgtype
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
type JSONBCodec struct{}
|
|
|
|
func (JSONBCodec) FormatSupported(format int16) bool {
|
|
return format == TextFormatCode || format == BinaryFormatCode
|
|
}
|
|
|
|
func (JSONBCodec) PreferredFormat() int16 {
|
|
return TextFormatCode
|
|
}
|
|
|
|
func (JSONBCodec) PlanEncode(ci *ConnInfo, oid uint32, format int16, value interface{}) EncodePlan {
|
|
switch format {
|
|
case BinaryFormatCode:
|
|
plan := JSONCodec{}.PlanEncode(ci, oid, TextFormatCode, value)
|
|
if plan != nil {
|
|
return &encodePlanJSONBCodecBinaryWrapper{textPlan: plan}
|
|
}
|
|
case TextFormatCode:
|
|
return JSONCodec{}.PlanEncode(ci, oid, format, value)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type encodePlanJSONBCodecBinaryWrapper struct {
|
|
textPlan EncodePlan
|
|
}
|
|
|
|
func (plan *encodePlanJSONBCodecBinaryWrapper) Encode(value interface{}, buf []byte) (newBuf []byte, err error) {
|
|
buf = append(buf, 1)
|
|
return plan.textPlan.Encode(value, buf)
|
|
}
|
|
|
|
func (JSONBCodec) PlanScan(ci *ConnInfo, oid uint32, format int16, target interface{}, actualTarget bool) ScanPlan {
|
|
switch format {
|
|
case BinaryFormatCode:
|
|
plan := JSONCodec{}.PlanScan(ci, oid, TextFormatCode, target, actualTarget)
|
|
if plan != nil {
|
|
return &scanPlanJSONBCodecBinaryUnwrapper{textPlan: plan}
|
|
}
|
|
case TextFormatCode:
|
|
return JSONCodec{}.PlanScan(ci, oid, format, target, actualTarget)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type scanPlanJSONBCodecBinaryUnwrapper struct {
|
|
textPlan ScanPlan
|
|
}
|
|
|
|
func (plan *scanPlanJSONBCodecBinaryUnwrapper) Scan(src []byte, dst interface{}) error {
|
|
if src == nil {
|
|
return plan.textPlan.Scan(src, dst)
|
|
}
|
|
|
|
if len(src) == 0 {
|
|
return fmt.Errorf("jsonb too short")
|
|
}
|
|
|
|
if src[0] != 1 {
|
|
return fmt.Errorf("unknown jsonb version number %d", src[0])
|
|
}
|
|
|
|
return plan.textPlan.Scan(src[1:], dst)
|
|
}
|
|
|
|
func (c JSONBCodec) DecodeDatabaseSQLValue(ci *ConnInfo, oid uint32, format int16, src []byte) (driver.Value, error) {
|
|
if src == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
switch format {
|
|
case BinaryFormatCode:
|
|
if len(src) == 0 {
|
|
return nil, fmt.Errorf("jsonb too short")
|
|
}
|
|
|
|
if src[0] != 1 {
|
|
return nil, fmt.Errorf("unknown jsonb version number %d", src[0])
|
|
}
|
|
|
|
dstBuf := make([]byte, len(src)-1)
|
|
copy(dstBuf, src[1:])
|
|
return dstBuf, nil
|
|
case TextFormatCode:
|
|
dstBuf := make([]byte, len(src))
|
|
copy(dstBuf, src)
|
|
return dstBuf, nil
|
|
default:
|
|
return nil, fmt.Errorf("unknown format code: %v", format)
|
|
}
|
|
}
|
|
|
|
func (c JSONBCodec) DecodeValue(ci *ConnInfo, oid uint32, format int16, src []byte) (interface{}, error) {
|
|
if src == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
switch format {
|
|
case BinaryFormatCode:
|
|
if len(src) == 0 {
|
|
return nil, fmt.Errorf("jsonb too short")
|
|
}
|
|
|
|
if src[0] != 1 {
|
|
return nil, fmt.Errorf("unknown jsonb version number %d", src[0])
|
|
}
|
|
|
|
src = src[1:]
|
|
case TextFormatCode:
|
|
default:
|
|
return nil, fmt.Errorf("unknown format code: %v", format)
|
|
}
|
|
|
|
var dst interface{}
|
|
err := json.Unmarshal(src, &dst)
|
|
return dst, err
|
|
}
|