varnitgoyal95/janus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

json support

Browse Source
This commit is contained in:
“nishant-sharma”
2021-04-27 16:40:13 +05:30
parent 256b56e9ad
commit 28a6c1ddff
6 changed files with 122 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
config/config.go
View File

@@ -11,7 +11,6 @@ type Configurations struct {
type ServerConfigurations struct {
Port string
Goroutines int
Kafka_Topic string
}
@@ -21,6 +20,8 @@ type KafkaConfigurations struct {
Retry_Backoff_MS int
Sasl_User string
Sasl_Password string
Kafka_Topic_Json string
Kafka_Topic_Protobuf string
}
type SchemaRegistryConfigurations struct {

3
config/config.yml
View File

@@ -1,7 +1,6 @@
server:
port: 8000
goroutines: 1000
kafka_topic: KAFKA_TOPIC
kafka:
bootstrap_servers: BOOTSTRAP_SERVERS # read from environment variable
@@ -9,6 +8,8 @@ kafka:
retry_backoff_ms: 500
sasl_user: KAFKA_SASL_USER # read from environment variable
sasl_password: KAFKA_SASL_PASSWORD # read from environment variable
kafka_topic_json: KAFKA_TOPIC_JSON
kafka_topic_protobuf: KAFKA_TOPIC_PROTOBUF
schemaRegistry:
endpoint: SCHEMA_REGISTRY_ENDPOINT

42
lib/RequestHandler.go
View File

@@ -11,7 +11,8 @@ import (
)
var (
RequestChannel = make(chan *RequestObject)
ProtobufRequestChannel = make(chan *RequestObject)
JsonRequestChannel = make(chan *RequestObject)
)
type RequestObject struct {
@@ -19,13 +20,21 @@ type RequestObject struct {
Header http.Header
}
func ProcessRequestChannel(topic string) {
func ProcessProtobufRequestChannel(topic string) {
for {
request := <- RequestChannel
request := <- ProtobufRequestChannel
ClickstreamProtobufEventHandler(*request, topic)
}
}
func ProcessJsonRequestChannel(topic string) {
for {
request := <- JsonRequestChannel
ClickstreamJsonEventHandler(*request, topic)
}
}
func ClickstreamProtobufEventHandler(request RequestObject, topic string) {
messageBytes := request.Body
@@ -54,5 +63,30 @@ func ClickstreamProtobufEventHandler(request RequestObject, topic string) {
}
producer_module.WriteMessageToKafkaAsync(message)
}
func ClickstreamJsonEventHandler(request RequestObject, topic string) {
messageBytes := request.Body
// to be of the format [magicByte] + [schemaID] + [messageIndex] + [value]
recordValue := []byte{}
// add [magicByte]
recordValue = append(recordValue, byte(0))
// add schemaID]
schemaIDBytes := make([]byte, 4)
binary.BigEndian.PutUint32(schemaIDBytes, uint32(producer_module.SchemaVersionMap[topic]))
recordValue = append(recordValue, schemaIDBytes...)
// Now write the bytes from the actual message...
recordValue = append(recordValue, messageBytes...)
message := &sarama.ProducerMessage{
Topic: topic,
Value: sarama.ByteEncoder(recordValue),
}
producer_module.WriteMessageToKafkaAsync(message)
}

9
main.go
View File

@@ -40,7 +40,8 @@ func init() {
configuration.Kafka.Sasl_Password = viper.GetString(configuration.Kafka.Sasl_Password)
configuration.SchemaRegistry.Endpoint = viper.GetString(configuration.SchemaRegistry.Endpoint)
configuration.SchemaRegistry.Topics = viper.GetString(configuration.SchemaRegistry.Topics)
configuration.Server.Kafka_Topic = viper.GetString(configuration.Server.Kafka_Topic)
configuration.Kafka.Kafka_Topic_Json = viper.GetString(configuration.Kafka.Kafka_Topic_Json)
configuration.Kafka.Kafka_Topic_Protobuf = viper.GetString(configuration.Kafka.Kafka_Topic_Protobuf)
port = configuration.Server.Port
log.Printf("PORT IS: ", port)
@@ -54,7 +55,11 @@ func init() {
producer_module.InitializeProducers(configuration.Kafka)
for i := 0; i < configuration.Server.Goroutines; i++ {
go lib.ProcessRequestChannel(configuration.Server.Kafka_Topic)
go lib.ProcessProtobufRequestChannel(configuration.Kafka.Kafka_Topic_Protobuf)
}
for i := 0; i < configuration.Server.Goroutines; i++ {
go lib.ProcessJsonRequestChannel(configuration.Kafka.Kafka_Topic_Json)
}
}

68
server/handlers.go
View File

@@ -7,26 +7,83 @@ import (
"io/ioutil"
"log"
"net/http"
"compress/gzip"
)
var (
healthyBool bool = true
)
func eventsHandler(w http.ResponseWriter, r *http.Request) {
var reader io.Reader
reader = r.Body
body, err := ioutil.ReadAll(reader)
if err == nil {
log.Printf(string(body))
// check if body is gzip compressed
if r.Header.Get("Content-Encoding") == "gzip" {
var err error
reader, err = gzip.NewReader(r.Body)
if err != nil {
// log.Printf(err)
w.Header().Set("Content-Type", "application/json")
http.Error(w, "Error while decompressing GZIP payload", http.StatusBadRequest)
return
}
} else {
reader = r.Body
}
body, err := ioutil.ReadAll(reader)
if err != nil {
// log.Printf(err)
w.Header().Set("Content-Type", "application/json")
http.Error(w, "Request body invalid", http.StatusBadRequest)
return
}
w.Header().Set("Content-Type", "application/json")
lib.RequestChannel <- &lib.RequestObject{Body: body, Header: r.Header}
lib.ProtobufRequestChannel <- &lib.RequestObject{Body: body, Header: r.Header}
io.WriteString(w, "ok")
}
func eventsHandlerJson(w http.ResponseWriter, r *http.Request) {
var reader io.Reader
for name, values := range r.Header {
// Loop over all values for the name.
for _, value := range values {
log.Println(name, value)
}
}
// check if body is gzip compressed
if r.Header.Get("Content-Encoding") == "gzip" {
var err error
reader, err = gzip.NewReader(r.Body)
if err != nil {
// log.Printf(err)
w.Header().Set("Content-Type", "application/json")
http.Error(w, "Error while decompressing GZIP payload", http.StatusBadRequest)
return
}
} else {
reader = r.Body
}
body, err := ioutil.ReadAll(reader)
if err != nil {
// log.Printf(err)
w.Header().Set("Content-Type", "application/json")
http.Error(w, "Request body invalid", http.StatusBadRequest)
return
}
w.Header().Set("Content-Type", "application/json")
lib.JsonRequestChannel <- &lib.RequestObject{Body: body, Header: r.Header}
io.WriteString(w, "ok")
}
func healthHandler(w http.ResponseWriter, r *http.Request) {
if healthyBool {
w.Header().Set("Content-Type", "application/json")
@@ -37,6 +94,7 @@ func healthHandler(w http.ResponseWriter, r *http.Request) {
}
}
func healthToggleHandler(w http.ResponseWriter, r *http.Request) {
healthyBool = !healthyBool
io.WriteString(w, "toggled")

3
server/server.go
View File

@@ -31,8 +31,10 @@ func NewServer(port string) (*Server, error) {
router := mux.NewRouter()
router.HandleFunc("/events", eventsHandler).Methods("POST")
router.HandleFunc("/events/json", eventsHandlerJson).Methods("POST")
router.HandleFunc("/health", healthHandler).Methods("GET")
router.HandleFunc("/health/toggle", healthToggleHandler).Methods("GET")
// router.HandleFunc("/test", testHandler).Methods("GET")
// router.HandleFunc("/schema/refresh", schemaRefreshHandler).Methods("POST")
// router.HandleFunc("/stop", stopHandler).Methods("POST")
@@ -41,4 +43,3 @@ func NewServer(port string) (*Server, error) {
return newServer, nil
}