From: hongzhidao <hongzhidao@gmail.com>
Date: Thu, 18 Apr 2019 16:28:31 +0000 (+0800)
Subject: Refactored variable reference.
X-Git-Tag: 0.3.2~64
X-Git-Url: http://www.kaiwu.me/postgresql/commit/?a=commitdiff_plain;h=caa6a8b9b12656f0e18846bb8264034473f8f8c2;p=njs.git

Refactored variable reference.
---

diff --git a/njs/njs_generator.c b/njs/njs_generator.c
index 12ebaf79..78a8a142 100644
--- a/njs/njs_generator.c
+++ b/njs/njs_generator.c
@@ -1648,13 +1648,11 @@ njs_generate_assignment(njs_vm_t *vm, njs_generator_t *generator,
 
     if (lvalue->token == NJS_TOKEN_NAME) {
 
-        index = njs_variable_index(vm, lvalue);
-        if (nxt_slow_path(index == NJS_INDEX_ERROR)) {
-            return NXT_ERROR;
+        ret = njs_generate_variable(vm, generator, lvalue);
+        if (nxt_slow_path(ret != NXT_OK)) {
+            return ret;
         }
 
-        lvalue->index = index;
-
         expr->dest = lvalue;
 
         ret = njs_generator(vm, generator, expr);
@@ -1757,6 +1755,7 @@ njs_generate_operation_assignment(njs_vm_t *vm, njs_generator_t *generator,
     lvalue = node->left;
 
     if (lvalue->token == NJS_TOKEN_NAME) {
+
         ret = njs_generate_variable(vm, generator, lvalue);
         if (nxt_slow_path(ret != NXT_OK)) {
             return ret;
diff --git a/njs/njs_parser.c b/njs/njs_parser.c
index 521e0a53..e40c7478 100644
--- a/njs/njs_parser.c
+++ b/njs/njs_parser.c
@@ -522,6 +522,19 @@ njs_parser_variable_node(njs_vm_t *vm, njs_parser_t *parser, nxt_str_t *name,
         return NULL;
     }
 
+    if (njs_is_null(&var->value)) {
+
+        switch (type) {
+
+        case NJS_VARIABLE_VAR:
+            var->value = njs_value_undefined;
+            break;
+
+        default:
+            break;
+        }
+    }
+
     node = njs_parser_node_new(vm, parser, NJS_TOKEN_NAME);
     if (nxt_slow_path(node == NULL)) {
         return NULL;
@@ -1389,7 +1402,7 @@ njs_parser_for_statement(njs_vm_t *vm, njs_parser_t *parser)
             init = parser->node;
 
             if (init->token == NJS_TOKEN_FOR_IN) {
-                return token;
+                goto done;
             }
 
         } else {
@@ -1404,7 +1417,12 @@ njs_parser_for_statement(njs_vm_t *vm, njs_parser_t *parser)
             init = parser->node;
 
             if (init->token == NJS_TOKEN_IN) {
-                return njs_parser_for_in_statement(vm, parser, &name, token);
+                token = njs_parser_for_in_statement(vm, parser, &name, token);
+                if (nxt_slow_path(token <= NJS_TOKEN_ILLEGAL)) {
+                    return token;
+                }
+
+                goto done;
             }
         }
     }
@@ -1475,6 +1493,8 @@ njs_parser_for_statement(njs_vm_t *vm, njs_parser_t *parser)
 
     parser->node = node;
 
+done:
+
     return token;
 }
 
diff --git a/njs/njs_variable.c b/njs/njs_variable.c
index 7eaa3597..d1ce5383 100644
--- a/njs/njs_variable.c
+++ b/njs/njs_variable.c
@@ -360,7 +360,6 @@ njs_variable_resolve(njs_vm_t *vm, njs_parser_node_t *node)
     nxt_uint_t                scope_index;
     njs_index_t               index;
     njs_variable_t            *var;
-    const njs_value_t         *default_value;
     njs_variable_reference_t  *vr;
 
     vr = &node->u.reference;
@@ -398,14 +397,7 @@ njs_variable_resolve(njs_vm_t *vm, njs_parser_node_t *node)
         var->argument = index;
     }
 
-    if (vr->type != NJS_DECLARATION && var->type <= NJS_VARIABLE_LET) {
-        goto not_found;
-    }
-
-    default_value = njs_is_object(&var->value) ? &var->value :
-                                                 &njs_value_undefined;
-
-    index = njs_scope_next_index(vm, vr->scope, scope_index, default_value);
+    index = njs_scope_next_index(vm, vr->scope, scope_index, &var->value);
 
     if (nxt_slow_path(index == NJS_INDEX_ERROR)) {
         return NULL;