ColumnTypeLength method return the type length for varbit type.

DengChan · oushu1dengxingchen1 · commit 5736d0976ddd · 2025-12-21T23:46:42.000+08:00
diff --git a/stdlib/sql.go b/stdlib/sql.go
@@ -676,6 +676,8 @@ func (r *Rows) ColumnTypeLength(index int) (int64, bool) {
 		return math.MaxInt64, true
 	case pgtype.VarcharOID, pgtype.BPCharArrayOID:
 		return int64(fd.TypeModifier - varHeaderSize), true
+	case pgtype.VarbitOID:
+		return int64(fd.TypeModifier), true	
 	default:
 		return 0, false
 	}
diff --git a/stdlib/sql_test.go b/stdlib/sql_test.go
@@ -1392,3 +1392,67 @@ func TestOptionShouldPing_HookCalledOnReuse(t *testing.T) {
 
 	require.True(t, hookCalled, "hook should be called on reuse")
 }
+
+func TestRowsColumnTypeLength(t *testing.T) {
+	testWithAllQueryExecModes(t, func(t *testing.T, db *sql.DB) {
+		columnTypeLengthTests := []struct {
+			Len int64
+			OK  bool
+		}{
+			{
+				math.MaxInt64,
+				true,
+			},
+			{
+				math.MaxInt64,
+				true,
+			},
+			{
+				255,
+				true,
+			},
+			{
+				10,
+				true,
+			},
+			{
+				50,
+				true,
+			},
+			{
+				0,
+				false,
+			},
+		}
+
+		_, err := db.Exec(`CREATE TEMPORARY TABLE temp_column_type_length (
+						   text_column TEXT,
+						   bytea_column BYTEA,
+						   varchar_column VARCHAR(255),
+						   bpcharA_column BPCHAR(10)[],
+						   varbit_column VARBIT(50),
+						   int_column INT
+					);`)
+		require.NoError(t, err)
+
+		rows, err := db.Query("SELECT * FROM temp_column_type_length")
+		require.NoError(t, err)
+
+		columns, err := rows.ColumnTypes()
+		require.NoError(t, err)
+		assert.Len(t, columns, 6)
+
+		for i, tt := range columnTypeLengthTests {
+			c := columns[i]
+
+			l, ok := c.Length()
+			if l != tt.Len {
+				t.Errorf("(%d) got: %d, want: %d", i, l, tt.Len)
+			}
+			if ok != tt.OK {
+				t.Errorf("(%d) got: %t, want: %t", i, ok, tt.OK)
+			}
+
+		}
+	})
+}

Original file line number	Diff line number	Diff line change
`@@ -676,6 +676,8 @@ func (r *Rows) ColumnTypeLength(index int) (int64, bool) {`
`676`	`676`	`return math.MaxInt64, true`
`677`	`677`	`case pgtype.VarcharOID, pgtype.BPCharArrayOID:`
`678`	`678`	`return int64(fd.TypeModifier - varHeaderSize), true`
	`679`	`+ case pgtype.VarbitOID:`
	`680`	`+ return int64(fd.TypeModifier), true`
`679`	`681`	`default:`
`680`	`682`	`return 0, false`
`681`	`683`	`}`