Simon Eisenmann
9 years ago
21 changed files with 700 additions and 157 deletions
@ -0,0 +1,190 @@
@@ -0,0 +1,190 @@
|
||||
/* |
||||
* Spreed WebRTC. |
||||
* Copyright (C) 2013-2015 struktur AG |
||||
* |
||||
* This file is part of Spreed WebRTC. |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
|
||||
package channelling |
||||
|
||||
import ( |
||||
"bytes" |
||||
"encoding/json" |
||||
"errors" |
||||
"log" |
||||
"sync" |
||||
"time" |
||||
|
||||
"github.com/strukturag/spreed-webrtc/go/buffercache" |
||||
) |
||||
|
||||
type PipelineFeedLine struct { |
||||
Seq int |
||||
Msg *DataOutgoing |
||||
} |
||||
|
||||
type Pipeline struct { |
||||
PipelineManager PipelineManager |
||||
mutex sync.RWMutex |
||||
namespace string |
||||
id string |
||||
from *Session |
||||
expires *time.Time |
||||
data []*DataOutgoing |
||||
sink Sink |
||||
} |
||||
|
||||
func NewPipeline(manager PipelineManager, |
||||
namespace string, |
||||
id string, |
||||
from *Session, |
||||
duration time.Duration) *Pipeline { |
||||
pipeline := &Pipeline{ |
||||
PipelineManager: manager, |
||||
namespace: namespace, |
||||
id: id, |
||||
from: from, |
||||
} |
||||
pipeline.Refresh(duration) |
||||
return pipeline |
||||
} |
||||
|
||||
func (pipeline *Pipeline) GetID() string { |
||||
return pipeline.id |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Refresh(duration time.Duration) { |
||||
pipeline.mutex.Lock() |
||||
expiration := time.Now().Add(duration) |
||||
pipeline.expires = &expiration |
||||
pipeline.mutex.Unlock() |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Add(msg *DataOutgoing) *Pipeline { |
||||
pipeline.mutex.Lock() |
||||
pipeline.data = append(pipeline.data, msg) |
||||
pipeline.mutex.Unlock() |
||||
|
||||
return pipeline |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Send(b buffercache.Buffer) { |
||||
// noop for now
|
||||
} |
||||
|
||||
func (pipeline *Pipeline) Index() uint64 { |
||||
return 0 |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Close() { |
||||
pipeline.mutex.Lock() |
||||
pipeline.expires = nil |
||||
if pipeline.sink != nil { |
||||
pipeline.sink.Close() |
||||
pipeline.sink = nil |
||||
} |
||||
pipeline.mutex.Unlock() |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Expired() bool { |
||||
var expired bool |
||||
pipeline.mutex.RLock() |
||||
if pipeline.expires == nil { |
||||
expired = true |
||||
} else { |
||||
expired = pipeline.expires.Before(time.Now()) |
||||
} |
||||
pipeline.mutex.RUnlock() |
||||
|
||||
return expired |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Session() *Session { |
||||
return pipeline.from |
||||
} |
||||
|
||||
func (pipeline *Pipeline) JSONFeed(since, limit int) ([]byte, error) { |
||||
pipeline.mutex.RLock() |
||||
var lineRaw []byte |
||||
var line *PipelineFeedLine |
||||
var buffer bytes.Buffer |
||||
var err error |
||||
data := pipeline.data[since:] |
||||
count := 0 |
||||
for seq, msg := range data { |
||||
line = &PipelineFeedLine{ |
||||
Seq: seq + since, |
||||
Msg: msg, |
||||
} |
||||
lineRaw, err = json.Marshal(line) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
buffer.Write(lineRaw) |
||||
buffer.WriteString("\n") |
||||
|
||||
count++ |
||||
if limit > 0 && count >= limit { |
||||
break |
||||
} |
||||
} |
||||
pipeline.mutex.RUnlock() |
||||
|
||||
return buffer.Bytes(), nil |
||||
} |
||||
|
||||
func (pipeline *Pipeline) FlushOutgoing(hub Hub, client *Client, to string, outgoing *DataOutgoing) bool { |
||||
pipeline.mutex.RLock() |
||||
log.Println("Flush outgoing via pipeline", to, client == nil) |
||||
if client == nil { |
||||
sink := pipeline.sink |
||||
pipeline.mutex.RUnlock() |
||||
pipeline.Add(outgoing) |
||||
|
||||
// It is possible to retrieve the userid for fake sessions here.
|
||||
if session, found := pipeline.PipelineManager.GetSession(to); found { |
||||
log.Println("Pipeline found userid via manager", session.Userid()) |
||||
} |
||||
|
||||
if sink == nil { |
||||
return true |
||||
} |
||||
// Sink it.
|
||||
pipeline.sink.Write(outgoing) |
||||
return true |
||||
} |
||||
pipeline.mutex.RUnlock() |
||||
|
||||
return false |
||||
} |
||||
|
||||
func (pipeline *Pipeline) Attach(sink Sink) error { |
||||
pipeline.mutex.Lock() |
||||
defer pipeline.mutex.Unlock() |
||||
if pipeline.sink != nil { |
||||
return errors.New("pipeline already attached to sink") |
||||
} |
||||
pipeline.sink = sink |
||||
|
||||
// Sink existing data first.
|
||||
log.Println("Attach sink to pipeline", pipeline.id) |
||||
for _, msg := range pipeline.data { |
||||
sink.Write(msg) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
@ -0,0 +1,120 @@
@@ -0,0 +1,120 @@
|
||||
/* |
||||
* Spreed WebRTC. |
||||
* Copyright (C) 2013-2015 struktur AG |
||||
* |
||||
* This file is part of Spreed WebRTC. |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
|
||||
package channelling |
||||
|
||||
import ( |
||||
"fmt" |
||||
"log" |
||||
"sync" |
||||
"time" |
||||
) |
||||
|
||||
const ( |
||||
PipelineNamespaceCall = "call" |
||||
) |
||||
|
||||
type PipelineManager interface { |
||||
BusManager |
||||
SessionStore |
||||
UserStore |
||||
GetPipelineByID(id string) (pipeline *Pipeline, ok bool) |
||||
GetPipeline(namespace string, sender Sender, session *Session, to string) *Pipeline |
||||
} |
||||
|
||||
type pipelineManager struct { |
||||
BusManager |
||||
SessionStore |
||||
UserStore |
||||
mutex sync.RWMutex |
||||
pipelines map[string]*Pipeline |
||||
duration time.Duration |
||||
} |
||||
|
||||
func NewPipelineManager(busManager BusManager, sessionStore SessionStore, userStore UserStore) PipelineManager { |
||||
plm := &pipelineManager{ |
||||
BusManager: busManager, |
||||
SessionStore: sessionStore, |
||||
UserStore: userStore, |
||||
pipelines: make(map[string]*Pipeline), |
||||
duration: 30 * time.Minute, |
||||
} |
||||
plm.start() |
||||
return plm |
||||
} |
||||
|
||||
func (plm *pipelineManager) cleanup() { |
||||
plm.mutex.Lock() |
||||
for id, pipeline := range plm.pipelines { |
||||
if pipeline.Expired() { |
||||
pipeline.Close() |
||||
delete(plm.pipelines, id) |
||||
} |
||||
} |
||||
plm.mutex.Unlock() |
||||
} |
||||
|
||||
func (plm *pipelineManager) start() { |
||||
c := time.Tick(30 * time.Second) |
||||
go func() { |
||||
for _ = range c { |
||||
plm.cleanup() |
||||
} |
||||
}() |
||||
} |
||||
|
||||
func (plm *pipelineManager) GetPipelineByID(id string) (*Pipeline, bool) { |
||||
plm.mutex.RLock() |
||||
pipeline, ok := plm.pipelines[id] |
||||
if !ok { |
||||
// XXX(longsleep): Hack for development
|
||||
for _, pipeline = range plm.pipelines { |
||||
ok = true |
||||
break |
||||
} |
||||
} |
||||
plm.mutex.RUnlock() |
||||
return pipeline, ok |
||||
} |
||||
|
||||
func (plm *pipelineManager) PipelineID(namespace string, sender Sender, session *Session, to string) string { |
||||
return fmt.Sprintf("%s.%s.%s", namespace, session.Id, to) |
||||
} |
||||
|
||||
func (plm *pipelineManager) GetPipeline(namespace string, sender Sender, session *Session, to string) *Pipeline { |
||||
id := plm.PipelineID(namespace, sender, session, to) |
||||
|
||||
plm.mutex.Lock() |
||||
pipeline, ok := plm.pipelines[id] |
||||
if ok { |
||||
// Refresh. We do not care if the pipeline is expired.
|
||||
pipeline.Refresh(plm.duration) |
||||
plm.mutex.Unlock() |
||||
return pipeline |
||||
} |
||||
|
||||
log.Println("Creating pipeline", namespace, id) |
||||
pipeline = NewPipeline(plm, namespace, id, session, plm.duration) |
||||
plm.pipelines[id] = pipeline |
||||
plm.mutex.Unlock() |
||||
|
||||
return pipeline |
||||
} |
||||
@ -0,0 +1,98 @@
@@ -0,0 +1,98 @@
|
||||
/* |
||||
* Spreed WebRTC. |
||||
* Copyright (C) 2013-2015 struktur AG |
||||
* |
||||
* This file is part of Spreed WebRTC. |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
|
||||
package server |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"net/http" |
||||
"strconv" |
||||
|
||||
"github.com/strukturag/spreed-webrtc/go/channelling" |
||||
|
||||
"github.com/gorilla/mux" |
||||
) |
||||
|
||||
type Pipelines struct { |
||||
channelling.PipelineManager |
||||
API channelling.ChannellingAPI |
||||
} |
||||
|
||||
func (pipelines *Pipelines) Get(request *http.Request) (int, interface{}, http.Header) { |
||||
vars := mux.Vars(request) |
||||
id, ok := vars["id"] |
||||
if !ok { |
||||
return http.StatusNotFound, "", nil |
||||
} |
||||
|
||||
pipeline, ok := pipelines.GetPipelineByID(id) |
||||
if !ok { |
||||
return http.StatusNotFound, "", nil |
||||
} |
||||
|
||||
since := 0 |
||||
limit := 0 |
||||
if sinceParam := request.Form.Get("since"); sinceParam != "" { |
||||
since, _ = strconv.Atoi(sinceParam) |
||||
} |
||||
if limitParam := request.Form.Get("limit"); limitParam != "" { |
||||
limit, _ = strconv.Atoi(limitParam) |
||||
} |
||||
|
||||
result, err := pipeline.JSONFeed(since, limit) |
||||
if err != nil { |
||||
return http.StatusInternalServerError, err.Error(), nil |
||||
} |
||||
|
||||
return http.StatusOK, result, nil |
||||
} |
||||
|
||||
func (pipelines *Pipelines) Post(request *http.Request) (int, interface{}, http.Header) { |
||||
vars := mux.Vars(request) |
||||
id, ok := vars["id"] |
||||
if !ok { |
||||
return http.StatusNotFound, "", nil |
||||
} |
||||
|
||||
pipeline, ok := pipelines.GetPipelineByID(id) |
||||
if !ok { |
||||
return http.StatusNotFound, "", nil |
||||
} |
||||
|
||||
var incoming channelling.DataIncoming |
||||
dec := json.NewDecoder(request.Body) |
||||
if err := dec.Decode(&incoming); err != nil { |
||||
return http.StatusBadRequest, err.Error(), nil |
||||
} |
||||
|
||||
result := &channelling.DataOutgoing{ |
||||
From: pipeline.Session().Id, |
||||
Iid: incoming.Iid, |
||||
} |
||||
reply, err := pipelines.API.OnIncoming(pipeline, pipeline.Session(), &incoming) |
||||
if err == nil { |
||||
result.Data = reply |
||||
} else { |
||||
result.Data = err |
||||
} |
||||
|
||||
return http.StatusOK, result, nil |
||||
} |
||||
@ -0,0 +1,32 @@
@@ -0,0 +1,32 @@
|
||||
/* |
||||
* Spreed WebRTC. |
||||
* Copyright (C) 2013-2015 struktur AG |
||||
* |
||||
* This file is part of Spreed WebRTC. |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
|
||||
package channelling |
||||
|
||||
import () |
||||
|
||||
// Sink connects a Pipeline with end points in both directions by
|
||||
// getting attached to a Pipeline.
|
||||
type Sink interface { |
||||
// Write sends outgoing data on the sink from the
|
||||
Write(interface{}) |
||||
Close() |
||||
} |
||||
@ -0,0 +1,26 @@
@@ -0,0 +1,26 @@
|
||||
/* |
||||
* Spreed WebRTC. |
||||
* Copyright (C) 2013-2015 struktur AG |
||||
* |
||||
* This file is part of Spreed WebRTC. |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
|
||||
package channelling |
||||
|
||||
type UserStore interface { |
||||
GetUser(id string) (user *User, ok bool) |
||||
} |
||||
Loading…
Reference in new issue