MySQL动态字符串处理DYNAMIC_STRING
MySQL中，常常会看到一些关于动态字符串的处理，列如：DYNAMIC_STRING。
为了记录动态字符串的实际长度，缓冲区的最大长度，以及每次字符串需要调整时，及时分配新的内存，以及调整长度。MySQL使用了DYNAMIC_STRING来保存动态字符串相关的信息:
typedefstructst_dynamic_string
{
char*str;
size_tlength,max_length,alloc_increment;
}DYNAMIC_STRING;
在这个结构体中，str存储实际字符串的首地址，length记录字符串的实际长度，max_length记录字符串缓冲区最多可以存放多少字符，alloc_increment表示当字符串需要分配内存时，每次分配多少内存。
下面看看这个结构体的初始化过程:
my_boolinit_dynamic_string(DYNAMIC_STRING*str,constchar*init_str,size_tinit_alloc,size_talloc_increment)
{
size_tlength;
DBUG_ENTER("init_dynamic_string");
if(!alloc_increment)
alloc_increment=128;
length=1;
if(init_str&&(length=strlen(init_str)+1)<init_alloc)
init_alloc=((length+alloc_increment-1)/alloc_increment)*alloc_increment;
if(!init_alloc)
init_alloc=alloc_increment;
if(!(str->str=(char*)my_malloc(init_alloc,MYF(MY_WME))))
DBUG_RETURN(TRUE);
str->length=length-1;
if(init_str)
memcpy(str->str,init_str,length);
str->max_length=init_alloc;
str->alloc_increment=alloc_increment;
DBUG_RETURN(FALSE);
}
从上述函数可以看到，初始化时，初始分配的字符串缓冲区大小init_alloc会根据需要初始的字符串来做判断。在分配好该DYNAMIC_STRING空间之后，我们会根据缓冲区的大小，字符串的实际长度，以及alloc_increment来初始化：
length:字符串的实际长度
max_length:缓冲区的最大长度
alloc_increment:空间不够时，下次分配内存的单元大小.
初始化这些内容之后，如果下次需要在该缓冲区添加更多字符，就可以根据这些值来判断是否需要对该缓冲区扩容：
my_booldynstr_append_mem(DYNAMIC_STRING*str,constchar*append,size_tlength)
{
char*new_ptr;
if(str->length+length>=str->max_length)/*如果新增字符串后，总长度超过缓冲区大小*/
{
/*需要分配多少个alloc_increment大小的内存，才能存下新增后的字符串*/
size_tnew_length=(str->length+length+str->alloc_increment)/
str->alloc_increment;
new_length*=str->alloc_increment;
if(!(new_ptr=(char*)my_realloc(str->str,new_length,MYF(MY_WME))))
return(TRUE);
str->str=new_ptr;
str->max_length=new_length;
}
/*将新分配的内容，append到str之后*/
memcpy(str->str+str->length,append,length);
str->length+=length;/*扩容之后str新的长度*/
str->str[str->length]=0;/*SafetyforCprograms*//*字符串最后一个字符为'\0'*/
return(FALSE);
}
从上述代码可以看到，在字符串初始化化好之后，之后如果需要给该字符串增加新的内容，只需要根据之前存储的信息来动态的realloc就好了。由于该结构体记录了字符串相关的完整内容，所以动态的扩容会非常方便处理。
当然，除了这些，还有比如字符串截断，字符串初始设置，转义OS的引号等等：
将字符串偏移大于N之后的截断。
my_booldynstr_trunc(DYNAMIC_STRING*str,size_tn)
{
str->length-=n;
str->str[str->length]='\0';
return(FALSE);
}
返回字符串中第一次出现某个字符的地址。若没有，则返回字符串结尾的地址(指向'')
char*strcend(registerconstchar*s,registerpcharc)
{
for(;;)
{
if(*s==(char)c)
return((char*)s);
if(!*s++)
return((char*)s-1);
}
}
字符串内容扩容:
my_booldynstr_realloc(DYNAMIC_STRING*str,size_tadditional_size)
{
DBUG_ENTER("dynstr_realloc");
if(!additional_size)
DBUG_RETURN(FALSE);
if(str->length+additional_size>str->max_length)/*如果新的字符串内容超过缓冲区的最大长度*/
{
str->max_length=((str->length+additional_size+str->alloc_increment-1)/
str->alloc_increment)*str->alloc_increment;
if(!(str->str=(char*)my_realloc(str->str,str->max_length,MYF(MY_WME))))
DBUG_RETURN(TRUE);
}
DBUG_RETURN(FALSE);
}
对字符串用引号括起来，对其中的单引号进行转义,主要用于执行一些系统命令(system(cmd))。
比如:ls-al会变成‘ls-al'
比如:ls-a'l会变成'ls-a\'l'
/*
*Concatenatesanynumberofstrings,escapesanyOSquoteintheresultthen
*surroundthewholeaffairinanothersetofquoteswhichisfinallyappended
*tospecifiedDYNAMIC_STRING.Thisfunctionisespeciallyusefulwhen
*buildingstringstobeexecutedwiththesystem()function.
*
*@paramstrDynamicStringwhichwillhaveaddtionalstringsappended.
*@paramappendStringtobeappended.
*@param...Optional.Additionalstring(s)tobeappended.
*
*@noteThefinalargumentinthelistmustbeNullSevenifnoadditional
*optionsarepassed.
*
*@returnTrue=Success.
*/
my_booldynstr_append_os_quoted(DYNAMIC_STRING*str,constchar*append,...)
{
constchar*quote_str="\'";
constuintquote_len=1;
my_boolret=TRUE;
va_listdirty_text;
ret&=dynstr_append_mem(str,quote_str,quote_len);/*Leadingquote*/
va_start(dirty_text,append);
while(append!=NullS)
{
constchar*cur_pos=append;
constchar*next_pos=cur_pos;
/*Searchforquoteineachstringandreplacewithescapedquote*/
while(*(next_pos=strcend(cur_pos,quote_str[0]))!='\0')
{
ret&=dynstr_append_mem(str,cur_pos,(uint)(next_pos-cur_pos));
ret&=dynstr_append_mem(str,"\\",1);
ret&=dynstr_append_mem(str,quote_str,quote_len);
cur_pos=next_pos+1;
}
ret&=dynstr_append_mem(str,cur_pos,(uint)(next_pos-cur_pos));
append=va_arg(dirty_text,char*);
}
va_end(dirty_text);
ret&=dynstr_append_mem(str,quote_str,quote_len);/*Trailingquote*/
return(ret);
}
通过定义动态字符串的结构体信息，每次分次进行字符串添加更多字符，都会根据字符串的当前的长度动态的扩容。而且每次扩容后，该结构体都记录的当前字符串的实际信息(当前字符串的长度，缓冲器可容纳字符串的长度，进行扩容的单元长度)。这样，动态字符串的处理操作就变得非常方便了。