Changeset 0.22.2 (#96)
This commit is contained in:
110
server/store/postgres/routing_policy.go
Normal file
110
server/store/postgres/routing_policy.go
Normal file
@@ -0,0 +1,110 @@
|
||||
package postgres
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
|
||||
"git.gobha.me/xcaliber/chat-switchboard/models"
|
||||
)
|
||||
|
||||
// RoutingPolicyStore implements store.RoutingPolicyStore for Postgres.
|
||||
type RoutingPolicyStore struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
func NewRoutingPolicyStore(db *sql.DB) *RoutingPolicyStore {
|
||||
return &RoutingPolicyStore{db: db}
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) Create(ctx context.Context, p *models.RoutingPolicy) error {
|
||||
configJSON, err := json.Marshal(p.Config)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return s.db.QueryRowContext(ctx, `
|
||||
INSERT INTO routing_policies (name, scope, team_id, priority, policy_type, config, is_active)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
||||
RETURNING id, created_at, updated_at
|
||||
`, p.Name, p.Scope, p.TeamID, p.Priority, p.Type, configJSON, p.IsActive,
|
||||
).Scan(&p.ID, &p.CreatedAt, &p.UpdatedAt)
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) Update(ctx context.Context, p *models.RoutingPolicy) error {
|
||||
configJSON, err := json.Marshal(p.Config)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = s.db.ExecContext(ctx, `
|
||||
UPDATE routing_policies
|
||||
SET name = $2, scope = $3, team_id = $4, priority = $5,
|
||||
policy_type = $6, config = $7, is_active = $8
|
||||
WHERE id = $1
|
||||
`, p.ID, p.Name, p.Scope, p.TeamID, p.Priority, p.Type, configJSON, p.IsActive)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) Delete(ctx context.Context, id string) error {
|
||||
_, err := s.db.ExecContext(ctx, `DELETE FROM routing_policies WHERE id = $1`, id)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) GetByID(ctx context.Context, id string) (*models.RoutingPolicy, error) {
|
||||
var p models.RoutingPolicy
|
||||
var configJSON []byte
|
||||
err := s.db.QueryRowContext(ctx, `
|
||||
SELECT id, name, scope, team_id, priority, policy_type, config, is_active, created_at, updated_at
|
||||
FROM routing_policies WHERE id = $1
|
||||
`, id).Scan(&p.ID, &p.Name, &p.Scope, &p.TeamID, &p.Priority, &p.Type, &configJSON, &p.IsActive, &p.CreatedAt, &p.UpdatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
json.Unmarshal(configJSON, &p.Config)
|
||||
return &p, nil
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) ListActive(ctx context.Context) ([]models.RoutingPolicy, error) {
|
||||
return s.query(ctx, `
|
||||
SELECT id, name, scope, team_id, priority, policy_type, config, is_active, created_at, updated_at
|
||||
FROM routing_policies
|
||||
WHERE is_active = true
|
||||
ORDER BY priority ASC, created_at ASC
|
||||
`)
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) ListAll(ctx context.Context) ([]models.RoutingPolicy, error) {
|
||||
return s.query(ctx, `
|
||||
SELECT id, name, scope, team_id, priority, policy_type, config, is_active, created_at, updated_at
|
||||
FROM routing_policies
|
||||
ORDER BY priority ASC, created_at ASC
|
||||
`)
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) ListForTeam(ctx context.Context, teamID string) ([]models.RoutingPolicy, error) {
|
||||
return s.query(ctx, `
|
||||
SELECT id, name, scope, team_id, priority, policy_type, config, is_active, created_at, updated_at
|
||||
FROM routing_policies
|
||||
WHERE is_active = true AND (scope = 'global' OR (scope = 'team' AND team_id = $1))
|
||||
ORDER BY priority ASC, created_at ASC
|
||||
`, teamID)
|
||||
}
|
||||
|
||||
func (s *RoutingPolicyStore) query(ctx context.Context, q string, args ...interface{}) ([]models.RoutingPolicy, error) {
|
||||
rows, err := s.db.QueryContext(ctx, q, args...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var result []models.RoutingPolicy
|
||||
for rows.Next() {
|
||||
var p models.RoutingPolicy
|
||||
var configJSON []byte
|
||||
if err := rows.Scan(&p.ID, &p.Name, &p.Scope, &p.TeamID, &p.Priority, &p.Type, &configJSON, &p.IsActive, &p.CreatedAt, &p.UpdatedAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
json.Unmarshal(configJSON, &p.Config)
|
||||
result = append(result, p)
|
||||
}
|
||||
return result, rows.Err()
|
||||
}
|
||||
Reference in New Issue
Block a user