Forráskód Böngészése

initial POC commit

teizz 3 éve
szülő
commit
172abc65f7
3 módosított fájl, 166 hozzáadás és 0 törlés
  1. 15 0
      Dockerfile
  2. 1 0
      VERSION
  3. 150 0
      main.go

+ 15 - 0
Dockerfile

@@ -0,0 +1,15 @@
+FROM golang:alpine AS builder
+RUN apk --no-cache --no-progress add make upx
+WORKDIR /go/src/lolcathost/pathway/
+
+COPY *.go VERSION ./
+RUN echo Building && \
+    env CGO_ENABLED=0 go build -trimpath -ldflags="all=-s -w -buildid= -X main.version=$(cat VERSION) -X main.buildtime=$(date +%FT%T%z)" && \
+    echo Compressing && \
+    upx pathway > /dev/null
+
+FROM scratch
+COPY --from=builder /go/src/lolcathost/pathway/pathway /pathway
+
+EXPOSE 8080
+CMD ["/pathway"]

+ 1 - 0
VERSION

@@ -0,0 +1 @@
+1.0-beta

+ 150 - 0
main.go

@@ -0,0 +1,150 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"strconv"
+	"sync"
+)
+
+var (
+	version   = "0.0-undefined"
+	buildtime = "0000-00-00T00:00:00+0000"
+	queues    = make(map[string]*Queue)
+	mut       = &sync.Mutex{}
+)
+
+type Transfer struct {
+	reader io.Reader
+	size   int
+}
+
+type Queue struct {
+	ch        chan Transfer
+	producers int
+	consumers int
+}
+
+func NewQueue() *Queue {
+	return &Queue{
+		ch: make(chan Transfer),
+	}
+}
+
+func (q *Queue) addConsumer() {
+	q.consumers = q.consumers + 1
+}
+
+func (q *Queue) remConsumer() {
+	q.consumers = q.consumers - 1
+}
+
+func (q *Queue) addProducer() {
+	q.producers = q.producers + 1
+}
+
+func (q *Queue) remProducer() {
+	q.producers = q.producers - 1
+}
+
+func (q *Queue) isEmpty() bool {
+	return q.producers == 0 && q.consumers == 0
+}
+
+func main() {
+	log.Printf("pathway version:%s buildtime:%s", version, buildtime)
+
+	handle := func(w http.ResponseWriter, r *http.Request) {
+		channelId := r.URL.Path
+
+		if r.Method == "GET" {
+			log.Printf("GET: %s", channelId)
+			mut.Lock()
+			queue, exists := queues[channelId]
+			if !exists {
+				queue = NewQueue()
+				queues[channelId] = queue
+			}
+
+			ch := queue.ch
+
+			queue.addConsumer()
+			mut.Unlock()
+
+			select {
+			case transfer := <-ch:
+				w.Header().Set("Content-Length", fmt.Sprintf("%d", transfer.size))
+				_, err := io.Copy(w, transfer.reader)
+				if err != nil {
+					if closer, ok := transfer.reader.(io.Closer); ok {
+						closer.Close()
+					}
+				}
+			case <-r.Context().Done():
+			}
+
+			mut.Lock()
+			queue.remConsumer()
+			if queue.isEmpty() {
+				delete(queues, channelId)
+			}
+			mut.Unlock()
+		} else {
+			log.Printf("POST: %s", channelId)
+			mut.Lock()
+			queue, exists := queues[channelId]
+			if !exists {
+				queue = NewQueue()
+				queues[channelId] = queue
+			}
+
+			ch := queue.ch
+
+			queue.addProducer()
+			mut.Unlock()
+
+			reader, writer := io.Pipe()
+
+			contentLength, err := strconv.Atoi(r.Header.Get("Content-Length"))
+			if err != nil {
+				contentLength = 0
+			}
+
+			transfer := Transfer{
+				reader: reader,
+				size:   contentLength,
+			}
+
+			select {
+			case ch <- transfer:
+				io.Copy(writer, r.Body)
+			case <-r.Context().Done():
+			}
+
+			writer.Close()
+
+			mut.Lock()
+			queue.remProducer()
+			if queue.isEmpty() {
+				delete(queues, channelId)
+			}
+			mut.Unlock()
+		}
+	}
+
+	health := func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		w.Write([]byte("OK"))
+	}
+
+	http.HandleFunc("/health", health)
+	http.HandleFunc("/", handle)
+
+	err := http.ListenAndServe(":8080", nil)
+	if err != nil {
+		log.Println(err)
+	}
+
+}