60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package writer
|
|
|
|
import (
|
|
"imposm3/cache"
|
|
"imposm3/database"
|
|
"imposm3/element"
|
|
"imposm3/expire"
|
|
"imposm3/geom/limit"
|
|
"imposm3/mapping"
|
|
"imposm3/stats"
|
|
"runtime"
|
|
"sync"
|
|
)
|
|
|
|
type ErrorLevel interface {
|
|
Level() int
|
|
}
|
|
|
|
type looper interface {
|
|
loop()
|
|
}
|
|
|
|
type OsmElemWriter struct {
|
|
osmCache *cache.OSMCache
|
|
diffCache *cache.DiffCache
|
|
progress *stats.Statistics
|
|
insertBuffer database.RowInserter
|
|
wg *sync.WaitGroup
|
|
limiter *limit.Limiter
|
|
writer looper
|
|
srid int
|
|
expireTiles *expire.Tiles
|
|
}
|
|
|
|
func (writer *OsmElemWriter) SetLimiter(limiter *limit.Limiter) {
|
|
writer.limiter = limiter
|
|
}
|
|
|
|
func (writer *OsmElemWriter) Start() {
|
|
for i := 0; i < runtime.NumCPU(); i++ {
|
|
writer.wg.Add(1)
|
|
go writer.writer.loop()
|
|
}
|
|
}
|
|
|
|
func (writer *OsmElemWriter) SetExpireTiles(expireTiles *expire.Tiles) {
|
|
writer.expireTiles = expireTiles
|
|
}
|
|
|
|
func (writer *OsmElemWriter) Close() {
|
|
writer.wg.Wait()
|
|
}
|
|
|
|
func (writer *OsmElemWriter) insertMatches(elem *element.OSMElem, matches []mapping.Match) {
|
|
for _, match := range matches {
|
|
row := match.Row(elem)
|
|
writer.insertBuffer.Insert(match.Table.Name, row)
|
|
}
|
|
}
|