contentserver/server/socket.go

package server

import (
	"errors"
	"fmt"
	json "github.com/foomo/ContentServer/server/jjson"
	"github.com/foomo/ContentServer/server/log"
	"github.com/foomo/ContentServer/server/repo"
	"github.com/foomo/ContentServer/server/requests"
	"net"
	"strconv"
	"strings"
)

// there should be sth. built in ?!
// anyway this ony concatenates two "ByteArrays"
func concat(a []byte, b []byte) []byte {
	newslice := make([]byte, len(a)+len(b))
	copy(newslice, a)
	copy(newslice[len(a):], b)
	return newslice
}

func handleSocketRequest(handler string, jsonBuffer []byte) (replyBytes []byte, err error) {
	countRequest()
	var reply interface{}
	var jsonErr error
	log.Record(fmt.Sprintf("socket.handleSocketRequest(%d): %s %s", numRequests(), handler, string(jsonBuffer)))
	switch handler {
	case "getURIs":
		getURIRequest := requests.NewURIs()
		jsonErr = json.Unmarshal(jsonBuffer, &getURIRequest)
		log.Debug("  getURIRequest: " + fmt.Sprint(getURIRequest))
		uris := contentRepo.GetURIs(getURIRequest.Region, getURIRequest.Language, getURIRequest.Ids)
		log.Debug("    resolved: " + fmt.Sprint(uris))
		reply = uris
		break
	case "content":
		contentRequest := requests.NewContent()
		jsonErr = json.Unmarshal(jsonBuffer, &contentRequest)
		log.Debug("  contentRequest: " + fmt.Sprint(contentRequest))
		content := contentRepo.GetContent(contentRequest)
		reply = content
		break
	case "getItemMap":
		itemMapRequest := requests.NewItemMap()
		jsonErr = json.Unmarshal(jsonBuffer, &itemMapRequest)
		log.Debug("  itemMapRequest: " + fmt.Sprint(itemMapRequest))
		itemMap := contentRepo.GetItemMap(itemMapRequest.Id)
		reply = itemMap
		break
	default:
		err = errors.New(log.Error("  can not handle this one " + handler))
	}
	if err == nil {
		if jsonErr != nil {
			log.Error("  could not read incoming json: " + fmt.Sprint(jsonErr))
			err = jsonErr
		} else {
			encodedBytes, jsonErr := json.MarshalIndent(map[string]interface{}{"reply": reply}, "", " ")
			if jsonErr != nil {
				err = jsonErr
				log.Error("  could not encode reply " + fmt.Sprint(jsonErr))
			} else {
				replyBytes = encodedBytes
			}
		}
	}
	return replyBytes, err
}

func handleConnection(conn net.Conn) {
	log.Debug("socket.handleConnection")
	var headerBuffer [1]byte
	header := ""
	for {
		_, readErr := conn.Read(headerBuffer[0:])
		if readErr != nil {
			log.Debug("  looks like the client closed the connection - this is my readError: " + fmt.Sprint(readErr))
			return
		}
		// read next byte
		current := string(headerBuffer[0:])
		if current == "{" {
			// json has started
			headerParts := strings.Split(header, ":")
			header = ""
			requestHandler := headerParts[0]
			jsonLength, _ := strconv.Atoi(headerParts[1])
			log.Debug(fmt.Sprintf("  found json with %d bytes", jsonLength))
			if jsonLength > 0 {
				jsonBuffer := make([]byte, jsonLength)
				jsonBuffer[0] = 123
				_, jsonReadErr := conn.Read(jsonBuffer[1:])
				if jsonReadErr != nil {
					log.Error("  could not read json - giving up with this client connection" + fmt.Sprint(jsonReadErr))
					return
				} else {
					log.Debug("  read json: " + string(jsonBuffer))
				}
				reply, handlingError := handleSocketRequest(requestHandler, jsonBuffer)
				if handlingError != nil {
					log.Error("socket.handleConnection: handlingError " + fmt.Sprint(handlingError))
					return
				} else {
					headerBytes := []byte(strconv.Itoa(len(reply)))
					reply = concat(headerBytes, reply)
					log.Debug("  replying: " + string(reply))
					_, writeError := conn.Write(reply)
					if writeError != nil {
						log.Error("socket.handleConnection: could not write my reply: " + fmt.Sprint(writeError))
						return
					} else {
						log.Debug("  replied. waiting for next request on open connection")
						//return
					}
				}
			} else {
				log.Error("can not read empty json")
				return
			}
		} else {
			// adding to header byte by byte
			header += string(headerBuffer[0:])
		}
	}
}

func RunSocketServer(server string, address string) {
	log.Record("building repo with content from " + server)
	contentRepo = repo.NewRepo(server)
	contentRepo.Update()
	ln, err := net.Listen("tcp", address)
	if err != nil {
		// failed to create socket
		log.Error("RunSocketServer: could not start the on \"" + address + "\" - error: " + fmt.Sprint(err))
	} else {
		// there we go
		log.Record("RunSocketServer: started to listen on " + address)
		for {
			conn, err := ln.Accept() // this blocks until connection or error
			if err != nil {
				log.Error("RunSocketServer: could not accept connection" + fmt.Sprint(err))
				continue
			} else {
				go handleConnection(conn) // a goroutine handles conn so that the loop can accept other connections
			}
		}
	}
}