Check earlier that the database has a tablespace mentioned in the config

Author: joto

src/flex-table.cpp

@@ -253,16 +253,6 @@ void table_connection_t::start(bool append)
 {
     assert(m_db_connection);
 
-    for (auto const &ts :
-         {table().data_tablespace(), table().index_tablespace()}) {
-        if (!has_tablespace(ts)) {
-            throw fmt_error(
-                "Tablespace '{0}' not available."
-                " Use 'CREATE TABLESPACE \"{0}\" ...;' to create it.",
-                ts);
-        }
-    }
-
     m_db_connection->exec("SET client_min_messages = WARNING");
 
     if (!append) {

src/output-flex.cpp

@@ -434,6 +434,16 @@ int output_flex_t::app_as_geometrycollection()
     return 1;
 }
 
+static void check_tablespace(std::string const &tablespace)
+{
+    if (!has_tablespace(tablespace)) {
+        throw fmt_error(
+            "Tablespace '{0}' not available."
+            " Use 'CREATE TABLESPACE \"{0}\" ...;' to create it.",
+            tablespace);
+    }
+}
+
 flex_table_t &output_flex_t::create_flex_table()
 {
     std::string const table_name =
@@ -491,6 +501,7 @@ flex_table_t &output_flex_t::create_flex_table()
     if (lua_isstring(lua_state(), -1)) {
         std::string const tablespace = lua_tostring(lua_state(), -1);
         check_identifier(tablespace, "data_tablespace field");
+        check_tablespace(tablespace);
         new_table.set_data_tablespace(tablespace);
     }
     lua_pop(lua_state(), 1);
@@ -500,6 +511,7 @@ flex_table_t &output_flex_t::create_flex_table()
     if (lua_isstring(lua_state(), -1)) {
         std::string const tablespace = lua_tostring(lua_state(), -1);
         check_identifier(tablespace, "index_tablespace field");
+        check_tablespace(tablespace);
         new_table.set_index_tablespace(tablespace);
     }
     lua_pop(lua_state(), 1);