qprotobufserializerregistry.cpp 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255
  1. /*
  2. * MIT License
  3. *
  4. * Copyright (c) 2019 Alexey Edelev <semlanik@gmail.com>
  5. *
  6. * This file is part of QtProtobuf project https://git.semlanik.org/semlanik/qtprotobuf
  7. *
  8. * Permission is hereby granted, free of charge, to any person obtaining a copy of this
  9. * software and associated documentation files (the "Software"), to deal in the Software
  10. * without restriction, including without limitation the rights to use, copy, modify,
  11. * merge, publish, distribute, sublicense, and/or sell copies of the Software, and
  12. * to permit persons to whom the Software is furnished to do so, subject to the following
  13. * conditions:
  14. *
  15. * The above copyright notice and this permission notice shall be included in all copies
  16. * or substantial portions of the Software.
  17. *
  18. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
  19. * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
  20. * PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
  21. * FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  22. * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  23. * DEALINGS IN THE SOFTWARE.
  24. */
  25. #include "qtprotobuflogging.h"
  26. #include "qprotobufserializerregistry_p.h"
  27. #include "qprotobufserializer.h"
  28. #include "qprotobufjsonserializer.h"
  29. #include "qprotobufserializationplugininterface.h"
  30. #include <QDir>
  31. #include <QString>
  32. #include <QHash>
  33. #include <QPluginLoader>
  34. #include <QJsonObject>
  35. #include <QJsonArray>
  36. namespace {
  37. const QLatin1String TypeNames("types");
  38. const QLatin1String MetaData("MetaData");
  39. const QLatin1String Version("version");
  40. const QLatin1String Rating("rating");
  41. const QLatin1String ProtoVersion("protobufVersion");
  42. const QLatin1String PluginName("name");
  43. const QLatin1String ProtobufSerializer("protobuf");
  44. const QLatin1String JsonSerializer("json");
  45. #ifdef _WIN32
  46. const QLatin1String LibExtension(".dll");
  47. const QLatin1String LibPrefix("");
  48. #else
  49. const QLatin1String LibExtension(".so");
  50. const QLatin1String LibPrefix("lib");
  51. #endif
  52. static const char *QtProtobufPluginPath = QT_PROTOBUF_PLUGIN_PATH;
  53. const QString DefaultImpl("Default");
  54. }
  55. namespace QtProtobuf {
  56. //! \private
  57. struct QProtobufSerializerRegistryPrivateRecord final
  58. {
  59. QProtobufSerializerRegistryPrivateRecord() : loader(nullptr) {}
  60. ~QProtobufSerializerRegistryPrivateRecord() {
  61. serializers.clear();
  62. if (loader && loader->isLoaded()) {
  63. loader->unload();
  64. }
  65. delete loader;
  66. }
  67. void createDefaultImpl()
  68. {
  69. if (serializers.find(ProtobufSerializer) == serializers.end()) {
  70. serializers[ProtobufSerializer] = std::shared_ptr<QAbstractProtobufSerializer>(new QProtobufSerializer);
  71. }
  72. if (serializers.find(JsonSerializer) == serializers.end()) {
  73. serializers[JsonSerializer] = std::shared_ptr<QAbstractProtobufSerializer>(new QProtobufJsonSerializer);
  74. }
  75. }
  76. void loadPluginMetadata(const QString &fullFilePath)
  77. {
  78. libPath = fullFilePath;
  79. loader = new QPluginLoader(fullFilePath);
  80. pluginData = loader->metaData();
  81. QVariantMap fullJson = pluginData.toVariantMap();
  82. if (!fullJson.isEmpty()) {
  83. metaData = fullJson.value(MetaData).toMap();
  84. pluginLoadedName = metaData.value(PluginName).toString();
  85. typeArray = metaData.value(TypeNames).toList();
  86. }
  87. }
  88. void loadPlugin()
  89. {
  90. QProtobufSerializationPluginInterface *loadedPlugin = qobject_cast<QProtobufSerializationPluginInterface*>(loadPluginImpl());
  91. if (!pluginData.isEmpty() && loadedPlugin) {
  92. for (int i = 0; i < typeArray.count(); i++) {
  93. QString typeName = typeArray.at(i).toString();
  94. serializers[typeName] = std::shared_ptr<QAbstractProtobufSerializer>(loadedPlugin->serializer(typeName));
  95. }
  96. }
  97. }
  98. QObject* loadPluginImpl()
  99. {
  100. if (loader == nullptr || !loader->load()) {
  101. qProtoWarning() << "Can't load plugin from" << libPath
  102. << "loader error" << (loader != nullptr ? loader->errorString() : "");
  103. return nullptr;
  104. }
  105. return loader->instance();
  106. }
  107. std::unordered_map<QString, std::shared_ptr<QAbstractProtobufSerializer>> serializers;
  108. QJsonObject pluginData;
  109. QVariantMap metaData;
  110. QString pluginLoadedName;
  111. QPluginLoader *loader;
  112. QString libPath;
  113. QVariantList typeArray;
  114. };
  115. //! \private
  116. class QProtobufSerializerRegistryPrivate
  117. {
  118. public:
  119. QProtobufSerializerRegistryPrivate()
  120. {
  121. // create default impl
  122. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> plugin = std::shared_ptr<QProtobufSerializerRegistryPrivateRecord>(new QProtobufSerializerRegistryPrivateRecord());
  123. plugin->createDefaultImpl();
  124. m_plugins[DefaultImpl] = plugin;
  125. m_pluginPath = QString::fromUtf8(QtProtobufPluginPath);
  126. QString envPluginPath = QString::fromUtf8(qgetenv("QT_PROTOBUF_PLUGIN_PATH"));
  127. if (!envPluginPath.isEmpty()) {
  128. m_pluginPath = envPluginPath;
  129. }
  130. }
  131. QString loadPlugin(const QString &name)
  132. {
  133. assert(!name.isEmpty());
  134. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> plugin = std::shared_ptr<QProtobufSerializerRegistryPrivateRecord>(new QProtobufSerializerRegistryPrivateRecord());
  135. QString libPath = m_pluginPath + QDir::separator() + LibPrefix + name + LibExtension;
  136. plugin->loadPluginMetadata(libPath);
  137. const QString &pluginName = plugin->pluginLoadedName;
  138. if (m_plugins.find(pluginName) == m_plugins.end()) {
  139. plugin->loadPlugin();
  140. m_plugins[pluginName] = plugin;
  141. } else {
  142. plugin->loader = nullptr;
  143. qProtoInfo() << "Serializer plugin with name" << pluginName << "is already loaded";
  144. }
  145. return pluginName;
  146. }
  147. std::unordered_map<QString/*pluginName*/, std::shared_ptr<QProtobufSerializerRegistryPrivateRecord>> m_plugins;
  148. QString m_pluginPath;
  149. };
  150. }
  151. using namespace QtProtobuf;
  152. QProtobufSerializerRegistry::QProtobufSerializerRegistry() :
  153. dPtr(new QProtobufSerializerRegistryPrivate())
  154. {
  155. }
  156. QProtobufSerializerRegistry::~QProtobufSerializerRegistry() = default;
  157. QString QProtobufSerializerRegistry::loadPlugin(const QString &name)
  158. {
  159. return dPtr->loadPlugin(name);
  160. }
  161. std::shared_ptr<QAbstractProtobufSerializer> QProtobufSerializerRegistry::getSerializer(const QString &id)
  162. {
  163. return dPtr->m_plugins[DefaultImpl]->serializers.at(id); //throws
  164. }
  165. std::shared_ptr<QAbstractProtobufSerializer> QProtobufSerializerRegistry::getSerializer(const QString &id, const QString &plugin)
  166. {
  167. return dPtr->m_plugins[plugin]->serializers.at(id); //throws
  168. }
  169. std::unique_ptr<QAbstractProtobufSerializer> QProtobufSerializerRegistry::acquireSerializer(const QString &/*id*/, const QString &/*plugin*/)
  170. {
  171. Q_ASSERT_X(false, "QProtobufSerializerRegistry", "acquireSerializer is unimplemented");
  172. return std::unique_ptr<QAbstractProtobufSerializer>();
  173. }
  174. float QProtobufSerializerRegistry::pluginVersion(const QString &plugin)
  175. {
  176. if (dPtr->m_plugins.find(plugin) == dPtr->m_plugins.end())
  177. return 0.0;
  178. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> implementation = dPtr->m_plugins[plugin];
  179. if (implementation->metaData.isEmpty())
  180. return 0.0;
  181. return implementation->metaData.value(Version).toFloat();
  182. }
  183. QStringList QProtobufSerializerRegistry::pluginSerializers(const QString &plugin)
  184. {
  185. QStringList strList;
  186. if (dPtr->m_plugins.find(plugin) == dPtr->m_plugins.end())
  187. return strList;
  188. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> implementation = dPtr->m_plugins[plugin];
  189. QVariantList typeArray = implementation->metaData.value(TypeNames).toList();
  190. foreach(QVariant value, typeArray) {
  191. if (!value.toString().isEmpty()) {
  192. strList.append(value.toString());
  193. }
  194. }
  195. return strList;
  196. }
  197. float QProtobufSerializerRegistry::pluginProtobufVersion(const QString &plugin)
  198. {
  199. if (dPtr->m_plugins.find(plugin) == dPtr->m_plugins.end())
  200. return 0.0;
  201. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> implementation = dPtr->m_plugins[plugin];
  202. if (implementation.get() && implementation->metaData.isEmpty())
  203. return 0.0;
  204. return implementation->metaData.value(ProtoVersion).toFloat();
  205. }
  206. int QProtobufSerializerRegistry::pluginRating(const QString &plugin)
  207. {
  208. if (dPtr->m_plugins.find(plugin) == dPtr->m_plugins.end())
  209. return 0;
  210. std::shared_ptr<QProtobufSerializerRegistryPrivateRecord> implementation = dPtr->m_plugins[plugin];
  211. if (implementation->metaData.isEmpty())
  212. return 0;
  213. return implementation->metaData.value(Rating).toInt();
  214. }