40 строки
786 B
Go
40 строки
786 B
Go
package netaddr
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"errors"
|
|
"net"
|
|
)
|
|
|
|
// Inet is a wrapper for transferring Inet values back and forth easily.
|
|
type Inet struct {
|
|
Inet net.IP
|
|
}
|
|
|
|
// Scan implements the Scanner interface.
|
|
func (i *Inet) Scan(value interface{}) error {
|
|
i.Inet = nil
|
|
if value == nil {
|
|
return nil
|
|
}
|
|
ipAsBytes, ok := value.([]byte)
|
|
if !ok {
|
|
return errors.New("could not convert scanned value to bytes")
|
|
}
|
|
parsedIP := net.ParseIP(string(ipAsBytes))
|
|
if parsedIP == nil {
|
|
return nil
|
|
}
|
|
i.Inet = parsedIP
|
|
return nil
|
|
}
|
|
|
|
// Value implements the driver Valuer interface. Note if
|
|
// i.IP is nil the database column value will be set to NULL.
|
|
func (i Inet) Value() (driver.Value, error) {
|
|
if i.Inet == nil {
|
|
return nil, nil
|
|
}
|
|
return []byte(i.Inet.String()), nil
|
|
}
|