initial POC commit
This commit is contained in:
15
Dockerfile
Normal file
15
Dockerfile
Normal file
@@ -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"]
|
||||||
150
main.go
Normal file
150
main.go
Normal file
@@ -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)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user