/
schema_cache.go
52 lines (42 loc) · 1.07 KB
/
schema_cache.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package configx
import (
"context"
"crypto/sha256"
"fmt"
"github.com/dgraph-io/ristretto"
"github.com/ory/jsonschema/v3"
)
var schemaCacheConfig = &ristretto.Config{
// Hold up to 25 schemas in cache. Usually we only need one.
MaxCost: 25,
NumCounters: 250,
BufferItems: 64,
Metrics: false,
IgnoreInternalCost: true,
Cost: func(value interface{}) int64 {
return 1
},
}
var schemaCache, _ = ristretto.NewCache(schemaCacheConfig)
func getSchema(ctx context.Context, schema []byte) (*jsonschema.Schema, error) {
key := fmt.Sprintf("%x", sha256.Sum256(schema))
if val, found := schemaCache.Get(key); found {
if validator, ok := val.(*jsonschema.Schema); ok {
return validator, nil
}
schemaCache.Del(key)
}
schemaID, comp, err := newCompiler(schema)
if err != nil {
return nil, err
}
validator, err := comp.Compile(ctx, schemaID)
if err != nil {
return nil, err
}
schemaCache.Set(key, validator, 1)
schemaCache.Wait()
return validator, nil
}