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
9688f6a4
Commit
9688f6a4
authored
Feb 08, 2023
by
Vitaly Arbuzov
Browse files
Make DBSIZE O(1) and use 64 bit random
parent
217a5eb7
Changes
2
Show whitespace changes
Inline
Side-by-side
src/db.c
View file @
9688f6a4
...
@@ -234,6 +234,7 @@ void dbAdd(redisDb *db, robj *key, robj *val) {
...
@@ -234,6 +234,7 @@ void dbAdd(redisDb *db, robj *key, robj *val) {
dictEntry
*
de
=
dictAddRaw
(
d
,
copy
,
NULL
);
dictEntry
*
de
=
dictAddRaw
(
d
,
copy
,
NULL
);
serverAssertWithInfo
(
NULL
,
key
,
de
!=
NULL
);
serverAssertWithInfo
(
NULL
,
key
,
de
!=
NULL
);
dictSetVal
(
d
,
de
,
val
);
dictSetVal
(
d
,
de
,
val
);
db
->
key_count
++
;
signalKeyAsReady
(
db
,
key
,
val
->
type
);
signalKeyAsReady
(
db
,
key
,
val
->
type
);
notifyKeyspaceEvent
(
NOTIFY_NEW
,
"new"
,
key
,
db
->
id
);
notifyKeyspaceEvent
(
NOTIFY_NEW
,
"new"
,
key
,
db
->
id
);
}
}
...
@@ -262,6 +263,7 @@ int dbAddRDBLoad(redisDb *db, sds key, robj *val) {
...
@@ -262,6 +263,7 @@ int dbAddRDBLoad(redisDb *db, sds key, robj *val) {
dictEntry
*
de
=
dictAddRaw
(
d
,
key
,
NULL
);
dictEntry
*
de
=
dictAddRaw
(
d
,
key
,
NULL
);
if
(
de
==
NULL
)
return
0
;
if
(
de
==
NULL
)
return
0
;
dictSetVal
(
d
,
de
,
val
);
dictSetVal
(
d
,
de
,
val
);
db
->
key_count
++
;
return
1
;
return
1
;
}
}
...
@@ -387,12 +389,7 @@ robj *dbRandomKey(redisDb *db) {
...
@@ -387,12 +389,7 @@ robj *dbRandomKey(redisDb *db) {
/* Return random non-empty dictionary from this DB. */
/* Return random non-empty dictionary from this DB. */
dict
*
getRandomDict
(
redisDb
*
db
)
{
dict
*
getRandomDict
(
redisDb
*
db
)
{
if
(
db
->
dict_count
==
1
)
return
db
->
dict
[
0
];
if
(
db
->
dict_count
==
1
)
return
db
->
dict
[
0
];
unsigned
long
long
target
=
rand
();
unsigned
long
target
=
randomULong
();
/* If number of keys is greater than max target value, then we should call target once more and get more entropy. */
if
(
db
->
key_count
>
RAND_MAX
)
{
target
<<=
32
;
target
|=
rand
();
}
unsigned
long
long
int
key_count
=
dbSize
(
db
);
unsigned
long
long
int
key_count
=
dbSize
(
db
);
if
(
!
key_count
)
return
db
->
dict
[
0
];
if
(
!
key_count
)
return
db
->
dict
[
0
];
target
%=
key_count
;
/* Random key index in range [0..KEY_COUNT). */
target
%=
key_count
;
/* Random key index in range [0..KEY_COUNT). */
...
@@ -531,6 +528,7 @@ long long emptyDbStructure(redisDb *dbarray, int dbnum, int async,
...
@@ -531,6 +528,7 @@ long long emptyDbStructure(redisDb *dbarray, int dbnum, int async,
/* Because all keys of database are removed, reset average ttl. */
/* Because all keys of database are removed, reset average ttl. */
dbarray
[
j
].
avg_ttl
=
0
;
dbarray
[
j
].
avg_ttl
=
0
;
dbarray
[
j
].
expires_cursor
=
0
;
dbarray
[
j
].
expires_cursor
=
0
;
dbarray
[
j
].
key_count
=
0
;
}
}
return
removed
;
return
removed
;
...
@@ -1157,14 +1155,7 @@ void dbsizeCommand(client *c) {
...
@@ -1157,14 +1155,7 @@ void dbsizeCommand(client *c) {
}
}
unsigned
long
long
int
dbSize
(
redisDb
*
db
)
{
unsigned
long
long
int
dbSize
(
redisDb
*
db
)
{
unsigned
long
long
size
=
0
;
return
db
->
key_count
;
dict
*
d
;
dbIterator
dbit
;
dbInitIterator
(
&
dbit
,
db
);
while
((
d
=
dbNextDict
(
&
dbit
)))
{
size
+=
dictSize
(
d
);
}
return
size
;
}
}
unsigned
long
dbSlots
(
redisDb
*
db
)
{
unsigned
long
dbSlots
(
redisDb
*
db
)
{
...
@@ -1568,12 +1559,14 @@ int dbSwapDatabases(int id1, int id2) {
...
@@ -1568,12 +1559,14 @@ int dbSwapDatabases(int id1, int id2) {
db1
->
avg_ttl
=
db2
->
avg_ttl
;
db1
->
avg_ttl
=
db2
->
avg_ttl
;
db1
->
expires_cursor
=
db2
->
expires_cursor
;
db1
->
expires_cursor
=
db2
->
expires_cursor
;
db1
->
dict_count
=
db2
->
dict_count
;
db1
->
dict_count
=
db2
->
dict_count
;
db1
->
key_count
=
db2
->
key_count
;
db2
->
dict
=
aux
.
dict
;
db2
->
dict
=
aux
.
dict
;
db2
->
expires
=
aux
.
expires
;
db2
->
expires
=
aux
.
expires
;
db2
->
avg_ttl
=
aux
.
avg_ttl
;
db2
->
avg_ttl
=
aux
.
avg_ttl
;
db2
->
expires_cursor
=
aux
.
expires_cursor
;
db2
->
expires_cursor
=
aux
.
expires_cursor
;
db2
->
dict_count
=
aux
.
dict_count
;
db2
->
dict_count
=
aux
.
dict_count
;
db2
->
key_count
=
aux
.
key_count
;
/* Now we need to handle clients blocked on lists: as an effect
/* Now we need to handle clients blocked on lists: as an effect
* of swapping the two DBs, a client that was waiting for list
* of swapping the two DBs, a client that was waiting for list
...
@@ -1612,12 +1605,14 @@ void swapMainDbWithTempDb(redisDb *tempDb) {
...
@@ -1612,12 +1605,14 @@ void swapMainDbWithTempDb(redisDb *tempDb) {
activedb
->
avg_ttl
=
newdb
->
avg_ttl
;
activedb
->
avg_ttl
=
newdb
->
avg_ttl
;
activedb
->
expires_cursor
=
newdb
->
expires_cursor
;
activedb
->
expires_cursor
=
newdb
->
expires_cursor
;
activedb
->
dict_count
=
newdb
->
dict_count
;
activedb
->
dict_count
=
newdb
->
dict_count
;
activedb
->
key_count
=
newdb
->
key_count
;
newdb
->
dict
=
aux
.
dict
;
newdb
->
dict
=
aux
.
dict
;
newdb
->
expires
=
aux
.
expires
;
newdb
->
expires
=
aux
.
expires
;
newdb
->
avg_ttl
=
aux
.
avg_ttl
;
newdb
->
avg_ttl
=
aux
.
avg_ttl
;
newdb
->
expires_cursor
=
aux
.
expires_cursor
;
newdb
->
expires_cursor
=
aux
.
expires_cursor
;
newdb
->
dict_count
=
aux
.
dict_count
;
newdb
->
dict_count
=
aux
.
dict_count
;
newdb
->
key_count
=
aux
.
key_count
;
/* Now we need to handle clients blocked on lists: as an effect
/* Now we need to handle clients blocked on lists: as an effect
* of swapping the two DBs, a client that was waiting for list
* of swapping the two DBs, a client that was waiting for list
...
...
src/server.c
View file @
9688f6a4
...
@@ -2593,6 +2593,7 @@ void initServer(void) {
...
@@ -2593,6 +2593,7 @@ void initServer(void) {
server
.
db
[
j
].
defrag_later
=
listCreate
();
server
.
db
[
j
].
defrag_later
=
listCreate
();
server
.
db
[
j
].
rehashing
=
listCreate
();
server
.
db
[
j
].
rehashing
=
listCreate
();
server
.
db
[
j
].
dict_count
=
slotCount
;
server
.
db
[
j
].
dict_count
=
slotCount
;
server
.
db
[
j
].
key_count
=
0
;
listSetFreeMethod
(
server
.
db
[
j
].
defrag_later
,(
void
(
*
)(
void
*
))
sdsfree
);
listSetFreeMethod
(
server
.
db
[
j
].
defrag_later
,(
void
(
*
)(
void
*
))
sdsfree
);
}
}
evictionPoolAlloc
();
/* Initialize the LRU keys pool. */
evictionPoolAlloc
();
/* Initialize the LRU keys pool. */
...
...
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