Fast-path with str.is_ascii()

coolreader18 · youknowone · coolreader18 · commit ab04157b96f6 · 2021-09-23T10:02:16.000-05:00
Co-authored-by: Jeong YunWon &lt;youknowone@users.noreply.github.com&gt;
diff --git a/vm/src/stdlib/codecs.rs b/vm/src/stdlib/codecs.rs
@@ -317,6 +317,9 @@ mod _codecs {
 
     #[pyfunction]
     fn ascii_encode(args: EncodeArgs, vm: &VirtualMachine) -> EncodeResult {
+        if args.s.is_ascii() {
+            return Ok(args.s.as_bytes());
+        }
         do_codec!(ascii::encode, args, vm)
     }
 

Original file line number	Diff line number	Diff line change
`@@ -317,6 +317,9 @@ mod _codecs {`
`317`	`317`
`318`	`318`	`#[pyfunction]`
`319`	`319`	`fn ascii_encode(args: EncodeArgs, vm: &VirtualMachine) -> EncodeResult {`
	`320`	`+ if args.s.is_ascii() {`
	`321`	`+ return Ok(args.s.as_bytes());`
	`322`	`+ }`
`320`	`323`	`do_codec!(ascii::encode, args, vm)`
`321`	`324`	`}`
`322`	`325`