可变长类型(Varlena)
Datum的typelen的约定
如果Datum类型是byval,则Datum表示一个值。
-
如果Datum类型不是byval,则Datum表示一个指针:
- typlen > 0, Datum 就指向固定长度字节流;
- typelen == -1, Datum 指向一个变长 varlena 结构体;
- typelen == -2, Datum 指向一个C语言风格的字符串;
因此,查看所有的变长数据类型:
SELECT typname FROM pg_type WHERE typlen = -1
Varlena数据类型
为了可以存储任意的数据,跳出CString中需要使用'\0'来作为终结符的弊端,PG采用了Varlena数据类型,优于不再有终结符表示数组的结束,所以必须存在一个长度字段只出当前的数组长度大小(Redis中的SDS也是类似的设计),以及是否经过了TOAST(是否压缩,是否行外存储等)。因此Varlena在数据开头引入了一个header。
/*
所有可变长度数据类型都将“ struct varlena”作为自己的header。
注意:对于TOASTable类型,varlena的表示太简单了,因为value可能会被压缩或移出行外存储。
* 但是,特定于数据类型的例程仅满足于处理de-TOASTed值,当然,客户端例程永远也不应看到TOASTed值。
* 但是,即使采用了de-TOASTed的值,仍然也不要直接使用vl_len_,因为它的并不直接表示长度(后30bit才是)
* 所以建议终使用宏 VARDATA_ANY ,VARSIZE_ANY ,VARSIZE_ANY_EXHDR ,VARDATA ,VARSIZE 和 SET_VARSIZE ,操作vl_len_字段
* 而不要直接引用结构字段!!
*/
struct varlena
{
// 头部的header,4字节,32bit
char vl_len_[4]; /* Do not touch this field directly!, */
char vl_dat[FLEXIBLE_ARRAY_MEMBER];/* Data content is here */
};
注意varlena
只是变长数据类型的基类,在具体使用中一般很少直接使用varlena类型。varlena
还分为很多种子类,每种格式的定义都不相同。我们在使用之前,需要根据它的第一个字节,转换为它对应格式:
- 第一个字节等于
000 00001
(小端序), 那么就是varattrib_1b_e
,用来存储和toast有关的 external 数据 - 第一个字节的最高位等于1,且后7bit不全为0,那么就是
varattrib_1b
,用来存储小数据 - 第一个字节的最高位等于0,那么就是
varattrib_4b
,可以存储不超过1GB的数据
另外需要注意的是,所有的Varlena类型中的vl_len_字段都是包括了header自身的大小,如果需要获得实际的数据大小需要减去header的长度。
varattrib_1b (small)
varattrib_1b
的结构如下所示:
typedef struct
{
uint8 va_header;
char va_data[FLEXIBLE_ARRAY_MEMBER]; /* Data begins here */
} varattrib_1b;
注意到varattrib_1b
的va_header
只有 8 位,最高位是标记位,值为 1。剩余的 7 位表示数据长度,所以varattrib_1b
类型只是用于存储长度不超过127 byte 的小数据,varattrib_1b
最常见的用法就是存放一个TOAST指针。varattrib_1b
的header
在小端序机器如下所示:
`varattrib_1b` header(8bit)
----------------------------------------
length(not zero) | tag |
-----------------------------------------
7 bit | 1 |
----------------------------------------
varattrib_4b(flat)
这种数据形式一般被称为“ flat”的形式,即没有经过TOAST机制的行外存储,也就是说其va_data
中存储就是实际的变长数据。
对于未压缩的数据,使用va_4byte
结构体存储,这是一个最原始的变长类型。观察va_4byte
对象,其内部的定义和一个原始的varlena结构体一模一样,所以varattrib_4b.va_4byte
是变长类型中唯一和TOAST没关系的类型。
对于压缩的数据,使用va_compressed
结构体存储,其内部存储了数据压缩之前的大小。varattrib_4b
使用union
来表示这两种情况。如下所示:
typedef union
{
// 这种数据形式一般被称为“ flat”的形式,即没有经过TOAST机制的行外存储,也就是说其存储仍然是连续的。
struct /* Normal varlena (4-byte length) */
{
uint32 va_header;
char va_data[FLEXIBLE_ARRAY_MEMBER];
} va_4byte;
struct /* Compressed-in-line format */
{
uint32 va_header;
uint32 va_rawsize;/* Original data size (excludes header) */
char va_data[FLEXIBLE_ARRAY_MEMBER]; /* Compressed data */
} va_compressed;
} varattrib_4b;
varattrib_4b
的头部va_header
是一个32bit大小的类型,其第一位为0,用于与varattrib_1b
类型进行区分,而其va_header
中的第二高位用于区分数据是否被压缩,为1,则表示存储的数据是压缩的。为0,则表示存储的数据是未压缩过的。剩下的 30 位表示数据的长度,所以只能支持不超过 1GB (2^30 - 1 bytes) 的数据。
varattrib_4b
的header在小端序机器如下所示:
--------------------------------------------------
length | compress | tag |
--------------------------------------------------
30 bit | 1 bit | 1 bit |
-------------------------------------------------
varattrib_1b_e (toast)
varattrib_1b_e
就是我们所说的“TOAST 指针”的父类,它并不存储数据,他的va_data
数据段存放的是TOAST指针,可以是三种不同类型: varatt_external
, varatt_indirect
, varatt_expanded
。这三种不同的TOAST指针具体在下一小节中会详细介绍。
varattrib_1b_e结构的首部header的第一个字节永远是 0x80(大端序) or 0x01 (小端序),在小端序机器如下所示:
`varattrib_1b_e ` header(8bit)
----------------------------------------
length(not zero) | tag |
-----------------------------------------
0000000 | 1 |
----------------------------------------
varattrib_1b_e
是 varattrib_1b
类型的一个子集,其和 varattrib_1b
类型的唯一区别就是多了一个 va_tag
字段,其可以指出在va_data
数据段中到底存放了哪一种的TOAST指针。同时需要注意的是varattrib_1b_e 类型和varattrib_1b
一样,内部的字段都是未对齐的(因为va_data是一个char数组),因此如果需要访问对应的va_data字段,只能使用memcpy的方法,将其va_data范围内的数据copy到varatt_external
, varatt_indirect
, varatt_expanded
结构体中,然后在对其进行访问,可以使用PG提供的宏VARATT_EXTERNAL_GET_POINTER
实现这一点。
/* TOAST pointers are a subset of varattrib_1b with an identifying tag byte
TOAST指针是varattrib_1b的子集,带有标识标签*/
typedef struct
{
uint8 va_header; /* Always 0x80(大端序) or 0x01 (小端序)*/
uint8 va_tag; /* Type of datum, 指出 va_data 域中TOAST指针的种类*/
char va_data[FLEXIBLE_ARRAY_MEMBER]; /* Type-specific data,存放三种不同的TOAST指针 */
} varattrib_1b_e;
// 第二个字节va_tag表示类型,有下面四种。每种类型下,它的 va_data存储的格式都不是一样的:
// 不同种“Toast 指针”的种类标签
typedef enum vartag_external
{
VARTAG_INDIRECT = 1, // 属于 varatt_indirect 类型的TOAST指针
VARTAG_EXPANDED_RO = 2, // 属于 ExpandedObjectHeader 类型的 只读指针
VARTAG_EXPANDED_RW = 3, // 属于 ExpandedObjectHeader 类型的 读写指针
VARTAG_ONDISK = 18 // 属于 varatt_external 类型的指针
} vartag_external;
varlena 类型 header 区分总结
总结下,在小端序的机器上,不同的varlena的头部可能有如下的情况:
- xxxxxx00 4-byte length word, aligned, uncompressed data (up to 1G)---------------------->varattrib_4b.va_4byte
- xxxxxx10 4-byte length word, aligned, compressed data (up to 1G) ------------------------> varattrib_4b.va_compressed
- 00000001 1-byte length word, unaligned, TOAST pointer--------------------------------------> varattrib_1b_e
- xxxxxxx1 1-byte length word, unaligned, uncompressed data (up to 126b)----------------> varattrib_1b
TOAST指针
刚才我们提到了varattrib_1b_e是所有TOAST指针的父类,而其下又有三种具体的,不同的类型TOAST指针,分别是varatt_external
, varatt_indirect
, varatt_expanded
。需要注意的是,这里的“TOAST pointer”并不是c语言意义上的指针,而是表示的是指出被Toasted的数据实际存放位置的结构体。
varatt_external(行外存储TOAST指针)
struct varatt_external
是一个传统的:TOAST指针“(也就在官方文档中提到的基于物理存储的TOAST指针)。 也就是说,其包含了行外存储在TOAST表中的Datum所需的信息。 仅当va_extsize
memcmp
比较这些结构体。
请再次注意,由于varatt_external
并未对齐地存储在实际的元组中,因此,在查看这些字段之前,需要将元组中的数据 memcpy 到本地struct变量中,然后才可以查看里面的字段! (我们之所以使用 memcmp ,是为了避免通过比较两个指针内部的字段值判断两个指针相等,而只检测两个TOAST指针(结构体)的值是否相等就可以了。)
/* TOAST指针的实现*/
typedef struct varatt_external
{
/*原始数据(未TOAST过的)大小(包含数据头部信息)*/
int32 va_rawsize; /* Original data size (includes header) */
/*行外存储(TOAST过)的数据大小(不包括头部)*/
int32 va_extsize; /* External saved size (doesn't) */
/*行外存储数据的OID,也是其在TOAST表的唯一标识*/
Oid va_valueid; /* Unique ID of value within TOAST table */
/*TOAST表的OID*/
Oid va_toastrelid; /* RelID of TOAST table containing it */
}varatt_external;
varatt_indirect(指向varlena的TOAST指针)
struct varatt_indirect
只是一个varlena
指针,可以指向varatt_external
,varatt_expanded
,或者是varattrib_1b
,varattrib_4b
类型的原始数据。其指向的必须是存储在内存中而不是行外磁盘存储的toast关系中的Datum。 创建者就需要完全负责被引用的空间的生存周期, 只要该引用Datum指针存在。
请注意,就像struct varatt_external一样,此结构未对齐地存储在任何包含元组中。
typedef struct varatt_indirect
{
struct varlena *pointer; /* Pointer to in-memory varlena */
} varatt_indirect;
varatt_expanded (内存扩展存储的TOAST指针)
struct varatt_expanded
是一个“ TOAST指针”,表示存储在内存中的行外数据, 采用某种特定于类型的,不一定物理连续的格式,便于计算而不是存储。 src/include/utils/expandeddatum.h
中提供了 ExpandedObjectHeader
类型的操作API。在下面的expandeddatum的章节中会专门介绍这种数据类型。
typedef struct ExpandedObjectHeader ExpandedObjectHeader;
typedef struct varatt_expanded
{
ExpandedObjectHeader *eohptr;
} varatt_expanded;
可变长类型的操作
上一小节介绍了三种可变长类型,在一般情况下,我们不能直接对这些结构体进行操作,因为数据并不是存放在一个栈上的结构体里面,这些结构体只是对存放的字节如何解释做出了定义,如果我们想要访问字段的值需要将Datum强制转换为可变长类型的指针,配合一系列的宏获取字段的值。
通用的宏
通用的宏:
VARSIZE_ANY(PTR) 返回任意varlena指针指向的可变对象的长度(包括header)
VARSIZE_ANY_EXHDR(PTR) 返回任意varlena指针指向的可变对象的数据长度(不包括header)
VARDATA_ANY(PTR) 返回任意varlena指针指向的可变对象的起始数据地址(不支持external or compressed-in-line Datum)
VARATT_IS_EXTENDED(PTR) 指针是否是扩展的类型,除了VARATT_IS_4B_U都是
设置varlena类型的长度
SET_VARSIZE(PTR, len)
SET_VARSIZE_SHORT(PTR, len)
SET_VARSIZE_COMPRESSED(PTR, len)
查看header的第一个字节:
- VARATT_IS_COMPRESSED(PTR) 数据是否是压缩的
类型相关的宏
varattrib_4b相关:
VARDATA(PTR) 获得varattrib_4b指针类型的数据起始地址
VARSIZE(PTR) 获得varattrib_4b指针类型指向的可变对象的长度(包括header)
varattrib_1b相关:
VARATT_IS_SHORT(PTR) 指针是否是 varattrib_1b 的
VARSIZE_SHORT(PTR) 同上
VARDATA_SHORT(PTR) 同上
varattrib_1b_e相关:
VARATT_IS_EXTERNAL(PTR) 指针是否是 varattrib_1b_e 的
SET_VARTAG_EXTERNAL(PTR, tag) 设置 varattrib_1b_e 指针类型的va_tag字段
VARTAG_EXTERNAL(PTR) 获得 varattrib_1b_e 指针类型的va_tag字段的起始地址
VARSIZE_EXTERNAL(PTR) 同上
VARDATA_EXTERNAL(PTR) 同上
VARATT_IS_EXTERNAL_ONDISK(PTR) 指针是否是 VARTAG_ONDISK
VARATT_IS_EXTERNAL_INDIRECT(PTR) 指针是否是 VARTAG_INDIRECT
VARATT_IS_EXTERNAL_EXPANDED_RO(PTR) 指针是否是VARTAG_EXPANDED_RO
VARATT_IS_EXTERNAL_EXPANDED_RW(PTR) 指针是否是VARTAG_EXPANDED_RW
内存expanded的数据类型
这部分的内容主要在expandedaatum.h
复杂的数据类型,尤其是诸如array
和record
之类的容器类型,通常在磁盘上具有紧凑的存储形式,但并利于修改。而且,当我们修改它们时可能会非常低效,因为我们不得不重新复制其余所有值。因此,PG提供了“扩展(expanded)”的概念,这一概念属于内存TOAST技术的一种,这种存储格式仅在内存中使用,内存扩展类型针对计算而非存储进行了更多优化。稍后我们会发现,Array
类型的expanded
结构是如何加速下标访问的。
我们将出现在磁盘上的格式称为数据类型的“扁平(flattened)”表示形式,flattened的存储格式是连续的字节blob(块)。但是该类型也可以具有expanded
表示形式用来加速内存中的计算,比如访问或者排序。如果一个数据类型支持expanded
的表示类型,其必须提供将expanded
的表示形式转换回flat
形式的方法。
PG中所有支持expanded
的数据结构都必须包含ExpandedObjectHeader
,其定义如下所示:
struct ExpandedObjectHeader
{
/* Phony varlena header Phony varlena标头 */
int32 vl_len_; /* 对于 ExpandedObjectHeader 对象来说,其vl_len域永远是 -1 */
const ExpandedObjectMethods *eoh_methods; // 扩展对象需要实现函数指针结构体,一个是获取flat格式方法,一个是获取flat size的方法
MemoryContext eoh_context; // 包含此 header 和 辅助数据 的内存上下文
char eoh_rw_ptr[EXPANDED_POINTER_SIZE]; // 读写指针(TOAST指针结构体)
char eoh_ro_ptr[EXPANDED_POINTER_SIZE]; // 只读指针(TOAST指针结构体)
};
ExpandedObjectMethods
ExpandedObjectMethods中定义了两个需要编码数据结构的程序员实现的方法。所有支持expand的数据类型都需要实现ExpandedObjectMethods中给出的两个方法,flatten_into
用于在detoast
的时候将一个expand
表示转换为flat
的表示。而get_flat_size
是方便获得一个expand
表示展开成flat
表示后的大小。
/* Struct of function pointers for an expanded object's methods */
typedef struct ExpandedObjectMethods
{
EOM_get_flat_size_method get_flat_size;
EOM_flatten_into_method flatten_into;
} ExpandedObjectMethods;
如何判断一个varlena类型是不是ExpandedObjectHeader
PG在设计ExpandedObjectHeader时,考虑到了对于只读函数,如果既能够处理同一种数据类型常规的 ”flat“ 的varlena输入(即varattrib_4b
类型),也能够处理其扩展的ExpandedObjectHeader 的输入,这是十分方便的。因此为了使得函数确定输入的varlena指针到底指向的是哪一种类型, ExpandedObjectHeader 的第一个int32始终是-1(定义为宏:EOH_HEADER_MAGIC
)。 -1
的二进制表示为1111 11111
,其不会和varattrib_4b
的header冲突。
这一判断方法被宏:VARATT_IS_EXPANDED_HEADER(PTR)
所封装,其返回true表示输入的指针指向的是一个ExpandedObjectHeader对象。
举个例子来说,在Array类型的实现中,Array类型的编码人员设计了一个名为AnyArrayType
的联合体,包含了这两种不同的array varlena类型。就简化了代码的处理逻辑,只需要以宏AARR_XXX开头的宏就可以同时处理这两种数据类型。
typedef union AnyArrayType
{
ArrayType flt; // flat格式的array类型
ExpandedArrayHeader xpn; // expand格式的array类型
} AnyArrayType;
/** Macros for working with AnyArrayType inputs. Beware multiple references!
为了篇幅删去了具体宏定义*/
#define AARR_NDIM(a)
#define AARR_HASNULL(a)
#define AARR_ELEMTYPE(a)
#define AARR_DIMS(a)
#define AARR_LBOUND(a)
ExpandedObjectHeader 和 varatt_expanded 的关系
我们之前在TOAST 指针中介绍过最后一种expand类型的TOAST指针varatt_expanded
结构,其内部只有一个ExpandedObjectHeader指针。而varatt_expanded
结构又是存放在结构体varattrib_1b_e
的va_data字段。也就是说函数传入的参数一般是一个varattrib_1b_e
指针(一个Datum)。
所以为了获取实际的ExpandedObjectHeader指针,我们首先把Datum强制转换为(varattrib_1b_e *)
指针,然后再利用宏VARDATA_EXTERNAL
确定varattrib_1b_e 结构中va_data域(指针域)的位置,然后使用memcpy将其拷贝到varatt_expanded结构体中,然后再从中提取出(ExpandedObjectHeader *
)指针。参加如下的函数:
// expandeddatu.c
// 给定一个作为expanded对象引用的Datum,在将其转化为varatt_indirect后返回其内部的ExpandedObjectHeader指针
ExpandedObjectHeader *
DatumGetEOHP(Datum d)
{
varattrib_1b_e *datum = (varattrib_1b_e *) DatumGetPointer(d);
varatt_expanded ptr;
Assert(VARATT_IS_EXTERNAL_EXPANDED(datum));
memcpy(&ptr, VARDATA_EXTERNAL(datum), sizeof(ptr)); // dest <=== src
Assert(VARATT_IS_EXPANDED_HEADER(ptr.eohptr));
return ptr.eohptr;
}