我试图弄清楚如何在 Go 中构建数据存储抽象。我想我了解接口的基础知识。但是,我遇到的问题是网上所有的例子都只给你展示了最简单的例子,没有其他的。
我想要做的是弄清楚如何以及在哪里放置 SQL 代码。我试图编写最简单的代码来说明我正在尝试做的事情(是的,没有错误代码,是的,路径结构不是惯用的)。我有一个包含两个表的数据库。一个用于存储 Circles,一个用于存储 Squares。我在 Go 中有对象来制作这些。我的目录结构是:
project/main.go
project/test.db
project/shapes/shape.go
project/shapes/circle/circle.go
project/shapes/square/square.go
project/datastore/datastore.go
project/datastore/sqlite3/sqlite3.go
我能想到的唯一方法是将 SQL INSERT 和 SELECT 代码放在实际的形状文件(circle.go 和 square.go)中。但这感觉真的不对。它似乎应该是 datastore/sqlite3 包的一部分。我只是不知道如何使这项工作。这是我到目前为止所拥有的:
main.go
package main
import (
"fmt"
"project/datastore/sqlite3"
"project/shapes/circle"
"project/shapes/square"
)
func main() {
c := circle.New(4)
area := c.Area()
fmt.Println("Area: ", area)
s := square.New(12)
area2 := s.Area()
fmt.Println("Area: ", area2)
db := sqlite3.New("test.db")
db.Put(c)
db.Close()
}
形状/形状.go
package shapes
type Shaper interface {
Area() float64
}
形状/圆/circle.go
package circle
import (
"project/shapes"
)
type CircleType struct {
Radius float64
}
func New(radius float64) shapes.Shaper {
var c CircleType
c.Radius = radius
return &c
}
func (c *CircleType) Area() float64 {
area := 3.1415 * c.Radius * c.Radius
return area
}
形状/正方形/square.go
package square
import (
"project/shapes"
)
type SquareType struct {
Side float64
}
func New(side float64) shapes.Shaper {
var s SquareType
s.Side = side
return &s
}
func (s *SquareType) Area() float64 {
area := s.Side * s.Side
return area
}
数据存储/datastore.go
package datastore
import (
"project/shapes"
)
type ShapeStorer interface {
Put(shape shapes.Shaper)
Close()
}
数据存储/sqlite3/sqlite3.go
package sqlite3
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
"log"
"project/datastore"
"project/shapes"
)
type Sqlite3DatastoreType struct {
DB *sql.DB
}
func New(filename string) datastore.ShapeStorer {
var ds Sqlite3DatastoreType
db, sqlerr := sql.Open("sqlite3", filename)
if sqlerr != nil {
log.Fatalln("Unable to open file due to error: ", sqlerr)
}
ds.DB = db
return &ds
}
func (ds *Sqlite3DatastoreType) Close() {
err := ds.DB.Close()
if err != nil {
log.Fatalln(err)
}
}
func (ds *Sqlite3DatastoreType) Put(shape shapes.Shaper) {
log.Println("Help")
// here you could either do a switch statement on the object type
// or you could do something like shape.Write(), if Write() was defined
// on the interface of shape/shapes.go Shaper interface and then
// implemented in the Square and Circle objects.
}
由于 Circle 和 Square 对象的数据库表会有所不同,因此我需要为每个对象设置一个方法。如果我在 circle 包和 square 包中添加一个方法来进行插入,我可以让它工作。但就像我上面说的,感觉就像我做错了。
提前非常感谢。