PostgreSQL 源码解读 node的模拟实现
node的实现是PostgreSQL的查询解析的基础，实现的关键是两个宏，makeNode和newNode。其他节点继承自Node节点，如果增加新的结构体，需要添加NodeTag中添加对应的枚举值，并在equal和nodetoString中添加对于的处理代码。当结构体少是很容易处理，如果结构体过多，维护会比较麻烦。PostgreSQL中大约有300个继承自node的结构体，写代码的人真是需要相当的勇气和毅力呀。
#include <iostream>
#include <string.h>
#include <assert.h>
#include <stdlib.h>
#include <stdio.h>
using namespace std;
enum NodeTag
{
T_Stmt,
T_Value
};
typedef struct Node
{
NodeTag type;
}Node;
Node *newNodeMacroHolder ;
#define newNode(size, tag) \
( \
assert((size) >= sizeof(Node)), /* need the tag, at least */ \
newNodeMacroHolder = (Node *) malloc(size), \
newNodeMacroHolder->type = (tag), \
newNodeMacroHolder \
)
#define makeNode(_type_) ((_type_ *)newNode(sizeof(_type_),T_##_type_))
#define nodeTag(nodeptr) (((const Node *)(nodeptr))->type)
typedef struct Stmt
{
NodeTag type;
char *text;
}Stmt;
typedef struct Value
{
NodeTag type;
long val;
}Value;
bool equal(void *a,void *b)
{
if(a == b)
return true;
if (a == NULL || b == NULL)
return false;
if(nodeTag(a) != nodeTag(b))
return false;
switch(nodeTag(a)){
case T_Stmt:
return strcmp(((const Stmt*)a)->text,((const Stmt *)b)->text)==0? true:false;
case T_Value:
return ((const Value *)a)->val==((const Value *)b)->val;
default:
cout<<"error:unknown type"<<endl;
}
return false;
}
char * nodetoString(void *obj)
{
char *r =(char *)malloc(1024);
if (obj == NULL){
strcpy(r,"<>");
}
switch(nodeTag(obj)){
case T_Stmt:
sprintf(r,"<Stmt:%s>",((const Stmt *)obj)->text);
break;
case T_Value:
sprintf(r,"<Value:%ld>",((const Value *)obj)->val);
break;
default:
strcpy(r,"<unknown node type>");
}
return r;
}
int main(int argc,char *argv[])
{
Stmt *s= makeNode(Stmt);
if(s){
char str[]="select * from a";
s->text=str;
}
Stmt *t= makeNode(Stmt);
if(t){
char str[]="select * from b";
t->text=str;
}
Value *v=makeNode(Value);
if(v){
v->val=100;
}
cout<<"t->text:"<<t->text<<endl;
cout<<"equal:"<<equal(s,t)<<endl;
cout<<nodetoString(t)<<endl;
free(s);
free(t);
free(v);
return 0;
}
如果准备使用C语言来实现node结构体的话，尤其是准备采用gcc编译的话，要注意将newNode的宏设置为
/* 针对gcc版本的newNode */
#define newNode(size, tag) \
({ Node *_result; \
AssertMacro((size) >= sizeof(Node));/* 检测申请的内存大小，>>=sizeof(Node) */ \
_result = (Node *) palloc0fast(size); /* 申请内存 */ \
_result->type = (tag); /*设置TypeTag */ \
_result; /*返回值*/\
})
参见我的另一篇笔记《PostgreSQL源码解读 基础结构 node》
