/
scheme.go
29 lines (27 loc) · 1.1 KB
/
scheme.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
package ydb
import (
"github.com/ydb-platform/ydb-go-sdk/v3/trace"
)
func scheme(_ *config) (t trace.Scheme) {
t.OnListDirectory = func(info trace.SchemeListDirectoryStartInfo) func(trace.SchemeListDirectoryDoneInfo) {
*info.Context = withFunctionID(*info.Context, info.Call.FunctionID())
return nil
}
t.OnDescribePath = func(info trace.SchemeDescribePathStartInfo) func(trace.SchemeDescribePathDoneInfo) {
*info.Context = withFunctionID(*info.Context, info.Call.FunctionID())
return nil
}
t.OnModifyPermissions = func(info trace.SchemeModifyPermissionsStartInfo) func(trace.SchemeModifyPermissionsDoneInfo) {
*info.Context = withFunctionID(*info.Context, info.Call.FunctionID())
return nil
}
t.OnMakeDirectory = func(info trace.SchemeMakeDirectoryStartInfo) func(trace.SchemeMakeDirectoryDoneInfo) {
*info.Context = withFunctionID(*info.Context, info.Call.FunctionID())
return nil
}
t.OnRemoveDirectory = func(info trace.SchemeRemoveDirectoryStartInfo) func(trace.SchemeRemoveDirectoryDoneInfo) {
*info.Context = withFunctionID(*info.Context, info.Call.FunctionID())
return nil
}
return t
}