Commit d53edb4

committed

feat: modify calculate response size logic for mongoengine queryset

Signed-off-by: ImMin5 <mino@megazone.com>

1 parent e956e34 commit d53edb4Copy full SHA for d53edb4

File tree

+12

-4

lines changed

+12

-4

lines changed

Lines changed: 12 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -271,12 +271,20 @@ def _pipeline(`
`271`	`271`
`272`	`272`	`def _get_response_size(response_or_iterator: Any) -> int:`
`273`	`273`	`try:`
`274`		`- if isinstance(response_or_iterator, dict):`
	`274`	`+ if response_or_iterator is None:`
	`275`	`+ return 0`
	`276`	`+`
	`277`	`+ if isinstance(response_or_iterator, tuple):`
	`278`	`+ response_or_iterator = response_or_iterator[0]`
	`279`	`+`
	`280`	`+ if isinstance(response_or_iterator, (dict, list)):`
`275`	`281`	`response_size = len(json.dumps(response_or_iterator, ensure_ascii=False))`
`276`		`- elif isinstance(response_or_iterator, (bytes, bytearray)):`
	`282`	`+ elif isinstance(response_or_iterator, (bytes, bytearray, str)):`
`277`	`283`	`response_size = len(response_or_iterator)`
`278`		`- elif response_or_iterator is None:`
`279`		`- response_size = 0`
	`284`	`+ elif hasattr(response_or_iterator, "to_json"):`
	`285`	`+ response_size = len(response_or_iterator.to_json())`
	`286`	`+ elif hasattr(response_or_iterator, "__dict__"):`
	`287`	`+ response_size = len(json.dumps(response_or_iterator, ensure_ascii=False))`
`280`	`288`	`else:`
`281`	`289`	`response_size = -1`
`282`	`290`	`except Exception:`

Comments

(0)