/**
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of 
 * CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, 
 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

#ifndef METADEF_CXX_ATTR_SERIALIZER_REGISTRY_H
#define METADEF_CXX_ATTR_SERIALIZER_REGISTRY_H
#include <functional>
#include <memory>
#include <mutex>
#include <map>

#include "attr_serializer.h"

#define REG_GEIR_SERIALIZER(serializer_name, cls, obj_type, bin_type)                              \
    REG_GEIR_SERIALIZER_BUILDER_UNIQ_HELPER(serializer_name, __COUNTER__, cls, obj_type, bin_type)

#define REG_GEIR_SERIALIZER_BUILDER_UNIQ_HELPER(name, ctr, cls, obj_type, bin_type)                \
    REG_GEIR_SERIALIZER_BUILDER_UNIQ(name, ctr, cls, obj_type, bin_type)

#define REG_GEIR_SERIALIZER_BUILDER_UNIQ(name, ctr, cls, obj_type, bin_type)               \
  static ::af::AttrSerializerRegistrar register_serialize_##name##ctr                      \
      __attribute__((unused)) =                                                            \
          ::af::AttrSerializerRegistrar([]()->std::unique_ptr<af::GeIrAttrSerializer>{     \
               return std::unique_ptr<af::GeIrAttrSerializer>(new(std::nothrow)cls());     \
          }, obj_type, bin_type)

namespace af {
using ge::TypeId;
using GeIrAttrSerializerBuilder = std::function<std::unique_ptr<GeIrAttrSerializer>()>;
class AttrSerializerRegistry {
 public:
  AttrSerializerRegistry(const AttrSerializerRegistry &) = delete;
  AttrSerializerRegistry(AttrSerializerRegistry &&) = delete;
  AttrSerializerRegistry &operator=(const AttrSerializerRegistry &) = delete;
  AttrSerializerRegistry &operator=(AttrSerializerRegistry &&) = delete;

  ~AttrSerializerRegistry() = default;

  static AttrSerializerRegistry &GetInstance();
  /**
   * 注册一个GE IR的序列化、反序列化handler
   * @param builder 调用该builder时，返回一个handler的实例
   * @param obj_type 内存中的数据类型，可以通过`GetTypeId<T>`函数获得
   * @param proto_type protobuf数据类型枚举值
   */
  void RegisterGeIrAttrSerializer(const GeIrAttrSerializerBuilder &builder,
                                  const TypeId obj_type,
                                  const af::GeIrAttrDef::ValueCase proto_type);

  GeIrAttrSerializer *GetSerializer(const TypeId obj_type);
  GeIrAttrSerializer *GetDeserializer(const af::GeIrAttrDef::ValueCase proto_type);

 private:
  AttrSerializerRegistry() = default;

  std::mutex mutex_;
  std::vector<std::unique_ptr<GeIrAttrSerializer>> serializer_holder_;
  std::map<TypeId, GeIrAttrSerializer *> serializer_map_;
  std::map<af::GeIrAttrDef::ValueCase, GeIrAttrSerializer *> deserializer_map_;
};

class AttrSerializerRegistrar {
 public:
  AttrSerializerRegistrar(const GeIrAttrSerializerBuilder builder,
                          const TypeId obj_type,
                          const af::GeIrAttrDef::ValueCase proto_type) noexcept;
  ~AttrSerializerRegistrar() = default;
};
}  // namespace af

#endif  // METADEF_CXX_ATTR_SERIALIZER_REGISTRY_H