From 29c2b0e6af6481af8a589b3cda331cbb71c8c3a8 Mon Sep 17 00:00:00 2001
From: nareix <hi.nareix@gmail.com>
Date: Thu, 30 Jun 2016 19:17:58 +0800
Subject: [PATCH] add handler

---
 rtmp.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/rtmp.go b/rtmp.go
index 87007f7..a64912d 100644
--- a/rtmp.go
+++ b/rtmp.go
@@ -14,6 +14,7 @@ import (
 	"github.com/nareix/pio"
 	"github.com/nareix/flv/flvio"
 	"github.com/nareix/av"
+	"github.com/nareix/av/avutil"
 	"github.com/nareix/codec"
 	"github.com/nareix/codec/h264parser"
 	"github.com/nareix/codec/aacparser"
@@ -1578,3 +1579,14 @@ func (self *Conn) handshakeServer() (err error) {
 	return
 }
 
+func Handler(h *avutil.RegisterHandler) {
+	h.UrlDemuxer = func(uri string) (ok bool, demuxer av.DemuxCloser, err error) {
+		if !strings.HasPrefix(uri, "rtmp://") {
+			return
+		}
+		ok = true
+		demuxer, err = Dial(uri)
+		return
+	}
+}
+