繁体   English   中英

工厂函数返回带有自定义删除器的unique_ptr

[英]Factory function returning unique_ptr with custom deleter

我正在制作一个帮助程序函数,用于加载共享库,并将结果放入带有自定义删除程序的std::unique_ptr (这是模块销毁函数)。

当我没有设置自定义删除器时,这很好用,但是一旦添加它们,我将收到一条错误消息,提示无法推导出自定义删除器的类型(这是公平的)。

问题是,如果我在调用该函数时在哪里指定删除器类型,它将看起来非常难看。

问题是,我的函数如何自动推断出定制删除器的类型? 以及如何声明生成的std::unique_ptrstd::vector

我的代码(我也愿意接受任何关于代码的建议):

template <typename T, typename D>
std::unique_ptr<T, D> openLib(const std::string &lib_path,
                       const std::string &csym = "create",
                       const std::string &dsym = "destroy")
{
  void *handle;
  if (!(handle = dlopen(lib_path.c_str(), RTLD_LAZY)))
  {
    std::cerr << "dlopen " << dlerror() << std::endl;
    return nullptr;
  }
  T *(*create)();
  if (!(create = (T * (*)()) dlsym(handle, csym.c_str())))
  {
    std::cerr << "dlsym " << csym << dlerror() << std::endl;
    return nullptr;
  }
  void (*destroy)();
  if (!(destroy = (void (*)()) dlsym(handle, dsym.c_str())))
  {
    std::cerr << "dlsym " << dsym << dlerror() << std::endl;
    return nullptr;
  }
  auto cDel = [destroy](T *lib) { destroy(); };
  std::unique_ptr<T, decltype(cDel)> lib_ptr((T *)create(), cDel);
  return lib_ptr;
}

编写一个删除器类型和工厂。

using deleter = void();
using ptr_deleter = deleter*;
struct lib_unloader {
  deleter* del = 0;
  void operator()(void*)const { if (del) del(); }
  explicit operator bool() const { return del; }
};
lib_unloader get_lib_unloader( void* handle, char const* destroy_name ) {
  auto del = ptr_deleter( dlsym( handle, destroy_name ) );
  return {del};
}
template<class T>
using lib_up = std::unique_ptr< T, lib_unloader >;

template<class T>>
lib_up<T> openLib(const std::string &lib_path,
                       const std::string &csym = "create",
                       const std::string &dsym = "destroy")
{
  void *handle;
  if (!(handle = dlopen(lib_path.c_str(), RTLD_LAZY)))
  {
    std::cerr << "dlopen " << dlerror() << std::endl;
    return nullptr;
  }
  T *(*create)();
  if (!(create = (T * (*)()) dlsym(handle, csym.c_str())))
  {
    std::cerr << "dlsym " << csym << dlerror() << std::endl;
    return nullptr;
  }
  auto destroy = get_lib_unloader( handle, dsym.c_str() );
  if (!destroy) {
    std::cerr << "dlsym " << dsym << dlerror() << std::endl;
    return nullptr;
  }
  return {create(), destroy};
}

我自己,我会写一个更好的句柄。 请注意,您的代码会像疯了似的泄漏动态库句柄。

using libhandle_ptr = std::unique_ptr<void, std::integral_constant<int(*)(void*), dlclose>>;

libhandle_ptr make_libhandle(const char* filename, int flags) {
  return libhandle_ptr( dlopen(filename, flags) );
}
template<class T>
T* get_sym( libhandle_ptr const& handle, const char* symbol ) {
  if (!handle || !symbol) return nullptr;
  return static_cast<T*>( dlsym( handle.get(), symbol ) );
}

现在我们得到:

lib_unloader get_lib_unloader( libhandle const& handle, char const* destroy_name ) {
  auto del = get_sym<deleter>( handle, destroy_name );
  return {del};
}
template<class T>>
lib_up<T> openLib( libhandle_ptr const& handle,
                       const char* csym = "create",
                       const char* dsym = "destroy")
{
  if (!handle)
    return {};
  auto create = get_sym<T*()>( handle, csym );
  if (!create)
  {
    std::cerr << "dlsym " << csym << dlerror() << std::endl;
    return {};
  }
  auto destroy = get_lib_unloader( handle, dsym.c_str() );
  if (!destroy) {
    std::cerr << "dlsym " << dsym << dlerror() << std::endl;
    return {};
  }
  return {create(), destroy};
}

其中,用户希望能够提供一个libhandle_ptr当他们wantna符号。

请注意,boost具有dll符号加载库,该库可能做得更好。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM