Restore lost code once again

Author: FloatingCrowbar

ydb/core/persqueue/key.h

@@ -35,8 +35,8 @@ class TKeyPrefix : public TBuffer
         : Partition(partition)
     {
         Resize(UnmarkedSize());
-        *PtrType() = type;
-        memcpy(PtrPartition(),  Sprintf("%.10" PRIu32, Partition.InternalPartitionId).data(), 10);
+        SetType(type);
+        memcpy(PtrPartition(), Sprintf("%.10" PRIu32, Partition.InternalPartitionId).data(), 10);
     }
 
     TKeyPrefix(EType type, const TPartitionId& partition, EMark mark)
@@ -62,25 +62,84 @@ class TKeyPrefix : public TBuffer
     static constexpr ui32 MarkPosition() { return UnmarkedSize(); }
     static constexpr ui32 MarkedSize() { return UnmarkedSize() + 1; }
 
+
     void SetType(EType type) {
-        *PtrType() = type;
+        if (!IsServicePartition()) {
+            *PtrType() = type;
+            return;
+        }
+        switch (type) {
+            case TypeNone:
+                *PtrType() = TypeNone;
+                return;
+	    case TypeData:
+                *PtrType() = ServiceTypeData;
+                return;
+            case TypeTmpData:
+                *PtrType() = ServiceTypeTmpData;
+                return;
+            case TypeInfo:
+                *PtrType() = ServiceTypeInfo;
+                return;
+            case TypeMeta:
+                *PtrType() = ServiceTypeMeta;
+                return;
+            case TypeTxMeta:
+               *PtrType() = ServiceTypeTxMeta;
+                return;
+            default:
+                Y_ABORT();
+        }
     }
 
+
     EType GetType() const {
-        return EType(*PtrType());
+        switch (*PtrType()) {
+            case TypeNone:
+                return TypeNone;
+            case TypeData:
+            case ServiceTypeData:
+                return TypeData;
+            case TypeTmpData:
+            case ServiceTypeTmpData:
+                return TypeTmpData;
+            case TypeInfo:
+            case ServiceTypeInfo:
+                return TypeInfo;
+            case TypeMeta:
+            case ServiceTypeMeta:
+                return TypeMeta;
+            case TypeTxMeta:
+            case ServiceTypeTxMeta:
+                return TypeTxMeta;
+        }
+        Y_ABORT();
+        return TypeNone;
     }
 
+    bool IsServicePartition() const {return Partition.WriteId.Defined();}
+
     const TPartitionId& GetPartition() const { return Partition; }
 
 protected:
     static constexpr ui32 UnmarkedSize() { return 1 + 10; }
 
     void ParsePartition()
     {
-        Partition.InternalPartitionId = FromString<ui32>(TStringBuf{PtrPartition(), 10});
+        Partition.OriginalPartitionId = FromString<ui32>(TStringBuf{PtrPartition(), 10});
+        Partition.InternalPartitionId = Partition.OriginalPartitionId;
     }
 
+
 private:
+    enum EServiceType : char {
+        ServiceTypeInfo = 'M',
+        ServiceTypeData = 'D',
+        ServiceTypeTmpData = 'X',
+        ServiceTypeMeta = 'J',
+        ServiceTypeTxMeta = 'K'
+    };
+
     char* PtrType() { return Data(); }
     char* PtrMark() { return Data() + UnmarkedSize(); }
     char* PtrPartition() { return Data() + 1; }
@@ -90,6 +149,7 @@ class TKeyPrefix : public TBuffer
     const char* PtrPartition() const { return Data() + 1; }
 
     TPartitionId Partition;
+
 };
 
 // {char type; ui32 partiton; ui64 offset; ui16 partNo; ui32 count, ui16 internalPartsCount}

ydb/core/persqueue/ut/internals_ut.cpp

@@ -233,10 +233,10 @@ Y_UNIT_TEST(StoreKeys) {
     UNIT_ASSERT_VALUES_EQUAL(keyOld.ToString(), "d0000000009_00000000000000000008_00007_0000000006_00005");
 
     TKey keyNew(TKeyPrefix::TypeData, TPartitionId{5, 1, 9}, 8, 7, 6, 5, false);
-    UNIT_ASSERT_VALUES_EQUAL(keyNew.ToString(), "D0000000000_90000000000000000008_00007_0000000006_00005");
+    UNIT_ASSERT_VALUES_EQUAL(keyNew.ToString(), "D0000000009_00000000000000000008_00007_0000000006_00005");
 
     keyNew.SetType(TKeyPrefix::TypeInfo);
-    UNIT_ASSERT_VALUES_EQUAL(keyNew.ToString(), "M0000000000_90000000000000000008_00007_0000000006_00005");
+    UNIT_ASSERT_VALUES_EQUAL(keyNew.ToString(), "M00000000009_00000000000000000008_00007_0000000006_00005");
 }
 
 Y_UNIT_TEST(RestoreKeys) {