Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ruanhaishen
redis
Commits
23f96494
Commit
23f96494
authored
May 31, 2010
by
Pieter Noordhuis
Browse files
support rdb saving/loading with dual list encoding
parent
b6eb9703
Changes
1
Hide whitespace changes
Inline
Side-by-side
redis.c
View file @
23f96494
...
@@ -3622,16 +3622,37 @@ static int rdbSaveObject(FILE *fp, robj *o) {
...
@@ -3622,16 +3622,37 @@ static int rdbSaveObject(FILE *fp, robj *o) {
if
(
rdbSaveStringObject
(
fp
,
o
)
==
-
1
)
return
-
1
;
if
(
rdbSaveStringObject
(
fp
,
o
)
==
-
1
)
return
-
1
;
}
else
if
(
o
->
type
==
REDIS_LIST
)
{
}
else
if
(
o
->
type
==
REDIS_LIST
)
{
/* Save a list value */
/* Save a list value */
list
*
list
=
o
->
ptr
;
if
(
o
->
encoding
==
REDIS_ENCODING_ZIPLIST
)
{
listIter
li
;
unsigned
char
*
p
;
listNode
*
ln
;
unsigned
char
*
vstr
;
unsigned
int
vlen
;
if
(
rdbSaveLen
(
fp
,
listLength
(
list
))
==
-
1
)
return
-
1
;
long
long
vlong
;
listRewind
(
list
,
&
li
);
while
((
ln
=
listNext
(
&
li
)))
{
if
(
rdbSaveLen
(
fp
,
ziplistLen
(
o
->
ptr
))
==
-
1
)
return
-
1
;
robj
*
eleobj
=
listNodeValue
(
ln
);
p
=
ziplistIndex
(
o
->
ptr
,
0
);
while
(
ziplistGet
(
p
,
&
vstr
,
&
vlen
,
&
vlong
))
{
if
(
vstr
)
{
if
(
rdbSaveRawString
(
fp
,
vstr
,
vlen
)
==
-
1
)
return
-
1
;
}
else
{
if
(
rdbSaveLongLongAsStringObject
(
fp
,
vlong
)
==
-
1
)
return
-
1
;
}
p
=
ziplistNext
(
o
->
ptr
,
p
);
}
}
else
if
(
o
->
encoding
==
REDIS_ENCODING_LIST
)
{
list
*
list
=
o
->
ptr
;
listIter
li
;
listNode
*
ln
;
if
(
rdbSaveStringObject
(
fp
,
eleobj
)
==
-
1
)
return
-
1
;
if
(
rdbSaveLen
(
fp
,
listLength
(
list
))
==
-
1
)
return
-
1
;
listRewind
(
list
,
&
li
);
while
((
ln
=
listNext
(
&
li
)))
{
robj
*
eleobj
=
listNodeValue
(
ln
);
if
(
rdbSaveStringObject
(
fp
,
eleobj
)
==
-
1
)
return
-
1
;
}
}
else
{
redisPanic
(
"Unknown list encoding"
);
}
}
}
else
if
(
o
->
type
==
REDIS_SET
)
{
}
else
if
(
o
->
type
==
REDIS_SET
)
{
/* Save a set value */
/* Save a set value */
...
@@ -3998,34 +4019,49 @@ static int rdbLoadDoubleValue(FILE *fp, double *val) {
...
@@ -3998,34 +4019,49 @@ static int rdbLoadDoubleValue(FILE *fp, double *val) {
/* Load a Redis object of the specified type from the specified file.
/* Load a Redis object of the specified type from the specified file.
* On success a newly allocated object is returned, otherwise NULL. */
* On success a newly allocated object is returned, otherwise NULL. */
static
robj
*
rdbLoadObject
(
int
type
,
FILE
*
fp
)
{
static
robj
*
rdbLoadObject
(
int
type
,
FILE
*
fp
)
{
robj
*
o
;
robj
*
o
,
*
ele
,
*
dec
;
size_t
len
;
redisLog
(
REDIS_DEBUG
,
"LOADING OBJECT %d (at %d)
\n
"
,
type
,
ftell
(
fp
));
redisLog
(
REDIS_DEBUG
,
"LOADING OBJECT %d (at %d)
\n
"
,
type
,
ftell
(
fp
));
if
(
type
==
REDIS_STRING
)
{
if
(
type
==
REDIS_STRING
)
{
/* Read string value */
/* Read string value */
if
((
o
=
rdbLoadEncodedStringObject
(
fp
))
==
NULL
)
return
NULL
;
if
((
o
=
rdbLoadEncodedStringObject
(
fp
))
==
NULL
)
return
NULL
;
o
=
tryObjectEncoding
(
o
);
o
=
tryObjectEncoding
(
o
);
}
else
if
(
type
==
REDIS_LIST
||
type
==
REDIS_SET
)
{
}
else
if
(
type
==
REDIS_LIST
)
{
/* Read list/set value */
/* Read list value */
uint32_t
listlen
;
if
((
len
=
rdbLoadLen
(
fp
,
NULL
))
==
REDIS_RDB_LENERR
)
return
NULL
;
o
=
createObject
(
REDIS_LIST
,
ziplistNew
());
o
->
encoding
=
REDIS_ENCODING_ZIPLIST
;
if
((
listlen
=
rdbLoadLen
(
fp
,
NULL
))
==
REDIS_RDB_LENERR
)
return
NULL
;
/* Load every single element of the list */
o
=
(
type
==
REDIS_LIST
)
?
createListObject
()
:
createSetObject
();
while
(
len
--
)
{
if
((
ele
=
rdbLoadEncodedStringObject
(
fp
))
==
NULL
)
return
NULL
;
if
(
o
->
encoding
==
REDIS_ENCODING_ZIPLIST
)
{
dec
=
getDecodedObject
(
ele
);
o
->
ptr
=
ziplistPush
(
o
->
ptr
,
dec
->
ptr
,
sdslen
(
dec
->
ptr
),
REDIS_TAIL
);
decrRefCount
(
dec
);
decrRefCount
(
ele
);
}
else
{
ele
=
tryObjectEncoding
(
ele
);
listAddNodeTail
(
o
->
ptr
,
ele
);
incrRefCount
(
ele
);
}
}
}
else
if
(
type
==
REDIS_SET
)
{
/* Read list/set value */
if
((
len
=
rdbLoadLen
(
fp
,
NULL
))
==
REDIS_RDB_LENERR
)
return
NULL
;
o
=
createSetObject
();
/* It's faster to expand the dict to the right size asap in order
/* It's faster to expand the dict to the right size asap in order
* to avoid rehashing */
* to avoid rehashing */
if
(
type
==
REDIS_SET
&&
list
len
>
DICT_HT_INITIAL_SIZE
)
if
(
len
>
DICT_HT_INITIAL_SIZE
)
dictExpand
(
o
->
ptr
,
list
len
);
dictExpand
(
o
->
ptr
,
len
);
/* Load every single element of the list/set */
/* Load every single element of the list/set */
while
(
listlen
--
)
{
while
(
len
--
)
{
robj
*
ele
;
if
((
ele
=
rdbLoadEncodedStringObject
(
fp
))
==
NULL
)
return
NULL
;
if
((
ele
=
rdbLoadEncodedStringObject
(
fp
))
==
NULL
)
return
NULL
;
ele
=
tryObjectEncoding
(
ele
);
ele
=
tryObjectEncoding
(
ele
);
if
(
type
==
REDIS_LIST
)
{
dictAdd
((
dict
*
)
o
->
ptr
,
ele
,
NULL
);
listAddNodeTail
((
list
*
)
o
->
ptr
,
ele
);
}
else
{
dictAdd
((
dict
*
)
o
->
ptr
,
ele
,
NULL
);
}
}
}
}
else
if
(
type
==
REDIS_ZSET
)
{
}
else
if
(
type
==
REDIS_ZSET
)
{
/* Read list/set value */
/* Read list/set value */
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment