用户自定义变量

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

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

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

  1. SET @var_name = expr [, @var_name = expr] ...;

  1. SET @var_name := expr;

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

例:

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

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

  1. SELECT @v1, @v2, @v3;
  1. +------+------+------+
  2. | @v1 | @v2 | @v3 |
  3. +------+------+------+
  4. | A | 65 | 65 |
  5. +------+------+------+
  6. 1 row in set (0.00 sec)
  1. SET @v1 = b'1000001';
  1. Query OK, 0 rows affected (0.00 sec)
  1. SET @v2 = b'1000001'+0;
  1. Query OK, 0 rows affected (0.00 sec)
  1. SET @v3 = CAST(b'1000001' AS UNSIGNED);
  1. Query OK, 0 rows affected (0.00 sec)
  1. SELECT @v1, @v2, @v3;
  1. +------+------+------+
  2. | @v1 | @v2 | @v3 |
  3. +------+------+------+
  4. | A | 65 | 65 |
  5. +------+------+------+
  6. 1 row in set (0.00 sec)

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

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

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

  1. select * from t;
  1. +------+
  2. | a |
  3. +------+
  4. | 1 |
  5. +------+
  6. 1 row in set (0.00 sec)
  1. SET @col = "a";
  1. Query OK, 0 rows affected (0.00 sec)
  1. SELECT @col FROM t;
  1. +------+
  2. | @col |
  3. +------+
  4. | a |
  5. +------+
  6. 1 row in set (0.00 sec)
  1. SELECT `@col` FROM t;
  1. ERROR 1054 (42S22): Unknown column '@col' in 'field list'
  1. SET @col = "`a`";
  1. Query OK, 0 rows affected (0.00 sec)
  1. SELECT @col FROM t;
  1. +------+
  2. | @col |
  3. +------+
  4. | `a` |
  5. +------+
  6. 1 row in set (0.01 sec)

但是在以下情况中,你可以在 PREPARE 语句中使用用户自定义变量:

  1. PREPARE stmt FROM "SELECT @c FROM t";
  1. Query OK, 0 rows affected (0.00 sec)
  1. EXECUTE stmt;
  1. +------+
  2. | @c |
  3. +------+
  4. | a |
  5. +------+
  6. 1 row in set (0.01 sec)
  1. DEALLOCATE PREPARE stmt;
  1. Query OK, 0 rows affected (0.00 sec)

更多细节