41 lines
864 B
Go
41 lines
864 B
Go
|
package taskAgent
|
||
|
|
|
||
|
|
import (
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"google.golang.org/grpc/credentials/insecure"
|
||
|
|
tt "task-processor/pkg/taskTransport/v1"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
const pkgLogHeader string = "Task agent |"
|
||
|
|
|
||
|
|
type Handler struct {
|
||
|
|
client tt.TaskProcessorClient
|
||
|
|
*service
|
||
|
|
}
|
||
|
|
|
||
|
|
type Deps struct {
|
||
|
|
Addr string
|
||
|
|
Timeout time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewHandler(deps Deps) *Handler {
|
||
|
|
var opts []grpc.DialOption
|
||
|
|
insec := grpc.WithTransportCredentials(insecure.NewCredentials())
|
||
|
|
opts = append(opts, insec)
|
||
|
|
|
||
|
|
conn, err := grpc.NewClient(deps.Addr, opts...)
|
||
|
|
if err != nil {
|
||
|
|
log.WithError(err).Fatalf("%v grpc connection failed", pkgLogHeader)
|
||
|
|
}
|
||
|
|
|
||
|
|
client := tt.NewTaskProcessorClient(conn)
|
||
|
|
log.WithField("address", deps.Addr).Debugf("%v client", pkgLogHeader)
|
||
|
|
|
||
|
|
return &Handler{
|
||
|
|
client: client,
|
||
|
|
service: newService(client, deps.Timeout),
|
||
|
|
}
|
||
|
|
}
|