-
Notifications
You must be signed in to change notification settings - Fork 1
/
server.go
85 lines (79 loc) · 1.98 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package server
import (
"fmt"
"github.com/apache/pulsar-client-go/pulsar"
"github.com/victor-nach/pulsar-microservice/src/destroyer-service/pb"
"github.com/victor-nach/pulsar-microservice/src/destroyer-service/server/db"
"github.com/victor-nach/pulsar-microservice/src/destroyer-service/server/handler"
"github.com/victor-nach/pulsar-microservice/src/destroyer-service/server/repo"
"google.golang.org/grpc"
"log"
"net"
"os"
"time"
)
//Run - run the server
func Run() error {
c, err := db.Connect()
if err != nil {
log.Fatal(err)
}
r := repo.Repository{Collection: c}
producer, err := getProducer()
if err != nil {
log.Printf("Failed to connect to broker : %v", err)
return err
}
h := handler.Handler{&r, producer}
s := grpc.NewServer()
pb.RegisterDestroyerServiceServer(s, &h)
port, ok := os.LookupEnv("PORT")
if !ok {
port = "5051"
}
lis, err := net.Listen("tcp", ":" + port)
if err != nil {
log.Printf("Failed to listen : %v", err)
return err
}
fmt.Println(lis.Addr())
fmt.Println("Serving destroyer service...")
if err := s.Serve(lis); err != nil {
log.Printf("failed to serve: %v\n", err)
return err
}
return nil
}
// getProducer - create a pulsar producer
func getProducer() (pulsar.Producer, error) {
// create a new pulsar client
url, ok := os.LookupEnv("PULSAR_URL")
if !ok {
url = "pulsar://localhost:6650"
}
fmt.Println(url)
client, err := pulsar.NewClient(pulsar.ClientOptions{
URL: url,
OperationTimeout: 30 * time.Second,
ConnectionTimeout: 30 * time.Second,
})
if err != nil {
log.Printf("Could not instantiate Pulsar client: %v", err)
return nil, err
}
// create a new producer instance
serviceName, ok := os.LookupEnv("SERVICE_NAME")
if !ok {
serviceName = "destroyer"
}
topic, ok := os.LookupEnv("TOPIC_NAME")
if !ok {
topic = "targets.acquired"
}
producer, err := client.CreateProducer(pulsar.ProducerOptions{
Topic: topic,
Name: serviceName,
})
log.Println("Producer is ready")
return producer, nil
}