james/yarr
4
0
Fork 0
mirror of https://github.com/nkanaev/yarr.git synced 2025-05-24 00:33:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

db parameter

Browse Source
This commit is contained in:
Nazar Kanaev 2020-09-18 21:21:54 +01:00
parent 1d977c543a
commit 65ddfc297d
3 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -2,3 +2,4 @@
/gofeed
/_output
/yarr
*.db

19
main.go
View File

@ -11,17 +11,24 @@ import (
)
func main() {
var addr, storageFile string
flag.StringVar(&addr, "addr", "127.0.0.1:7070", "address to run server on")
flag.StringVar(&storageFile, "db", "", "storage file path")
flag.Parse()
logger := log.New(os.Stdout, "", log.Ldate|log.Ltime|log.Lshortfile)
configPath, err := os.UserConfigDir()
if err != nil {
logger.Fatal("Failed to get config dir: ", err)
}
storagePath := filepath.Join(configPath, "yarr")
storageFile := filepath.Join(storagePath, "storage.db")
if err := os.MkdirAll(storagePath, 0755); err != nil {
logger.Fatal("Failed to create app config dir: ", err)
if storageFile == "" {
storagePath := filepath.Join(configPath, "yarr")
if err := os.MkdirAll(storagePath, 0755); err != nil {
logger.Fatal("Failed to create app config dir: ", err)
}
storageFile = filepath.Join(storagePath, "storage.db")
}
db, err := storage.New(storageFile, logger)
@ -29,10 +36,6 @@ func main() {
logger.Fatal("Failed to initialise database: ", err)
}
var addr string
flag.StringVar(&addr, "addr", "127.0.0.1:7070", "address to run server on")
flag.Parse()
srv := server.New(db, logger, addr)
platform.Start(srv)
}

2
server/server.go
View File

@ -33,7 +33,7 @@ func (h *Handler) Start() {
h.startJobs()
s := &http.Server{Addr: h.Addr, Handler: h}
s.ListenAndServe()
}
}
func (h Handler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
route, vars := getRoute(req)