diff --git a/backend/services/browser_service.go b/backend/services/browser_service.go index 09d290a..358533c 100644 --- a/backend/services/browser_service.go +++ b/backend/services/browser_service.go @@ -46,6 +46,7 @@ type connectionItem struct { cursor map[int]uint64 // current cursor of databases entryCursor map[int]entryCursor // current entry cursor of databases stepSize int64 + db int // current database index } type browserService struct { @@ -288,10 +289,14 @@ func (b *browserService) getRedisClient(connName string, db int) (item connectio b.connMap[connName] = item } - if db >= 0 { + if db >= 0 && item.db != db { var rdb *redis.Client if rdb, ok = client.(*redis.Client); ok && rdb != nil { - _ = rdb.Do(item.ctx, "select", strconv.Itoa(db)).Err() + if err = rdb.Do(item.ctx, "select", strconv.Itoa(db)).Err(); err != nil { + return + } + item.db = db + b.connMap[connName] = item } } return @@ -299,21 +304,8 @@ func (b *browserService) getRedisClient(connName string, db int) (item connectio // load current database size func (b *browserService) loadDBSize(ctx context.Context, client redis.UniversalClient) int64 { - if cluster, isCluster := client.(*redis.ClusterClient); isCluster { - var keyCount atomic.Int64 - cluster.ForEachMaster(ctx, func(ctx context.Context, cli *redis.Client) error { - if size, serr := cli.DBSize(ctx).Result(); serr != nil { - return serr - } else { - keyCount.Add(size) - } - return nil - }) - return keyCount.Load() - } else { - keyCount, _ := client.DBSize(ctx).Result() - return keyCount - } + keyCount, _ := client.DBSize(ctx).Result() + return keyCount } // save current scan cursor @@ -387,10 +379,10 @@ func (b *browserService) ServerInfo(name string) (resp types.JSResp) { // OpenDatabase open select database, and list all keys // @param path contain connection name and db name -func (b *browserService) OpenDatabase(connName string, db int) (resp types.JSResp) { - b.setClientCursor(connName, db, 0) +func (b *browserService) OpenDatabase(server string, db int) (resp types.JSResp) { + b.setClientCursor(server, db, 0) - item, err := b.getRedisClient(connName, db) + item, err := b.getRedisClient(server, db) if err != nil { resp.Msg = err.Error() return diff --git a/frontend/src/components/sidebar/BrowserPane.vue b/frontend/src/components/sidebar/BrowserPane.vue index d11237a..7aedffe 100644 --- a/frontend/src/components/sidebar/BrowserPane.vue +++ b/frontend/src/components/sidebar/BrowserPane.vue @@ -35,7 +35,6 @@ const props = defineProps({ const themeVars = useThemeVars() const i18n = useI18n() const dialogStore = useDialogStore() -// const prefStore = usePreferencesStore() const tabStore = useTabStore() const browserStore = useBrowserStore() const connectionStore = useConnectionStore() @@ -158,14 +157,12 @@ const handleSelectDB = async (db) => { browserStore.closeDatabase(props.server, props.db) browserStore.setKeyFilter(props.server, {}) await browserStore.openDatabase(props.server, db) + await nextTick() + await connectionStore.saveLastDB(props.server, db) + tabStore.upsertTab({ server: props.server, db }) // browserTreeRef.value?.resetExpandKey(props.server, db) fullyLoaded.value = await browserStore.loadMoreKeys(props.server, db) browserTreeRef.value?.refreshTree() - - nextTick().then(() => { - connectionStore.saveLastDB(props.server, db) - tabStore.upsertTab({ server: props.server, db }) - }) } catch (e) { $message.error(e.message) } finally { @@ -283,10 +280,10 @@ onMounted(() => onReload())