pika_slot.h

#ifndef PIKA_SLOT_H_
#define PIKA_SLOT_H_

#include "pika_command.h"
#include "pika_client_conn.h"
#include "strings.h"

const std::string SlotKeyPrefix = "_internal:slotkey:4migrate:";
const size_t MaxKeySendSize = 10 * 1024;
//crc 32
#define HASH_SLOTS_MASK 0x000003ff
#define HASH_SLOTS_SIZE (HASH_SLOTS_MASK + 1)

const uint32_t IEEE_POLY = 0xedb88320;
extern uint32_t crc32tab[256];

void CRC32TableInit(uint32_t poly);

extern void InitCRC32Table();

extern uint32_t CRC32Update(uint32_t crc, const char *buf, int len);

extern int SlotNum(const std::string &str);

extern int KeyType(const std::string key, std::string &key_type);

extern void SlotKeyAdd(const std::string type, const std::string key);

extern void SlotKeyRem(const std::string key);

extern void KeyNotExistsRem(const std::string type, const std::string key);


class SlotsMgrtTagSlotCmd : public Cmd {
public:
    SlotsMgrtTagSlotCmd() { }

    virtual void Do();

private:
    std::string dest_ip_;
    int64_t dest_port_;
    int64_t timeout_ms_;
    int64_t slot_num_;
    std::string key_;
    char key_type_;

    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);

    int SlotKeyPop();
};

class SlotsMgrtTagOneCmd : public Cmd {
public:
    SlotsMgrtTagOneCmd() { }

    virtual void Do();

private:
    std::string dest_ip_;
    int64_t dest_port_;
    int64_t timeout_ms_;
    std::string key_;
    int64_t slot_num_;
    char key_type_;

    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);

    int KeyTypeCheck();

    int SlotKeyRemCheck();
};

class SlotsInfoCmd : public Cmd {
public:
    SlotsInfoCmd() { }

    virtual void Do();

private:
    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);
};

class SlotsHashKeyCmd : public Cmd {
public:
    SlotsHashKeyCmd() { }

    virtual void Do();

private:
    std::vector<std::string> keys_;

    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);
};

class SlotsReloadCmd : public Cmd {
public:
    SlotsReloadCmd() { }

    virtual void Do();

private:
    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);
};

class SlotsReloadOffCmd : public Cmd {
public:
    SlotsReloadOffCmd() { }

    virtual void Do();

private:
    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);
};

class SlotsDelCmd : public Cmd {
public:
    SlotsDelCmd() { }

    virtual void Do();

private:
    std::vector<std::string> slots_;

    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);
};

class SlotsScanCmd : public Cmd {
public:
    SlotsScanCmd() : pattern_("*"), count_(10) { }

    virtual void Do();

private:
    std::string key_, pattern_;
    int64_t cursor_, count_;

    virtual void DoInitial(PikaCmdArgsType &argvs, const CmdInfo *const ptr_info);

    virtual void Clear() {
        pattern_ = "*";
        count_ = 10;
    }
};

#endif