diff --git a/main.go b/main.go index f25a7b9..b2edc4b 100644 --- a/main.go +++ b/main.go @@ -18,7 +18,7 @@ var uploadFolder string func main() { var err error - dbpool, err = pgxpool.New(context.Background(), "postgresql://postgres:postgres@172.19.0.3:5432/postgres") + dbpool, err = pgxpool.New(context.Background(), "postgresql://postgres:postgres@172.18.0.3:5432/postgres") if err != nil { log.Fatal(err) } @@ -54,6 +54,16 @@ func main() { } uploadFolder = fmt.Sprintf("%s/videos", currentDir) + if _, err := os.Stat(uploadFolder); err != nil { + if !os.IsNotExist(err) { + log.Panicf("main: %v\n", err) + } + + if err := os.Mkdir(uploadFolder, 0750); err != nil { + log.Panicf("main: %v\n", err) + } + } + router := gin.Default() router.SetTrustedProxies(nil) @@ -103,7 +113,7 @@ func receiveChunk(c *gin.Context) { return } - f, err := os.OpenFile(fmt.Sprintf("%s/%s", uploadFolder, c.GetHeader("file-name")), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) + f, err := os.OpenFile(fmt.Sprintf("%s/%s", uploadFolder, c.GetHeader("file-name")), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0640) if err != nil { c.JSON(http.StatusInternalServerError, "") log.Panicf("receiveChunk: %v\n", err)