102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/pebbe/zmq4"
|
|
"github.com/satori/go.uuid"
|
|
"bitbucket.org/nicdex/adaptech-goes"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
func main() {
|
|
fmt.Println("Simple ZeroMQ server for goes.")
|
|
|
|
//TODO: config/flag
|
|
storagePath := path.Join(os.TempDir(), uuid.NewV4().String())
|
|
storagePath = "c:\\dev\\go\\events"
|
|
|
|
goes.SetStorage(goes.NewDiskStorage(storagePath))
|
|
goes.SetSerializer(goes.NewPassthruSerializer())
|
|
|
|
context, err := zmq4.NewContext()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer context.Term()
|
|
|
|
replySocket, err := context.NewSocket(zmq4.REP)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer replySocket.Close()
|
|
|
|
//TODO: config/flag
|
|
listenAddr := "tcp://*:12345"
|
|
err = replySocket.Bind(listenAddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for {
|
|
message, err := replySocket.RecvMessageBytes(0)
|
|
if err != nil {
|
|
fmt.Println("Error receiving command from client", err)
|
|
continue
|
|
}
|
|
|
|
command := string(message[0])
|
|
switch command {
|
|
case "AddEvent":
|
|
aggregateId, err := uuid.FromBytes(message[1])
|
|
if err != nil {
|
|
fmt.Println("Wrong format for AggregateId", err)
|
|
break
|
|
}
|
|
fmt.Println("->", command, aggregateId.String())
|
|
data := message[2]
|
|
err = goes.AddEvent(goes.Event{aggregateId, data})
|
|
if err != nil {
|
|
replySocket.Send(fmt.Sprintf("Error: %v", err), 0)
|
|
fmt.Println(err)
|
|
break
|
|
}
|
|
replySocket.Send("Ok", 0)
|
|
case "ReadStream":
|
|
aggregateId, err := uuid.FromBytes(message[1])
|
|
if err != nil {
|
|
fmt.Println("Wrong format for AggregateId", err)
|
|
break
|
|
}
|
|
fmt.Println("->", command, aggregateId.String())
|
|
events, err := goes.RetrieveFor(aggregateId)
|
|
if err != nil {
|
|
replySocket.Send(fmt.Sprintf("Error: %v", err), 0)
|
|
fmt.Println(err)
|
|
break
|
|
}
|
|
sendEvents(replySocket, events)
|
|
case "ReadAll":
|
|
fmt.Println("->", command)
|
|
events, err := goes.RetrieveAll()
|
|
if err != nil {
|
|
replySocket.Send(fmt.Sprintf("Error: %v", err), 0)
|
|
fmt.Println(err)
|
|
break
|
|
}
|
|
sendEvents(replySocket, events)
|
|
}
|
|
}
|
|
}
|
|
|
|
func sendEvents(socket *zmq4.Socket, events []*goes.Event) {
|
|
len := len(events)
|
|
socket.Send(fmt.Sprintf("%v", len), zmq4.SNDMORE)
|
|
|
|
i := 0
|
|
for ; i < len-1; i++ {
|
|
socket.SendBytes(events[i].Payload.([]byte), zmq4.SNDMORE)
|
|
}
|
|
socket.SendBytes(events[i].Payload.([]byte), 0)
|
|
fmt.Println("<-", len, "events")
|
|
} |