• 用户自定义变量

    用户自定义变量

    用户自定义变量格式为 @var_name。var_name 目前只支持字母，数字，_$组成。用户自定义变量是大小写不敏感的。

    用户自定义变量是跟 session 绑定的，也就是说只有当前连接可以看见设置的用户变量，其他客户端连接无法查看到。

    用 SET 语句可以设置用户自定义变量：

    1. SET @var_name = expr [, @var_name = expr] ...
    2. 或 
    3. SET @var_name := expr

    对于 SET 语句，赋值操作符可以是 = 也可以是 :=

    例：

    1. mysql> SET @a1=1, @a2=2, @a3:=4;
    2. mysql> SELECT @a1, @a2, @t3, @a4 := @a1+@a2+@a3;
    3. +------+------+------+--------------------+
    4. | @a1  | @a2  | @a3  | @a4 := @a1+@a2+@a3 |
    5. +------+------+------+--------------------+
    6. |    1 |    2 |    4 |                  7 |
    7. +------+------+------+--------------------+

    如果设置用户变量用了 HEX 或者 BIT 值，TiDB会把它当成二进制字符串。如果你要将其设置成数字，那么需要手动加上 CAST转换: CAST(.. AS UNSIGNED)：

    1. mysql> SELECT @v1, @v2, @v3;
    2. +------+------+------+
    3. | @v1  | @v2  | @v3  |
    4. +------+------+------+
    5. | A    | 65   | 65   |
    6. +------+------+------+
    7. 1 row in set (0.00 sec)
    9. mysql> SET @v1 = b'1000001';
    10. Query OK, 0 rows affected (0.00 sec)
    12. mysql> SET @v2 = b'1000001'+0;
    13. Query OK, 0 rows affected (0.00 sec)
    15. mysql> SET @v3 = CAST(b'1000001' AS UNSIGNED);
    16. Query OK, 0 rows affected (0.00 sec)
    18. mysql> SELECT @v1, @v2, @v3;
    19. +------+------+------+
    20. | @v1  | @v2  | @v3  |
    21. +------+------+------+
    22. | A    | 65   | 65   |
    23. +------+------+------+
    24. 1 row in set (0.00 sec)

    如果获取一个没有设置过的变量，会返回一个 NULL：

    1. mysql> select @not_exist;
    2. +------------+
    3. | @not_exist |
    4. +------------+
    5. | NULL       |
    6. +------------+
    7. 1 row in set (0.00 sec)

    用户自定义变量不能直接在 SQL 语句中被当成 identifier，例：

    1. mysql> select * from t;
    2. +------+
    3. | a    |
    4. +------+
    5. |    1 |
    6. +------+
    7. 1 row in set (0.00 sec)
    9. mysql> SET @col = "a";
    10. Query OK, 0 rows affected (0.00 sec)
    12. mysql> SELECT @col FROM t;
    13. +------+
    14. | @col |
    15. +------+
    16. | a    |
    17. +------+
    18. 1 row in set (0.00 sec)
    20. mysql> SELECT `@col` FROM t;
    21. ERROR 1054 (42S22): Unknown column '@col' in 'field list'
    23. mysql> SET @col = "`a`";
    24. Query OK, 0 rows affected (0.00 sec)
    26. mysql> SELECT @col FROM t;
    27. +------+
    28. | @col |
    29. +------+
    30. | `a`  |
    31. +------+
    32. 1 row in set (0.01 sec)

    但是有一个例外是如果你在 PREPARE 语句中使用它，是可以的：

    1. mysql> PREPARE stmt FROM "SELECT @c FROM t";
    2. Query OK, 0 rows affected (0.00 sec)
    4. mysql> EXECUTE stmt;
    5. +------+
    6. | @c   |
    7. +------+
    8. | a    |
    9. +------+
    10. 1 row in set (0.01 sec)
    12. mysql> DEALLOCATE PREPARE stmt;
    13. Query OK, 0 rows affected (0.00 sec)

    更多细节。