register.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. package pairec
  2. import (
  3. "gitlab.alibaba-inc.com/pai_biz_arch/pairec/filter"
  4. "gitlab.alibaba-inc.com/pai_biz_arch/pairec/recconf"
  5. "gitlab.alibaba-inc.com/pai_biz_arch/pairec/service/recall"
  6. "gitlab.alibaba-inc.com/pai_biz_arch/pairec/sort"
  7. )
  8. func register(conf *recconf.RecommendConfig) {
  9. registerDao(conf)
  10. registerRecall(conf)
  11. registerFilter(conf)
  12. registerSort(conf)
  13. // registerUserRecommendCleanHook()
  14. }
  15. func registerFilter(conf *recconf.RecommendConfig) {
  16. filter.RegisterFilterWithConfig(conf)
  17. filter.RegisterFilter("UniqueFilter", filter.NewUniqueFilter())
  18. }
  19. func registerSort(conf *recconf.RecommendConfig) {
  20. for _, conf_ := range conf.DPPConf {
  21. sort.RegisterSort(conf_.Name, sort.NewDPPSort(conf_))
  22. }
  23. }
  24. func registerDao(conf *recconf.RecommendConfig) {
  25. /**
  26. if conf.UserDaoConf.AdapterType == recconf.DaoConf_Adapter_Redis {
  27. module.RegisterUserDao("UserRedisDao", module.NewUserRedisDao())
  28. }
  29. if conf.ItemDaoConf.AdapterType == recconf.DaoConf_Adapter_Redis {
  30. module.RegisterItemDao("ItemRedisDao", module.NewItemRedisDao())
  31. // module.RegisterItemDao("ItemRedisV3Dao", module.NewItemRedisV3Dao())
  32. }
  33. module.RegisterUserDao("UserMockDao", module.NewUserMockDao())
  34. module.RegisterItemDao("ItemMockDao", module.NewItemMockDao())
  35. **/
  36. }
  37. func registerRecall(conf *recconf.RecommendConfig) {
  38. recall.Load(conf)
  39. }