grpc.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. package conf
  2. import (
  3. "github.com/xtls/xray-core/transport/internet/grpc"
  4. "google.golang.org/protobuf/proto"
  5. )
  6. type GRPCConfig struct {
  7. ServiceName string `json:"serviceName" `
  8. MultiMode bool `json:"multiMode"`
  9. IdleTimeout int32 `json:"idle_timeout"`
  10. HealthCheckTimeout int32 `json:"health_check_timeout"`
  11. PermitWithoutStream bool `json:"permit_without_stream"`
  12. InitialWindowsSize int32 `json:"initial_windows_size"`
  13. UserAgent string `json:"user_agent"`
  14. }
  15. func (g *GRPCConfig) Build() (proto.Message, error) {
  16. if g.IdleTimeout <= 0 {
  17. g.IdleTimeout = 0
  18. }
  19. if g.HealthCheckTimeout <= 0 {
  20. g.HealthCheckTimeout = 0
  21. }
  22. if g.InitialWindowsSize < 0 {
  23. // default window size of gRPC-go
  24. g.InitialWindowsSize = 0
  25. }
  26. return &grpc.Config{
  27. ServiceName: g.ServiceName,
  28. MultiMode: g.MultiMode,
  29. IdleTimeout: g.IdleTimeout,
  30. HealthCheckTimeout: g.HealthCheckTimeout,
  31. PermitWithoutStream: g.PermitWithoutStream,
  32. InitialWindowsSize: g.InitialWindowsSize,
  33. UserAgent: g.UserAgent,
  34. }, nil
  35. }