Skip to content

Commit

Permalink
Add old constructor for retro compatibility
Browse files Browse the repository at this point in the history
  • Loading branch information
Felippe Durán committed Sep 15, 2020
1 parent 564eb72 commit 1c43a0b
Show file tree
Hide file tree
Showing 6 changed files with 25 additions and 11 deletions.
16 changes: 15 additions & 1 deletion client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,10 @@ import (
"time"

uuid "github.com/satori/go.uuid"
"github.com/sirupsen/logrus"
"github.com/spf13/viper"
"github.com/topfreegames/eventsgateway/logger"
logruswrapper "github.com/topfreegames/eventsgateway/logger/logrus"
pb "github.com/topfreegames/protos/eventsgateway/grpc/generated"
"google.golang.org/grpc"
)
Expand All @@ -31,9 +33,21 @@ type Client struct {
serverAddress string
}

// NewClient ctor
// NewClient ctor (DEPRECATED, use New() instead)
// configPrefix is whatever comes before `client` subpart of config
func NewClient(
configPrefix string,
config *viper.Viper,
logger logrus.FieldLogger,
client pb.GRPCForwarderClient,
opts ...grpc.DialOption,
) (*Client, error) {
return New(configPrefix, config, logruswrapper.NewWithLogger(logger), client, opts...)
}

// New ctor
// configPrefix is whatever comes before `client` subpart of config
func New(
configPrefix string,
config *viper.Viper,
logger logger.Logger,
Expand Down
10 changes: 5 additions & 5 deletions client/client_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,29 +34,29 @@ var _ = Describe("Client", func() {

BeforeEach(func() {
var err error
c, err = client.NewClient("", config, log, mockGRPCClient)
c, err = client.New("", config, log, mockGRPCClient)
Expect(err).NotTo(HaveOccurred())
now = time.Now().UnixNano() / 1000000
})

Describe("NewClient", func() {
It("should return client if no error", func() {
c, err := client.NewClient("", config, log, mockGRPCClient)
c, err := client.New("", config, log, mockGRPCClient)
Expect(err).NotTo(HaveOccurred())
Expect(c).NotTo(BeNil())
})

It("should return an error if no kafka topic", func() {
config.Set("client.kafkatopic", "")
c, err := client.NewClient("", config, log, mockGRPCClient)
c, err := client.New("", config, log, mockGRPCClient)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("no kafka topic informed"))
Expect(c).To(BeNil())
})

It("should return an error if no server address", func() {
config.Set("client.grpc.serveraddress", "")
c, err := client.NewClient("", config, log, mockGRPCClient)
c, err := client.New("", config, log, mockGRPCClient)
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("no grpc server address informed"))
Expect(c).To(BeNil())
Expand All @@ -76,7 +76,7 @@ var _ = Describe("Client", func() {

BeforeEach(func() {
var err error
c, err = client.NewClient("", config, log, mockGRPCClient)
c, err = client.New("", config, log, mockGRPCClient)
Expect(err).NotTo(HaveOccurred())
now = time.Now().UnixNano() / int64(time.Millisecond)
})
Expand Down
2 changes: 1 addition & 1 deletion client/client_whitebox_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ var _ = Describe("Client Whitebox", func() {
mockCtrl := gomock.NewController(GinkgoT())
mockGRPCClient := mockpb.NewMockGRPCForwarderClient(mockCtrl)
config.Set("client.async", true)
c, err = NewClient(
c, err = New(
"",
config,
logruswrapper.NewWithLogger(logger),
Expand Down
4 changes: 2 additions & 2 deletions client/sync_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ var _ = Describe("Sync Client", func() {

BeforeEach(func() {
var err error
c, err = client.NewClient("", config, log, mockGRPCClient)
c, err = client.New("", config, log, mockGRPCClient)
Expect(err).NotTo(HaveOccurred())
now = time.Now().UnixNano() / 1000000
})
Expand All @@ -52,7 +52,7 @@ var _ = Describe("Sync Client", func() {

BeforeEach(func() {
var err error
c, err = client.NewClient("", config, log, mockGRPCClient)
c, err = client.New("", config, log, mockGRPCClient)
Expect(err).NotTo(HaveOccurred())
now = time.Now().UnixNano() / 1000000
})
Expand Down
2 changes: 1 addition & 1 deletion loadtest/runner.go
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ func newRunner(

func (r *runner) configure() error {
r.config.Set("client.kafkatopic", randomTopic())
c, err := client.NewClient("", r.config, r.log, nil)
c, err := client.New("", r.config, r.log, nil)
if err != nil {
return err
}
Expand Down
2 changes: 1 addition & 1 deletion producer/producer.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ func NewProducer(
}

func (p *Producer) configure() error {
c, err := client.NewClient("", p.config, p.log, nil)
c, err := client.New("", p.config, p.log, nil)
if err != nil {
return err
}
Expand Down

0 comments on commit 1c43a0b

Please sign in to comment.