Create intermediate dirs for sqlite database (#1281)

datacommonsorg · Oct 5, 2023 · d268ecf · d268ecf
1 parent bdb0b80
commit d268ecf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -189,11 +189,11 @@ func main() {
 	if *useSQLite {
 		sqlClient, err = sqlite.CreateDB(*sqlDataPath)
 		if err != nil {
-			log.Fatalf("Can not open sqlite3 database from: %s", *sqlDataPath)
+			log.Fatalf("Cannot open sqlite3 database from: %s: %v", *sqlDataPath, err)
 		}
 		err := sqldb.CreateTables(sqlClient)
 		if err != nil {
-			log.Fatalf("Can not create tables %v", err)
+			log.Fatalf("Cannot create tables %v", err)
 		}
 		defer sqlClient.Close()
 	}
@@ -204,11 +204,11 @@ func main() {
 		} else {
 			sqlClient, err = cloudsql.ConnectWithConnector(*cloudSQLInstance)
 			if err != nil {
-				log.Fatalf("Can not open cloud sql database from %s: %v", *cloudSQLInstance, err)
+				log.Fatalf("Cannot open cloud sql database from %s: %v", *cloudSQLInstance, err)
 			}
 			err := sqldb.CreateTables(sqlClient)
 			if err != nil {
-				log.Fatalf("Can not create tables %v", err)
+				log.Fatalf("Cannot create tables %v", err)
 			}
 			defer sqlClient.Close()
 		}

diff --git a/internal/sqldb/sqlite/sqlite.go b/internal/sqldb/sqlite/sqlite.go
@@ -21,6 +21,10 @@ import (
 )
 
 func CreateDB(fileDir string) (*sql.DB, error) {
+	// Create all intermediate directories.
+	if err := os.MkdirAll(fileDir, 0755); err != nil {
+		return nil, err
+	}
 	dbPath := filepath.Join(fileDir, "datacommons.db")
 	_, err := os.Stat(dbPath)
 	if err == nil {