使用Gorm插入和选择PostGIS Geometry

Z时代
2024-01-10
分类：问答

我一直在尝试找到一种使用Golang尤其是库gorm插入和检索几何类型的方法。我还尝试使用库orb，它为几何定义了不同的类型，并提供了不同格式之间的编码/解码。

Orb 已经为每种类型实现了Scan()和Value()方法。这使得go’s

Insert()和Scan()函数可以使用除基本类型以外的其他类型。但是，Orb期望使用以众所周知的二进制（WKB）格式表示的几何。

orb文档显示，要实现此目的，您只需将字段包装在PostGIS函数中ST_AsBinary()，ST_GeomFromWKB()并分别用于查询和插入。例如，表定义为：

_, err = db.Exec(`
        CREATE TABLE IF NOT EXISTS orbtest (
            id SERIAL PRIMARY KEY,
            name TEXT NOT NULL,
            geom geometry(POLYGON, 4326) NOT NULL
        );
    `)

您可以这样做：

rows, err := db.Query("SELECT id, name, ST_AsBinary(geom) FROM orbtest LIMIT 1")

对于插入（其中p是orb.Point）：

db.Exec("INSERT INTO orbtest (id, name, geom) VALUES ($1, $2, ST_GeomFromWKB($3))", 1, "Test", wkb.Value(p))

这是我的问题：通过使用GORM，我无法使用这些功能来构建那些查询。GORM将自动将给定结构的值插入数据库，并将数据扫描到结构的整个层次结构中。这些Scan()和Value()方法被称为幕后，没有我的控制。

尝试将二进制数据直接插入到几何列中是行不通的，而直接查询几何列将以十六进制形式给出结果。

我尝试了多种数据库方法来解决此问题。我试图创建视图，以自动在几何列上调用所需的函数。这适用于查询，但不能插入。

是否可以做出某种触发或规则，以自动对传入/传出的数据调用所需的功能？

我还应该注意，我正在使用的库的工作完全独立于数据和架构，因此我没有进行任何形式的查询硬编码的奢侈。我当然可以编写一个函数来扫描整个数据模型，并从头开始生成查询，但是我希望有更好的选择。

有谁知道用SQL进行这项工作的方法吗？只需查询列本身就能够自动调用列上的函数？

任何建议将不胜感激。

回答：

我最终使用的另一个解决方案是go-geos，因为我发现我需要使用GEOS

C库。这样，我就可以将结构转换WKT为要插入的结构（因为postgis将其作为常规文本接受）并可以WKB在扫描时进行转换。

type Geometry4326 *geos.Geometry
// Value converts the given Geometry4326 struct into WKT such that it can be stored in a 
// database. Implements Valuer interface for use with database operations.
func (g Geometry4326) Value() (driver.Value, error) {
    str, err := g.ToWKT()
    if err != nil {
        return nil, err
    }
    return "SRID=4326;" + str, nil
}
// Scan converts the hexadecimal representation of geometry into the given Geometry4326 
// struct. Implements Scanner interface for use with database operations.
func (g *Geometry4326) Scan(value interface{}) error {
    bytes, ok := value.([]byte)
    if !ok {
        return errors.New("cannot convert database value to geometry")
    }
    str := string(bytes)
    geom, err := geos.FromHex(str)
    if err != nil {
        return errors.Wrap(err, "cannot get geometry from hex")
    }
    geometry := Geometry4326(geom)
    *g = geometry
    return nil
}