108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
type cacheOptions struct {
|
|
CacheSizeM int
|
|
MaxOpenFiles int
|
|
BlockRestartInterval int
|
|
WriteBufferSizeM int
|
|
BlockSizeK int
|
|
MaxFileSizeM int
|
|
}
|
|
|
|
type coordsCacheOptions struct {
|
|
cacheOptions
|
|
BunchSize int
|
|
BunchCacheCapacity int
|
|
}
|
|
type osmCacheOptions struct {
|
|
Coords coordsCacheOptions
|
|
Ways cacheOptions
|
|
Nodes cacheOptions
|
|
Relations cacheOptions
|
|
CoordsIndex cacheOptions
|
|
WaysIndex cacheOptions
|
|
}
|
|
|
|
const defaultConfig = `
|
|
{
|
|
"Coords": {
|
|
"CacheSizeM": 16,
|
|
"WriteBufferSizeM": 64,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 64,
|
|
"MaxFileSizeM": 32,
|
|
"BlockRestartInterval": 256,
|
|
"BunchSize": 32,
|
|
"BunchCacheCapacity": 8096
|
|
},
|
|
"Nodes": {
|
|
"CacheSizeM": 16,
|
|
"WriteBufferSizeM": 64,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 64,
|
|
"MaxFileSizeM": 32,
|
|
"BlockRestartInterval": 128
|
|
},
|
|
"Ways": {
|
|
"CacheSizeM": 16,
|
|
"WriteBufferSizeM": 64,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 64,
|
|
"MaxFileSizeM": 32,
|
|
"BlockRestartInterval": 128
|
|
},
|
|
"Relations": {
|
|
"CacheSizeM": 16,
|
|
"WriteBufferSizeM": 64,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 64,
|
|
"MaxFileSizeM": 32,
|
|
"BlockRestartInterval": 128
|
|
},
|
|
"CoordsIndex": {
|
|
"CacheSizeM": 32,
|
|
"WriteBufferSizeM": 128,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 256,
|
|
"MaxFileSizeM": 8,
|
|
"BlockRestartInterval": 256
|
|
},
|
|
"WaysIndex": {
|
|
"CacheSizeM": 16,
|
|
"WriteBufferSizeM": 64,
|
|
"BlockSizeK": 0,
|
|
"MaxOpenFiles": 64,
|
|
"MaxFileSizeM": 8,
|
|
"BlockRestartInterval": 128
|
|
}
|
|
}
|
|
`
|
|
|
|
var globalCacheOptions osmCacheOptions
|
|
|
|
func init() {
|
|
err := json.Unmarshal([]byte(defaultConfig), &globalCacheOptions)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cacheConfFile := os.Getenv("IMPOSM_CACHE_CONFIG")
|
|
if cacheConfFile != "" {
|
|
data, err := ioutil.ReadFile(cacheConfFile)
|
|
if err != nil {
|
|
log.Println("Unable to read cache config:", err)
|
|
}
|
|
err = json.Unmarshal(data, &globalCacheOptions)
|
|
if err != nil {
|
|
log.Println("Unable to parse cache config:", err)
|
|
}
|
|
}
|
|
}
|