feature request: return affectedRows for COPY command

[skyqrose]

I'm using a COPY command to load a csv, and want to know how many rows were added. Postgres does return this data (Postgres docs, see the Outputs section), but I can't access it via the PGlite API.

The query API Result includes a affectedRows field docs, but only for INSERT, UPDATE, and DELETE commands. (Side note: those docs could use a description of when the field is populated.)

I believe that this should be a simple addition, just adding a case to retrieveRowCount() for COPY.

My workaround for now is to make a separate SELECT count(*) query after doing the COPY.

[samwillis]

Thanks for the report, quite right thats where to add it. This would be a great first contribution for someone, it's just an additional CASE on the switch branch along with UPDATE/DELETE:

pglite/packages/pglite/src/parse.ts

Lines 85 to 96 in 9aff673

	function retrieveRowCount(msg: CommandCompleteMessage): number {
	const parts = msg.text.split(' ')
	switch (parts[0]) {
	case 'INSERT':
	return parseInt(parts[2], 10)
	case 'UPDATE':
	case 'DELETE':
	return parseInt(parts[1], 10)
	default:
	return 0
	}
	}