From 4ca0a296ee08ec73f631d0006e3ab2c7927c7be0 Mon Sep 17 00:00:00 2001
From: ldy <lidanyang12@huawei.com>
Date: Mon, 9 Jun 2025 10:06:49 +0000
Subject: [PATCH] =?UTF-8?q?=E5=88=86=E5=8C=BA=E8=A1=A8=E6=A8=A1=E5=9D=97?=
 =?UTF-8?q?=E5=8F=AF=E6=B5=8B=E8=AF=95=E6=80=A7=E6=8F=90=E5=8D=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: ldy <lidanyang12@huawei.com>
---
 services/include/package/pkg_manager.h           |  7 +++++++
 services/package/pkg_manager/pkg_managerImpl.cpp | 10 ++++++++++
 services/ptable_parse/ptable_manager.h           |  8 ++++++++
 3 files changed, 25 insertions(+)

diff --git a/services/include/package/pkg_manager.h b/services/include/package/pkg_manager.h
index 29b7e8f8..0f8978c3 100644
--- a/services/include/package/pkg_manager.h
+++ b/services/include/package/pkg_manager.h
@@ -223,6 +223,13 @@ using PkgDecodeProgress = std::function<void(int type, size_t writeDataLen, cons
  */
 class PkgManager {
 public:
+    class PkgManagerFactory {
+    public:
+        virtual ~PkgManagerFactory() = default;
+        virtual PkgManagerPtr CreatePackageManager(void) const;
+        virtual void ReleasePkgManager(PkgManagerPtr pkgManger) const;
+    };
+
     using PkgManagerPtr = PkgManager *;
     using FileInfoPtr = FileInfo *;
     using PkgInfoPtr = PkgInfo *;
diff --git a/services/package/pkg_manager/pkg_managerImpl.cpp b/services/package/pkg_manager/pkg_managerImpl.cpp
index e2d0d394..852f418c 100644
--- a/services/package/pkg_manager/pkg_managerImpl.cpp
+++ b/services/package/pkg_manager/pkg_managerImpl.cpp
@@ -51,6 +51,16 @@ constexpr int32_t DIGEST_FLAGS_HAS_SIGN = 2;
 constexpr int32_t DIGEST_FLAGS_SIGNATURE = 4;
 constexpr uint32_t VERIFY_FINSH_PERCENT = 100;
 
+PkgManager::PkgManagerPtr PkgManager::PkgManagerFactory::CreatePackageManager(void) const
+{
+    return PkgManager::CreatePackageInstance();
+}
+
+void PkgManager::PkgManagerFactory::ReleasePkgManager(PkgManagerPtr pkgManger) const
+{
+    PkgManager::ReleasePackageInstance(pkgManger);
+}
+
 PkgManager::PkgManagerPtr PkgManager::CreatePackageInstance()
 {
     return new(std::nothrow) PkgManagerImpl();
diff --git a/services/ptable_parse/ptable_manager.h b/services/ptable_parse/ptable_manager.h
index 13abd0e1..979f975e 100644
--- a/services/ptable_parse/ptable_manager.h
+++ b/services/ptable_parse/ptable_manager.h
@@ -47,6 +47,14 @@ public:
     {
         pPtable_ = std::move(ptable);
     }
+    std::unique_ptr<Ptable> GetPtable(void)
+    {
+        return std::move(pPtable_);
+    }
+    void SetDeviceStorageType(StorageType storageType)
+    {
+        storage_ = storageType;
+    }
 #endif
 
     std::unique_ptr<Ptable> pPtable_;
-- 
Gitee