go/src/v.io/x/js.core/stress/stressd/server.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package main

import (
	"bytes"
	"fmt"
	"time"

	"v.io/v23/context"
	"v.io/v23/rpc"
	"v.io/x/js.core/stress"
)

const payloadSize = 1000

type impl struct{}

func NewStressService() stress.StressServerStub {
	return stress.StressServer(&impl{})
}

func (s *impl) Echo(_ *context.T, _ rpc.ServerCall, payload []byte) ([]byte, error) {
	return payload, nil
}

func (s *impl) ServerEcho(ctx *context.T, _ rpc.ServerCall, totalTime time.Duration, name string) (stress.StressResults, error) {
	stub := stress.StressClient(name)
	payload := make([]byte, payloadSize)
	for i := range payload {
		payload[i] = byte(i & 0xff)
	}
	var iterations int64
	var res stress.StressResults
	start := time.Now()
	for {
		got, err := stub.Echo(ctx, payload)
		if err != nil {
			return res, err
		}
		if !bytes.Equal(got, payload) {
			return res, fmt.Errorf("Echo returned %v, but expected %v", got, payload)
		}
		iterations++

		if time.Since(start) >= totalTime {
			break
		}
	}
	duration := time.Since(start).Seconds()
	res.Iterations = iterations
	res.Qps = float64(iterations) / duration
	res.MsecsPerRpc = 1000 / res.Qps
	return res, nil
}