59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package crud
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
type scopeKey struct{}
|
||
|
||
// RowScope 行级组织范围：OrgUnitIDs 非空时限制可读范围；WriteOrgUnit 写入新建行。
|
||
type RowScope struct {
|
||
OrgUnitIDs []int64
|
||
WriteOrgUnit int64
|
||
}
|
||
|
||
func WithRowScope(ctx context.Context, s RowScope) context.Context {
|
||
return context.WithValue(ctx, scopeKey{}, s)
|
||
}
|
||
|
||
func RowScopeFrom(ctx context.Context) RowScope {
|
||
v, _ := ctx.Value(scopeKey{}).(RowScope)
|
||
return v
|
||
}
|
||
|
||
func appendOrgFilter(where []string, args []any, argN int, orgIDs []int64) ([]string, []any, int) {
|
||
if len(orgIDs) == 0 {
|
||
return where, args, argN
|
||
}
|
||
ph := make([]string, 0, len(orgIDs))
|
||
for _, id := range orgIDs {
|
||
ph = append(ph, fmt.Sprintf("$%d", argN))
|
||
args = append(args, id)
|
||
argN++
|
||
}
|
||
where = append(where, fmt.Sprintf("(org_unit_id IS NULL OR org_unit_id IN (%s))", strings.Join(ph, ",")))
|
||
return where, args, argN
|
||
}
|
||
|
||
func matchOrgScope(row map[string]any, orgIDs []int64) bool {
|
||
if len(orgIDs) == 0 {
|
||
return true
|
||
}
|
||
v, ok := row["org_unit_id"]
|
||
if !ok || v == nil {
|
||
return true
|
||
}
|
||
oid := toInt64(v)
|
||
if oid == 0 {
|
||
return true
|
||
}
|
||
for _, id := range orgIDs {
|
||
if id == oid {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|