e65cbde97e
New example files: entitlement, task, lock, log, drive, config, command, info. Every major source file now has a dedicated *_example_test.go with compilable, tested examples. 561 tests, 84.8% coverage. Co-Authored-By: Virgil <virgil@lethean.io>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package core_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
. "dappco.re/go/core"
|
|
)
|
|
|
|
func ExampleEntitlement_UsagePercent() {
|
|
e := Entitlement{Limit: 100, Used: 75}
|
|
fmt.Println(e.UsagePercent())
|
|
// Output: 75
|
|
}
|
|
|
|
func ExampleCore_SetEntitlementChecker() {
|
|
c := New()
|
|
c.SetEntitlementChecker(func(action string, qty int, _ context.Context) Entitlement {
|
|
limits := map[string]int{"social.accounts": 5, "ai.credits": 100}
|
|
usage := map[string]int{"social.accounts": 3, "ai.credits": 95}
|
|
|
|
limit, ok := limits[action]
|
|
if !ok {
|
|
return Entitlement{Allowed: false, Reason: "not in package"}
|
|
}
|
|
used := usage[action]
|
|
remaining := limit - used
|
|
if qty > remaining {
|
|
return Entitlement{Allowed: false, Limit: limit, Used: used, Remaining: remaining, Reason: "limit exceeded"}
|
|
}
|
|
return Entitlement{Allowed: true, Limit: limit, Used: used, Remaining: remaining}
|
|
})
|
|
|
|
fmt.Println(c.Entitled("social.accounts", 2).Allowed)
|
|
fmt.Println(c.Entitled("social.accounts", 5).Allowed)
|
|
fmt.Println(c.Entitled("ai.credits").NearLimit(0.9))
|
|
// Output:
|
|
// true
|
|
// false
|
|
// true
|
|
}
|
|
|
|
func ExampleCore_RecordUsage() {
|
|
c := New()
|
|
var recorded string
|
|
c.SetUsageRecorder(func(action string, qty int, _ context.Context) {
|
|
recorded = fmt.Sprintf("%s:%d", action, qty)
|
|
})
|
|
|
|
c.RecordUsage("ai.credits", 10)
|
|
fmt.Println(recorded)
|
|
// Output: ai.credits:10
|
|
}
|