Create a pip package for pipe_io_server

This commit is contained in:
Kristóf Tóth
2019-08-02 15:32:38 +02:00
parent 499ce8af76
commit c39527d99e
19 changed files with 98 additions and 0 deletions

View File

@ -0,0 +1,29 @@
#include "pipe_io.h"
void run2() {
pipe_io pipeio = pipeio_new_io("in", "out");
const char* line;
while ((line = pipeio_recv_msg(&pipeio.reader))) {
pipeio_send_msg(&pipeio.writer, line);
}
pipeio_close_io(&pipeio);
}
void msg_handler(const char* msg, void* ctx) {
pipe_writer* writer = (pipe_writer*)(ctx);
pipeio_send_msg(writer, msg);
}
void run1() {
pipe_io pipeio = pipeio_new_io("in", "out");
pipeio_set_msg_handler(&pipeio.reader, &msg_handler, &pipeio.writer);
pipeio_run(&pipeio.reader);
pipeio_close_io(&pipeio);
}
int main() {
run1(); // or run2();
// note that run1 and run2 do the same thing
// and you can use whichever API you prefer
}

View File

@ -0,0 +1,13 @@
#include <iostream>
#include <string>
#include <fstream>
#include "pipe_io.hpp"
int main() {
PipeIO pipe_io("in", "out");
pipe_io.reader.set_message_handler(
[&pipe_io](std::string msg){pipe_io.writer.send_message(msg);}
);
pipe_io.reader.run();
}

View File

@ -0,0 +1,147 @@
#pragma once
#include <stdlib.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>
typedef struct pipe_reader {
FILE* stream;
char* line_buf;
size_t buf_size;
void (*msg_handler)(const char*, void*);
void* handler_ctx;
} pipe_reader;
static void null_handler(const char* msg, void* ctx) {}
pipe_reader pipeio_new_reader(const char* pipe_path) {
pipe_reader preader;
preader.stream = fopen(pipe_path, "r");
preader.line_buf = NULL;
preader.buf_size = 0;
preader.msg_handler = &null_handler;
return preader;
}
bool pipeio_check_reader(const pipe_reader* preader) {
if (!preader || preader->stream == NULL) {
return false;
}
return true;
}
const char* pipeio_recv_msg(pipe_reader* preader) {
if (!pipeio_check_reader(preader)) {
return NULL;
}
ssize_t read = getline(&preader->line_buf, &preader->buf_size, preader->stream);
if (read > 0) {
preader->line_buf[read - 1] = '\0';
return preader->line_buf;
}
return NULL;
}
void pipeio_set_msg_handler(pipe_reader* preader, void (*msg_handler)(const char*, void*), void* handler_ctx) {
if (msg_handler != NULL && handler_ctx != NULL) {
preader->msg_handler = msg_handler;
preader->handler_ctx = handler_ctx;
}
}
void pipeio_run(pipe_reader* preader) {
const char* line;
while ((line = pipeio_recv_msg(preader))) {
(*preader->msg_handler)(line, preader->handler_ctx);
}
}
void pipeio_close_reader(pipe_reader* preader) {
if (!pipeio_check_reader(preader)) {
return;
}
fclose(preader->stream);
preader->stream = NULL;
if (preader->line_buf) {
free(preader->line_buf);
preader->line_buf = NULL;
}
}
typedef struct pipe_writer {
FILE* stream;
} pipe_writer;
pipe_writer pipeio_new_writer(const char* pipe_path) {
pipe_writer pwriter;
pwriter.stream = fopen(pipe_path, "w");
return pwriter;
}
bool pipeio_check_writer(const pipe_writer* pwriter) {
if (!pwriter || pwriter->stream == NULL) {
return false;
}
return true;
}
bool pipeio_send_msg(pipe_writer* pwriter, const char* msg) {
if (!pipeio_check_writer(pwriter)) {
return false;
}
fprintf(pwriter->stream, "%s\n", msg);
fflush(pwriter->stream);
return true;
}
void pipeio_close_writer(pipe_writer* pwriter) {
if (!pipeio_check_writer(pwriter)) {
return;
}
fclose(pwriter->stream);
pwriter->stream = NULL;
}
typedef struct pipe_io {
pipe_reader reader;
pipe_writer writer;
} pipe_io;
pipe_io pipeio_new_io(const char* in_pipe_path, const char* out_pipe_path) {
pipe_io pipeio;
pipeio.reader = pipeio_new_reader(in_pipe_path);
pipeio.writer = pipeio_new_writer(out_pipe_path);
return pipeio;
}
bool pipeio_check_io(const pipe_io* pipeio) {
if (!pipeio) {
return false;
}
return (pipeio_check_reader(&pipeio->reader) && pipeio_check_writer(&pipeio->writer));
}
void pipeio_close_io(pipe_io* pipeio) {
if (!pipeio_check_io(pipeio)) {
return;
}
pipeio_close_reader(&pipeio->reader);
pipeio_close_writer(&pipeio->writer);
}
// const char* recv_msg_pipe_io(pipe_io* pipeio) {
// if (!check_pipe_io(pipeio)) {
// return NULL;
// }
// return recv_msg_pipe_reader(&pipeio->preader);
// }
// bool send_msg_pipe_io(pipe_io* pipeio, const char* msg) {
// if (!check_pipe_io(pipeio)) {
// return false;
// }
// return send_msg_pipe_writer(&pipeio->pwriter, msg);
// }

View File

@ -0,0 +1,79 @@
extern "C" {
#include "pipe_io.h"
}
#include <string>
#include <iostream>
#include <functional>
class PipeReader {
public:
PipeReader(const std::string& pipe_path)
:preader(pipeio_new_reader(pipe_path.c_str())) {}
virtual ~PipeReader() {
pipeio_close_reader(&this->preader);
}
PipeReader(PipeReader&) = delete;
PipeReader(PipeReader&&) = delete;
PipeReader& operator=(const PipeReader&) = delete;
PipeReader& operator=(const PipeReader&&) = delete;
void set_message_handler(std::function<void(std::string)> message_handler) {
this->message_handler = message_handler;
}
void run() {
while (const char* linebuf = this->recv_message()) {
std::string msg(linebuf);
this->message_handler(msg);
}
}
const char* recv_message() {
return pipeio_recv_msg(&this->preader);
}
private:
pipe_reader preader;
std::function<void(std::string)> message_handler = [](std::string){};
};
class PipeWriter {
public:
PipeWriter(const std::string& pipe_path)
:pwriter(pipeio_new_writer(pipe_path.c_str())) {}
virtual ~PipeWriter() {
pipeio_close_writer(&this->pwriter);
}
PipeWriter(PipeReader&) = delete;
PipeWriter(PipeReader&&) = delete;
PipeWriter& operator=(const PipeReader&) = delete;
PipeWriter& operator=(const PipeReader&&) = delete;
void send_message(std::string msg) {
pipeio_send_msg(&this->pwriter, msg.c_str());
}
private:
pipe_writer pwriter;
};
class PipeIO {
public:
PipeReader reader;
PipeWriter writer;
PipeIO(const std::string& in_pipe_path, const std::string& out_pipe_path)
:reader(in_pipe_path), writer(out_pipe_path) {}
PipeIO(PipeReader&) = delete;
PipeIO(PipeReader&&) = delete;
PipeIO& operator=(const PipeReader&) = delete;
PipeIO& operator=(const PipeReader&&) = delete;
};

View File

@ -0,0 +1,14 @@
using System;
using Pipe.IO;
class Program
{
static void Main(string[] args)
{
using (var t = new PipeIO("in", "out"))
{
t.Reader.OnMessage += (String msg) => t.Writer.SendMessage(msg);
t.Reader.Run();
}
}
}

View File

@ -0,0 +1,83 @@
using System;
using System.IO;
namespace Pipe.IO
{
public class PipeReader : IDisposable
{
public delegate void MessageHandler(String msg);
public event MessageHandler OnMessage;
private StreamReader pipe;
public PipeReader(String pipe_path)
{
var fileStream = File.OpenRead(pipe_path);
this.pipe = new StreamReader(fileStream);
}
public void Dispose()
{
this.pipe.Close();
}
public void Run()
{
String msg;
while ((msg = this.RecvMessage()) != null)
{
if (this.OnMessage != null)
{
this.OnMessage(msg);
}
}
}
public String RecvMessage()
{
return this.pipe.ReadLine();
}
}
public class PipeWriter : IDisposable
{
private StreamWriter pipe;
public PipeWriter(String pipe)
{
var fileStream = File.OpenWrite(pipe);
this.pipe = new StreamWriter(fileStream);
}
public void Dispose()
{
this.pipe.Close();
}
public void SendMessage(String msg)
{
this.pipe.WriteLine(msg);
this.pipe.Flush();
}
}
public class PipeIO : IDisposable
{
public PipeReader Reader;
public PipeWriter Writer;
public PipeIO(String in_pipe_path, String out_pipe_path)
{
this.Reader = new PipeReader(in_pipe_path);
this.Writer = new PipeWriter(out_pipe_path);
}
public void Dispose()
{
this.Reader.Dispose();
this.Writer.Dispose();
}
}
}

View File

@ -0,0 +1,20 @@
package main
import (
"log"
"./pipeio"
)
func main() {
pipeio, err := pipeio.NewPipeIO("in", "out")
if err != nil {
log.Fatal(err)
}
defer pipeio.Close()
pipeio.Reader.SetMessageHandler(func(msg []byte) {
pipeio.Writer.SendMessage(msg)
})
pipeio.Reader.Run()
}

View File

@ -0,0 +1,116 @@
package pipeio
import (
"os"
"io"
"bufio"
"errors"
)
const defaultBufSize = 65536
type pipeReader struct {
pipe *os.File
scanner *bufio.Scanner
messageHandler messageHandlerFunc
}
type messageHandlerFunc func([]byte)
func NewPipeReader(pipePath string) (*pipeReader, error) {
pipe, err := os.Open(pipePath)
scanner := bufio.NewScanner(pipe)
scanner.Split(bufio.ScanLines)
if err != nil {
return nil, err
}
scanner.Buffer(make([]byte, defaultBufSize), defaultBufSize)
return &pipeReader{
pipe: pipe,
scanner: scanner,
messageHandler: func([]byte){},
}, nil
}
func (preader *pipeReader) Close() {
preader.pipe.Close()
}
func (preader *pipeReader) SetMessageHandler(fun messageHandlerFunc) {
preader.messageHandler = fun
}
func (preader *pipeReader) Run() {
for {
msg := preader.RecvMessage()
if msg == nil {
break
}
preader.messageHandler(msg)
}
}
func (preader *pipeReader) RecvMessage() []byte {
if preader.scanner.Scan() {
return preader.scanner.Bytes()
}
if err := preader.scanner.Err(); err != nil && err != io.EOF {
if err == bufio.ErrTooLong {
err = errors.New("pipeReader buffer too small!")
}
panic(err)
}
return nil
}
func (preader *pipeReader) BufSize(size int) {
preader.scanner.Buffer(make([]byte, defaultBufSize), size)
}
type pipeWriter struct {
pipe *os.File
}
func NewPipeWriter(pipePath string) (*pipeWriter, error) {
pipe, err := os.OpenFile(pipePath, os.O_WRONLY, 0644)
if err != nil {
return nil, err
}
return &pipeWriter{pipe: pipe}, nil
}
func (pwriter *pipeWriter) Close() {
pwriter.pipe.Close()
}
func (pwriter *pipeWriter) SendMessage(msg []byte) {
pwriter.pipe.Write(append(msg, '\n'))
}
type pipeIO struct {
Reader pipeReader
Writer pipeWriter
}
func NewPipeIO(inPipePath, outPipePath string) (*pipeIO, error) {
reader, err := NewPipeReader(inPipePath)
if err != nil {
return nil, err
}
writer, err := NewPipeWriter(outPipePath)
if err != nil {
return nil, err
}
return &pipeIO{
Reader: *reader,
Writer: *writer,
}, nil
}
func (pipeio *pipeIO) Close() {
pipeio.Reader.Close()
pipeio.Writer.Close()
}

View File

@ -0,0 +1,13 @@
public class Example {
public static void main(String[] args) {
try (PipeIO pipeio = new PipeIO("in", "out")) {
pipeio.reader.setMessageHandler(new MessageHandler(){
@Override
public void call(String msg) {
pipeio.writer.sendMessage(msg);
}
});
pipeio.reader.run();
}
}
}

View File

@ -0,0 +1,3 @@
public interface MessageHandler {
public void call(String msg);
}

View File

@ -0,0 +1,17 @@
import java.io.IOException;
public class PipeIO implements AutoCloseable {
public PipeReader reader;
public PipeWriter writer;
public PipeIO(String in_pipe_path, String out_pipe_path) {
this.reader = new PipeReader(in_pipe_path);
this.writer = new PipeWriter(out_pipe_path);
}
public void close() {
this.reader.close();
this.writer.close();
}
}

View File

@ -0,0 +1,50 @@
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.lang.AutoCloseable;
public class PipeReader implements AutoCloseable {
private BufferedReader pipe;
private MessageHandler messageHandler;
public PipeReader(String pipe_path) {
try {
this.pipe = new BufferedReader(new FileReader(pipe_path));
} catch (IOException e) {
throw new UncheckedIOException(e);
}
this.messageHandler = new MessageHandler(){
@Override
public void call(String msg) {}
};
}
public void close() {
try {
this.pipe.close();
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
public void setMessageHandler(MessageHandler handler) {
this.messageHandler = handler;
}
public void run() {
String msg;
while ((msg = this.recvMessage()) != null) {
this.messageHandler.call(msg);
}
}
public String recvMessage() {
try {
return this.pipe.readLine();
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
}

View File

@ -0,0 +1,35 @@
import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.lang.AutoCloseable;
public class PipeWriter implements AutoCloseable {
private BufferedWriter pipe;
public PipeWriter(String pipe_path) {
try {
this.pipe = new BufferedWriter(new FileWriter(pipe_path));
} catch(IOException e) {
throw new UncheckedIOException(e);
}
}
public void close() {
try {
this.pipe.close();
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
public void sendMessage(String msg) {
try {
this.pipe.write(msg + "\n");
this.pipe.flush();
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
}

View File

@ -0,0 +1,7 @@
from pipe_io import PipeIO
if __name__ == '__main__':
with PipeIO('in', 'out') as pipeio:
pipeio.reader.message_handler = pipeio.writer.send_message
pipeio.reader.run()

View File

@ -0,0 +1,69 @@
from typing import Callable
class PipeReader:
def __init__(self, pipe_path):
self._pipe = open(pipe_path, 'rb')
self._message_handler = lambda msg: None
def __enter__(self):
return self
def __exit__(self, type_, value, traceback):
self.close()
def close(self):
self._pipe.close()
@property
def message_handler(self):
return self._message_handler
@message_handler.setter
def message_handler(self, value):
if not isinstance(value, Callable):
raise ValueError("message_handler must be callable!")
self._message_handler = value
def run(self):
msg = self.recv_message()
while msg:
self._message_handler(msg)
msg = self.recv_message()
def recv_message(self):
return self._pipe.readline()[:-1]
class PipeWriter:
def __init__(self, pipe_path):
self._pipe = open(pipe_path, 'wb')
def __enter__(self):
return self
def __exit__(self, type_, value, traceback):
self.close()
def close(self):
self._pipe.close()
def send_message(self, message):
self._pipe.write(message + b'\n')
self._pipe.flush()
class PipeIO:
def __init__(self, in_pipe_path, out_pipe_path):
self.reader = PipeReader(in_pipe_path)
self.writer = PipeWriter(out_pipe_path)
def __enter__(self):
return self
def __exit__(self, type_, value, traceback):
self.close()
def close(self):
self.reader.close()
self.writer.close()