fix FilterBy func

github · Dec 18, 2024 · 6c2a036 · 6c2a036
1 parent c041017
commit 6c2a036
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/go/sql/types.go b/go/sql/types.go
@@ -254,6 +254,7 @@ func (this *ColumnList) FilterBy(f func(Column) bool) *ColumnList {
 			filteredCols = append(filteredCols, column)
 		}
 	}
+	filteredList.columns = filteredCols
 	return filteredList
 }
 

diff --git a/localtests/generated-columns/create.sql b/localtests/generated-columns/create.sql
@@ -27,4 +27,6 @@ begin
   insert into gh_ost_test (id, a, b) values (null, 2,0);
   insert into gh_ost_test (id, a, b) values (null, 2,1);
   insert into gh_ost_test (id, a, b) values (null, 2,2);
+  update gh_ost_test set b=b+1 where id < 5;
+  update gh_ost_test set b=b-1 where id >= 5;
 end ;;
-Original file line number
+Diff line change
@@ Expand Up @@
     			filteredCols = append(filteredCols, column)
     		}
     	}
+    	filteredList.columns = filteredCols
     	return filteredList
     }
@@ Expand Down @@