mirror of https://github.com/go-gitea/gitea.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.1 KiB
48 lines
1.1 KiB
// Copyright 2023 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package lqinternal |
|
|
|
import ( |
|
"bytes" |
|
"encoding/binary" |
|
|
|
"github.com/syndtr/goleveldb/leveldb" |
|
"github.com/syndtr/goleveldb/leveldb/opt" |
|
) |
|
|
|
func QueueItemIDBytes(id int64) []byte { |
|
buf := make([]byte, 8) |
|
binary.PutVarint(buf, id) |
|
return buf |
|
} |
|
|
|
func QueueItemKeyBytes(prefix []byte, id int64) []byte { |
|
key := make([]byte, len(prefix), len(prefix)+1+8) |
|
copy(key, prefix) |
|
key = append(key, '-') |
|
return append(key, QueueItemIDBytes(id)...) |
|
} |
|
|
|
func RemoveLevelQueueKeys(db *leveldb.DB, namePrefix []byte) { |
|
keyPrefix := make([]byte, len(namePrefix)+1) |
|
copy(keyPrefix, namePrefix) |
|
keyPrefix[len(namePrefix)] = '-' |
|
|
|
it := db.NewIterator(nil, &opt.ReadOptions{Strict: opt.NoStrict}) |
|
defer it.Release() |
|
for it.Next() { |
|
if bytes.HasPrefix(it.Key(), keyPrefix) { |
|
_ = db.Delete(it.Key(), nil) |
|
} |
|
} |
|
} |
|
|
|
func ListLevelQueueKeys(db *leveldb.DB) (res [][]byte) { |
|
it := db.NewIterator(nil, &opt.ReadOptions{Strict: opt.NoStrict}) |
|
defer it.Release() |
|
for it.Next() { |
|
res = append(res, it.Key()) |
|
} |
|
return res |
|
}
|
|
|