Implement const block inference

This commit is contained in:
Lukas Wirth 2020-12-23 12:24:24 +01:00
parent 0a780c0ab3
commit a142beaf01
5 changed files with 25 additions and 6 deletions

View file

@ -246,6 +246,10 @@ impl ExprCollector<'_> {
let body = self.collect_block_opt(e.block_expr()); let body = self.collect_block_opt(e.block_expr());
self.alloc_expr(Expr::Async { body }, syntax_ptr) self.alloc_expr(Expr::Async { body }, syntax_ptr)
} }
ast::Effect::Const(_) => {
let body = self.collect_block_opt(e.block_expr());
self.alloc_expr(Expr::Const { body }, syntax_ptr)
}
}, },
ast::Expr::BlockExpr(e) => self.collect_block(e), ast::Expr::BlockExpr(e) => self.collect_block(e),
ast::Expr::LoopExpr(e) => { ast::Expr::LoopExpr(e) => {

View file

@ -114,6 +114,9 @@ pub enum Expr {
Async { Async {
body: ExprId, body: ExprId,
}, },
Const {
body: ExprId,
},
Cast { Cast {
expr: ExprId, expr: ExprId,
type_ref: TypeRef, type_ref: TypeRef,
@ -253,7 +256,10 @@ impl Expr {
f(*expr); f(*expr);
} }
} }
Expr::TryBlock { body } | Expr::Unsafe { body } | Expr::Async { body } => f(*body), Expr::TryBlock { body }
| Expr::Unsafe { body }
| Expr::Async { body }
| Expr::Const { body } => f(*body),
Expr::Loop { body, .. } => f(*body), Expr::Loop { body, .. } => f(*body),
Expr::While { condition, body, .. } => { Expr::While { condition, body, .. } => {
f(*condition); f(*condition);

View file

@ -155,7 +155,7 @@ impl<'a> InferenceContext<'a> {
} }
None => self.infer_block(statements, *tail, expected), None => self.infer_block(statements, *tail, expected),
}, },
Expr::Unsafe { body } => self.infer_expr(*body, expected), Expr::Unsafe { body } | Expr::Const { body } => self.infer_expr(*body, expected),
Expr::TryBlock { body } => { Expr::TryBlock { body } => {
let _inner = self.infer_expr(*body, expected); let _inner = self.infer_expr(*body, expected);
// FIXME should be std::result::Result<{inner}, _> // FIXME should be std::result::Result<{inner}, _>

View file

@ -1894,6 +1894,7 @@ fn effects_smoke_test() {
let x = unsafe { 92 }; let x = unsafe { 92 };
let y = async { async { () }.await }; let y = async { async { () }.await };
let z = try { () }; let z = try { () };
let w = const { 92 };
let t = 'a: { 92 }; let t = 'a: { 92 };
} }
@ -1905,7 +1906,7 @@ fn effects_smoke_test() {
} }
"#, "#,
expect![[r#" expect![[r#"
16..136 '{ ...2 }; }': () 16..162 '{ ...2 }; }': ()
26..27 'x': i32 26..27 'x': i32
30..43 'unsafe { 92 }': i32 30..43 'unsafe { 92 }': i32
37..43 '{ 92 }': i32 37..43 '{ 92 }': i32
@ -1921,9 +1922,13 @@ fn effects_smoke_test() {
99..109 'try { () }': {unknown} 99..109 'try { () }': {unknown}
103..109 '{ () }': () 103..109 '{ () }': ()
105..107 '()': () 105..107 '()': ()
119..120 't': i32 119..120 'w': i32
127..133 '{ 92 }': i32 123..135 'const { 92 }': i32
129..131 '92': i32 129..135 '{ 92 }': i32
131..133 '92': i32
145..146 't': i32
153..159 '{ 92 }': i32
155..157 '92': i32
"#]], "#]],
) )
} }

View file

@ -358,6 +358,7 @@ pub enum Effect {
Async(SyntaxToken), Async(SyntaxToken),
Unsafe(SyntaxToken), Unsafe(SyntaxToken),
Try(SyntaxToken), Try(SyntaxToken),
Const(SyntaxToken),
// Very much not an effect, but we stuff it into this node anyway // Very much not an effect, but we stuff it into this node anyway
Label(ast::Label), Label(ast::Label),
} }
@ -373,6 +374,9 @@ impl ast::EffectExpr {
if let Some(token) = self.try_token() { if let Some(token) = self.try_token() {
return Effect::Try(token); return Effect::Try(token);
} }
if let Some(token) = self.const_token() {
return Effect::Const(token);
}
if let Some(label) = self.label() { if let Some(label) = self.label() {
return Effect::Label(label); return Effect::Label(label);
} }