Add database/sql support to pgtype
This commit is contained in:
+15
-37
@@ -2,47 +2,13 @@ package pgtype
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql/driver"
|
||||
"errors"
|
||||
)
|
||||
|
||||
func DatabaseSQLValue(ci *ConnInfo, src Value) (interface{}, error) {
|
||||
switch src := src.(type) {
|
||||
case *Bool:
|
||||
return src.Bool, nil
|
||||
case *Bytea:
|
||||
return src.Bytes, nil
|
||||
case *Date:
|
||||
if src.InfinityModifier == None {
|
||||
return src.Time, nil
|
||||
}
|
||||
case *Float4:
|
||||
return float64(src.Float), nil
|
||||
case *Float8:
|
||||
return src.Float, nil
|
||||
case *GenericBinary:
|
||||
return src.Bytes, nil
|
||||
case *GenericText:
|
||||
return src.String, nil
|
||||
case *Int2:
|
||||
return int64(src.Int), nil
|
||||
case *Int4:
|
||||
return int64(src.Int), nil
|
||||
case *Int8:
|
||||
return int64(src.Int), nil
|
||||
case *Text:
|
||||
return src.String, nil
|
||||
case *Timestamp:
|
||||
if src.InfinityModifier == None {
|
||||
return src.Time, nil
|
||||
}
|
||||
case *Timestamptz:
|
||||
if src.InfinityModifier == None {
|
||||
return src.Time, nil
|
||||
}
|
||||
case *Unknown:
|
||||
return src.String, nil
|
||||
case *Varchar:
|
||||
return src.String, nil
|
||||
if valuer, ok := src.(driver.Valuer); ok {
|
||||
return valuer.Value()
|
||||
}
|
||||
|
||||
buf := &bytes.Buffer{}
|
||||
@@ -64,3 +30,15 @@ func DatabaseSQLValue(ci *ConnInfo, src Value) (interface{}, error) {
|
||||
|
||||
return nil, errors.New("cannot convert to database/sql compatible value")
|
||||
}
|
||||
|
||||
func encodeValueText(src TextEncoder) (interface{}, error) {
|
||||
buf := &bytes.Buffer{}
|
||||
null, err := src.EncodeText(nil, buf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if null {
|
||||
return nil, nil
|
||||
}
|
||||
return buf.String(), err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user