coprocess_native.go

// +build coprocess
// +build !grpc

package main

/*
#cgo python CFLAGS: -DENABLE_PYTHON
#include <stdio.h>
#include <stdlib.h>

#include "coprocess/api.h"

#ifdef ENABLE_PYTHON
#include "coprocess/python/dispatcher.h"
#include "coprocess/python/binding.h"
#endif

*/
import "C"

import (
	"errors"

	"github.com/golang/protobuf/proto"

	"github.com/TykTechnologies/tyk/coprocess"

	"encoding/json"
	"unsafe"
)

// Dispatch prepares a CoProcessMessage, sends it to the GlobalDispatcher and gets a reply.
func (c *CoProcessor) Dispatch(object *coprocess.Object) (*coprocess.Object, error) {
	if GlobalDispatcher == nil {
		return nil, errors.New("Dispatcher not initialized")
	}

	var objectMsg []byte
	var err error
	switch MessageType {
	case coprocess.ProtobufMessage:
		objectMsg, err = proto.Marshal(object)
	case coprocess.JsonMessage:
		objectMsg, err = json.Marshal(object)
	}
	if err != nil {
		return nil, err
	}

	objectMsgStr := string(objectMsg)

	CObjectStr := C.CString(objectMsgStr)
	defer C.free(unsafe.Pointer(CObjectStr))

	objectPtr := (*C.struct_CoProcessMessage)(C.malloc(C.size_t(unsafe.Sizeof(C.struct_CoProcessMessage{}))))
	objectPtr.p_data = unsafe.Pointer(CObjectStr)
	objectPtr.length = C.int(len(objectMsg))
	defer C.free(unsafe.Pointer(objectPtr))

	newObjectPtr := (*C.struct_CoProcessMessage)(GlobalDispatcher.Dispatch(unsafe.Pointer(objectPtr)))
	defer C.free(unsafe.Pointer(newObjectPtr))

	if newObjectPtr == nil {
		return nil, errors.New("Dispatch error")
	}

	newObjectBytes := C.GoBytes(newObjectPtr.p_data, newObjectPtr.length)

	newObject := &coprocess.Object{}

	switch MessageType {
	case coprocess.ProtobufMessage:
		err = proto.Unmarshal(newObjectBytes, newObject)
	case coprocess.JsonMessage:
		err = json.Unmarshal(newObjectBytes, newObject)
	}
	if err != nil {
		return nil, err
	}

	return newObject, nil
}