47 lines
925 B
Go
47 lines
925 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package testservice
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
type Simple struct {
|
|
Name string
|
|
DC string
|
|
}
|
|
|
|
func (s *Simple) Flow(_ *Req, flow Simple_FlowServer) error {
|
|
for flow.Context().Err() == nil {
|
|
resp := &Resp{ServerName: "one", Datacenter: s.DC}
|
|
if err := flow.Send(resp); err != nil {
|
|
return err
|
|
}
|
|
time.Sleep(time.Millisecond)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Simple) Something(_ context.Context, _ *Req) (*Resp, error) {
|
|
return &Resp{ServerName: s.Name, Datacenter: s.DC}, nil
|
|
}
|
|
|
|
type SimplePanic struct {
|
|
Name, DC string
|
|
}
|
|
|
|
func (s *SimplePanic) Flow(_ *Req, flow Simple_FlowServer) error {
|
|
for flow.Context().Err() == nil {
|
|
time.Sleep(time.Millisecond)
|
|
panic("panic from Flow")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *SimplePanic) Something(_ context.Context, _ *Req) (*Resp, error) {
|
|
time.Sleep(time.Millisecond)
|
|
panic("panic from Something")
|
|
}
|