GOMODULES/pgx
1
0
Fork 0
mirror of https://github.com/jackc/pgx.git synced 2025-05-31 11:42:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add format code helpers to ConnInfo

Browse Source
This commit is contained in:
Jack Christensen 2019-08-25 00:32:11 -05:00
parent 7d83f9ba53
commit b1e25e4ea4
1 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
pgtype.go
View File

@ -170,6 +170,8 @@ type ConnInfo struct {
oidToDataType map[uint32]*DataType oidToDataType map[uint32]*DataType
nameToDataType map[string]*DataType nameToDataType map[string]*DataType
reflectTypeToDataType map[reflect.Type]*DataType reflectTypeToDataType map[reflect.Type]*DataType
oidToParamFormatCode map[uint32]int16
oidToResultFormatCode map[uint32]int16
} }
func NewConnInfo() *ConnInfo { func NewConnInfo() *ConnInfo {
@ -177,6 +179,8 @@ func NewConnInfo() *ConnInfo {
oidToDataType: make(map[uint32]*DataType, 128), oidToDataType: make(map[uint32]*DataType, 128),
nameToDataType: make(map[string]*DataType, 128), nameToDataType: make(map[string]*DataType, 128),
reflectTypeToDataType: make(map[reflect.Type]*DataType, 128), reflectTypeToDataType: make(map[reflect.Type]*DataType, 128),
oidToParamFormatCode: make(map[uint32]int16, 128),
oidToResultFormatCode: make(map[uint32]int16, 128),
} }
ci.RegisterDataType(DataType{Value: &ACLItemArray{}, Name: "_aclitem", OID: ACLItemArrayOID}) ci.RegisterDataType(DataType{Value: &ACLItemArray{}, Name: "_aclitem", OID: ACLItemArrayOID})
@ -262,6 +266,22 @@ func (ci *ConnInfo) RegisterDataType(t DataType) {
ci.oidToDataType[t.OID] = &t ci.oidToDataType[t.OID] = &t
ci.nameToDataType[t.Name] = &t ci.nameToDataType[t.Name] = &t
ci.reflectTypeToDataType[reflect.ValueOf(t.Value).Type()] = &t ci.reflectTypeToDataType[reflect.ValueOf(t.Value).Type()] = &t
{
var formatCode int16
if _, ok := t.Value.(BinaryEncoder); ok {
formatCode = BinaryFormatCode
}
ci.oidToParamFormatCode[t.OID] = formatCode
}
{
var formatCode int16
if _, ok := t.Value.(BinaryDecoder); ok {
formatCode = BinaryFormatCode
}
ci.oidToResultFormatCode[t.OID] = formatCode
}
} }
func (ci *ConnInfo) DataTypeForOID(oid uint32) (*DataType, bool) { func (ci *ConnInfo) DataTypeForOID(oid uint32) (*DataType, bool) {
@ -279,6 +299,22 @@ func (ci *ConnInfo) DataTypeForValue(v Value) (*DataType, bool) {
return dt, ok return dt, ok
} }
func (ci *ConnInfo) ParamFormatCodeForOID(oid uint32) int16 {
fc, ok := ci.oidToParamFormatCode[oid]
if ok {
return fc
}
return TextFormatCode
}
func (ci *ConnInfo) ResultFormatCodeForOID(oid uint32) int16 {
fc, ok := ci.oidToResultFormatCode[oid]
if ok {
return fc
}
return TextFormatCode
}
// DeepCopy makes a deep copy of the ConnInfo. // DeepCopy makes a deep copy of the ConnInfo.
func (ci *ConnInfo) DeepCopy() *ConnInfo { func (ci *ConnInfo) DeepCopy() *ConnInfo {
ci2 := &ConnInfo{ ci2 := &ConnInfo{