Skip to content

Commit e1a80cb

Browse files
committed
Add feature value unpack step to load test
Signed-off-by: khorshuheng <[email protected]>
1 parent f666e75 commit e1a80cb

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

main.go

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,10 +59,11 @@ func main() {
5959
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
6060
defer cancel()
6161
if len(requests) == 1 {
62-
_, err := client.GetOnlineFeatures(ctx, &requests[0])
62+
resp, err := client.GetOnlineFeatures(ctx, &requests[0])
6363
if err != nil {
6464
log.Fatalf("%v", err)
6565
}
66+
postProcessResponse(resp)
6667
w.WriteHeader(200)
6768
} else {
6869
var wg sync.WaitGroup
@@ -75,10 +76,11 @@ func main() {
7576
request := request
7677
go func() {
7778
defer wg.Done()
78-
_, err := client.GetOnlineFeatures(ctx, &request)
79+
resp, err := client.GetOnlineFeatures(ctx, &request)
7980
if err != nil {
8081
fatalErrors <- err
8182
}
83+
postProcessResponse(resp)
8284
}()
8385
}
8486

@@ -120,3 +122,11 @@ func main() {
120122
}
121123
}
122124

125+
func postProcessResponse(resp *feast.OnlineFeaturesResponse) {
126+
for _, fieldValue := range resp.RawResponse.FieldValues {
127+
for _, field := range fieldValue.Fields {
128+
field.GetVal()
129+
}
130+
}
131+
}
132+

0 commit comments

Comments
 (0)