我引用了irbanan关于支持PostGIS的空间数据类型的回答。我正在使用MySQL,并尝试为自定义数据类型EWKBGeomPoint
实现Value()
。
我的Gorm模型:
import (
"github.com/twpayne/go-geom"
"github.com/twpayne/go-geom/encoding/ewkb"
)
type EWKBGeomPoint geom.Point
type Tag struct {
Name string `json:"name"`
json:"siteID"` // forign key
Loc EWKBGeomPoint `json:"loc"`
}
据我所知,MySQL支持这样的插入:
INSERT INTO `tag` (`name`,`loc`) VALUES ('tag name',ST_GeomFromText('POINT(10.000000 20.000000)'))
或
INSERT INTO `tag` (`name`,`loc`) VALUES ('tag name', ST_GeomFromWKB(X'0101000000000000000000F03F000000000000F03F'))
如果我自己做一个Value()
来满足database/sql
的Valuer
接口:
func (g EWKBGeomPoint) Value() (driver.Value, error) {
log.Println("EWKBGeomPoint value called")
b := geom.Point(g)
bp := &b
floatArr := bp.Coords()
return fmt.Sprintf("ST_GeomFromText('POINT(%f %f)')", floatArr[0], floatArr[1]), nil
}
包括ST_GeomFromText()
在内的整个值都引用在Gorm的一句话中,因此它不起作用:
INSERT INTO `tag` (`name`,`loc`) VALUES ('tag name','ST_GeomFromText('POINT(10.000000 20.000000)')');
我该如何让它发挥作用?
编辑1:
我追踪到Gorm代码,最终得到了callback_create.go
的createCallback
函数。在它内部检查if primaryField == nil
,这是真的,它进入调用scope.SQLDB().Exec
,然后我无法进一步跟踪。
scope.SQL是字符串INSERT INTO
标记(
名称,
位置) VALUES (?,?)
和scope.SQLVars
打印[tag name {{1 2 [10 20] 0}}]
。看起来插值发生在这个调用内部。
这是在调用database/sql
代码吗?
编辑2:
在这里发现了一个类似的Stackoverflow问题。但我不明白解决办法。
这里有另一种方法;使用二进制编码。
根据该文档,MySQL使用4个字节存储几何体值,以指示SRID(空间参考ID(,然后是该值的WKB(众所周知的二进制(表示。
因此,一个类型可以使用WKB编码,并在Value((和Scan((函数中添加和删除四字节前缀。在其他答案中找到的go geom库有一个WKB编码包,github.com/twpayne/go-geom/encoding/WKB。
例如:
type MyPoint struct {
Point wkb.Point
}
func (m *MyPoint) Value() (driver.Value, error) {
value, err := m.Point.Value()
if err != nil {
return nil, err
}
buf, ok := value.([]byte)
if !ok {
return nil, fmt.Errorf("did not convert value: expected []byte, but was %T", value)
}
mysqlEncoding := make([]byte, 4)
binary.LittleEndian.PutUint32(mysqlEncoding, 4326)
mysqlEncoding = append(mysqlEncoding, buf...)
return mysqlEncoding, err
}
func (m *MyPoint) Scan(src interface{}) error {
if src == nil {
return nil
}
mysqlEncoding, ok := src.([]byte)
if !ok {
return fmt.Errorf("did not scan: expected []byte but was %T", src)
}
var srid uint32 = binary.LittleEndian.Uint32(mysqlEncoding[0:4])
err := m.Point.Scan(mysqlEncoding[4:])
m.Point.SetSRID(int(srid))
return err
}
使用MyPoint类型定义标签:
type Tag struct {
Name string `gorm:"type:varchar(50);primary_key"`
Loc *MyPoint `gorm:"column:loc"`
}
func (t Tag) String() string {
return fmt.Sprintf("%s @ Point(%f, %f)", t.Name, t.Loc.Point.Coords().X(), t.Loc.Point.Coords().Y())
}
使用类型创建标签:
tag := &Tag{
Name: "London",
Loc: &MyPoint{
wkb.Point{
geom.NewPoint(geom.XY).MustSetCoords([]float64{0.1275, 51.50722}).SetSRID(4326),
},
},
}
err = db.Create(&tag).Error
if err != nil {
log.Fatalf("create: %v", err)
}
MySQL结果:
mysql> describe tag;
+-------+-------------+------+-----+---------+-------+
| Field | Type | Null | Key | Default | Extra |
+-------+-------------+------+-----+---------+-------+
| name | varchar(50) | NO | PRI | NULL | |
| loc | geometry | YES | | NULL | |
+-------+-------------+------+-----+---------+-------+
mysql> select name, st_astext(loc) from tag;
+--------+------------------------+
| name | st_astext(loc) |
+--------+------------------------+
| London | POINT(0.1275 51.50722) |
+--------+------------------------+
- (ArcGIS表示4326是世界上存储引用数据最常见的空间引用。它是PostGIS空间数据库和GeoJSON标准的默认值。它也默认用于大多数web映射库。(
更新:这种方法不起作用
钩子可以让你把柱子设置成一个gorm。在Gorm的sql生成之前的Expr。
例如,插入之前的内容如下:
func (t *Tag) BeforeCreate(scope *gorm.Scope) error {
x, y := .... // tag.Loc coordinates
text := fmt.Sprintf("POINT(%f %f)", x, y)
expr := gorm.Expr("ST_GeomFromText(?)", text)
scope.SetColumn("loc", expr)
return nil
}