Merge pull request #42 from goccy/fix-temp-table

Fix drop table query for create temp table

exec_test.go

@@ -119,3 +119,26 @@ COMMIT TRANSACTION;
 		})
 	}
 }
+
+func TestCreateTempTable(t *testing.T) {
+	now := time.Now()
+	ctx := context.Background()
+	ctx = zetasqlite.WithCurrentTime(ctx, now)
+	db, err := sql.Open("zetasqlite", ":memory:")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+	if _, err := db.ExecContext(ctx, "CREATE TEMP TABLE tmp_table (id INT64)"); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := db.ExecContext(ctx, "CREATE TEMP TABLE tmp_table (id INT64)"); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := db.ExecContext(ctx, "CREATE TABLE tmp_table (id INT64)"); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := db.ExecContext(ctx, "CREATE TABLE tmp_table (id INT64)"); err == nil {
+		t.Fatal("expected error")
+	}
+}

internal/analyzer.go

@@ -279,7 +279,7 @@ func (it *AnalyzerOutputIterator) analyzeCreateTableStmt(ctx context.Context, no
 		return nil, err
 	}
 	execContext := func(ctx context.Context, conn *Conn) (driver.Result, error) {
-		dropTableQuery := fmt.Sprintf("DROP TABLE IF EXISTS `%s`", spec.TableName())
+		dropTableQuery := fmt.Sprintf("DROP TABLE IF EXISTS `%s`", strings.Join(spec.NamePath, "."))
 		if spec.CreateMode == ast.CreateOrReplaceMode {
 			if _, err := conn.ExecContext(ctx, dropTableQuery); err != nil {
 				return nil, err
@@ -343,7 +343,7 @@ func (it *AnalyzerOutputIterator) analyzeCreateTableAsSelectStmt(ctx context.Con
 		return nil, err
 	}
 	execContext := func(ctx context.Context, conn *Conn) (driver.Result, error) {
-		dropTableQuery := fmt.Sprintf("DROP TABLE IF EXISTS `%s`", spec.TableName())
+		dropTableQuery := fmt.Sprintf("DROP TABLE IF EXISTS `%s`", strings.Join(spec.NamePath, "."))
 		if spec.CreateMode == ast.CreateOrReplaceMode {
 			if _, err := conn.ExecContext(ctx, dropTableQuery); err != nil {
 				return nil, err