feat: get swagger instance from register (#1298)

Signed-off-by: Eray Ates <eates23@gmail.com>

Signed-off-by: Eray Ates <eates23@gmail.com>

swagger.go

@@ -39,6 +39,15 @@ func Register(name string, swagger Swagger) {
 	swags[name] = swagger
 }
 
+// GetSwagger returns the swagger instance for given name.
+// If not found, returns nil.
+func GetSwagger(name string) Swagger {
+	swaggerMu.RLock()
+	defer swaggerMu.RUnlock()
+
+	return swags[name]
+}
+
 // ReadDoc reads swagger document. An optional name parameter can be passed to read a specific document.
 // The default name is "swagger".
 func ReadDoc(optionalName ...string) (string, error) {

swagger_test.go

@@ -219,3 +219,14 @@ func TestCalledTwicelRegister(t *testing.T) {
 func setup() {
 	swags = nil
 }
+
+func TestGetSwagger(t *testing.T) {
+	setup()
+	instance := &s{}
+	Register(Name, instance)
+	swagger := GetSwagger(Name)
+	assert.Equal(t, instance, swagger)
+
+	swagger = GetSwagger("invalid")
+	assert.Nil(t, swagger)
+}