rule_chain.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. package main
  2. import "sparrow/pkg/models"
  3. func (r *Registry) FindRuleChains(vendorId string, reply *[]*models.RuleChain) error {
  4. db, err := getDB()
  5. if err != nil {
  6. return err
  7. }
  8. return db.Where("vendor_id = ?", vendorId).
  9. Find(reply).Error
  10. }
  11. func (r *Registry) FindRuleChainById(ruleChainId string, reply *models.RuleChain) error {
  12. db, err := getDB()
  13. if err != nil {
  14. return err
  15. }
  16. return db.Where("record_id = ?", ruleChainId).
  17. First(reply).Error
  18. }
  19. func (r *Registry) FindRuleNodeById(nodeId string, reply *models.RuleNode) error {
  20. db, err := getDB()
  21. if err != nil {
  22. return err
  23. }
  24. return db.Where("record_id = ?", nodeId).First(reply).Error
  25. }
  26. func (r *Registry) FindRuleChainNodes(ruleChainId string, reply *[]*models.RuleNode) error {
  27. db, err := getDB()
  28. if err != nil {
  29. return err
  30. }
  31. return db.Where("rule_chain_id = ?", ruleChainId).
  32. Find(reply).Error
  33. }
  34. func (r *Registry) GetRuleNodeRelations(nodeId string, reply*[]*models.Relation) error {
  35. db, err := getDB()
  36. if err != nil {
  37. return err
  38. }
  39. return db.Where("from_id = ?", nodeId).Find(reply).Error
  40. }