Mirrors/iptv-proxy
1
0
Fork 0
mirror of https://github.com/pierre-emmanuelJ/iptv-proxy.git synced 2026-03-13 14:46:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix cache and improve it (use multiple cache by url and store it in /tmp)

Browse Source 
Signed-off-by: Pierre-Emmanuel Jacquier <pierre-emmanuel.jacquier@exoscale.ch>
This commit is contained in:
Pierre-Emmanuel Jacquier
2019-10-01 21:00:36 +00:00
parent ff179ab21a
commit 468ae7fb64
1 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
pkg/routes/xtream.go
View File

@@ -1,11 +1,13 @@
package routes
import (
"encoding/base64"
"fmt"
"io/ioutil"
"log"
"net/http"
"net/url"
"path/filepath"
"strings"
"sync"
"time"
@@ -18,8 +20,7 @@ import (
// XXX Add one cache per url and store it on the local storage or key/value storage e.g: etcd, redis...
// and remove that dirty globals
var xtreamM3uCache []byte
var xtreamM3uCacheLastURL string
var xtreamM3uCache map[string]string = map[string]string{}
var lock = sync.RWMutex{}
func (p *proxy) cacheXtreamM3u(m3uURL *url.URL) error {
@@ -39,12 +40,27 @@ func (p *proxy) cacheXtreamM3u(m3uURL *url.URL) error {
}
lock.Lock()
xtreamM3uCache = []byte(result)
path, err := writeCacheTmp([]byte(result), m3uURL.String())
if err != nil {
return err
}
xtreamM3uCache[m3uURL.String()] = path
lock.Unlock()
return nil
}
func writeCacheTmp(data []byte, url string) (string, error) {
filename := base64.StdEncoding.EncodeToString([]byte(url))
path := filepath.Join("/tmp", filename)
if err := ioutil.WriteFile(path, data, 0644); err != nil {
return "", err
}
return path, nil
}
func (p *proxy) xtreamGet(c *gin.Context) {
rawURL := fmt.Sprintf("%s/get.php?username=%s&password=%s", p.XtreamBaseURL, p.XtreamUser, p.XtreamPassword)
@@ -66,12 +82,10 @@ func (p *proxy) xtreamGet(c *gin.Context) {
// XXX Add cache per url and store it on the local storage or key/value storage e.g: etcd, redis...
lock.RLock()
if xtreamM3uCacheLastURL != m3uURL.String() {
_, ok := xtreamM3uCache[m3uURL.String()]
if !ok {
log.Printf("[iptv-proxy] %v | %s | xtream cache m3u file\n", time.Now().Format("2006/01/02 - 15:04:05"), c.ClientIP())
lock.RUnlock()
lock.Lock()
xtreamM3uCacheLastURL = m3uURL.String()
lock.Unlock()
if err := p.cacheXtreamM3u(m3uURL); err != nil {
c.AbortWithError(http.StatusInternalServerError, err)
return
@@ -82,8 +96,15 @@ func (p *proxy) xtreamGet(c *gin.Context) {
c.Header("Content-Disposition", "attachment; filename=\"iptv.m3u\"")
lock.RLock()
c.Data(http.StatusOK, "application/octet-stream", xtreamM3uCache)
path := xtreamM3uCache[m3uURL.String()]
lock.RUnlock()
data, err := ioutil.ReadFile(path)
if err != nil {
c.AbortWithError(http.StatusInternalServerError, err)
return
}
c.Data(http.StatusOK, "application/octet-stream", data)
}
func (p *proxy) xtreamPlayerAPIGET(c *gin.Context) {