[Enhancement] Allow `new_key_type!` to declare generic key type

[qouteall]

Such as

new_key_type! {
    pub struct NodeKey<T>;
}