How can I focus a controller only works on localhost?

Question

I have a site with a controller with so many post and get methods.

Recently, this site will be available to all people. However, I want to maintain this controller only working on localhost.

In spite of this I can add logic to every method, for example:

if (HttpContext.Request.Host.Host.ToLower() != "localhost")
            {
                return NotFound();
            }

I need to add the code above to the method one by one, it is so troublesome.

Is there any way I focus a controller only works on localhost?

Answer 1

Hi @melon NG ,

According to your description, I suggest you could consider writing a custom middleware to achieve your requirement. This middleware will check every request's host name and path value, if it find the host name is localhost and it wants to access the specific controller, then you could directly return the response.

            app.Use(async (context, next) =>  
            {  
                if (context.Request.Host.Host.ToLower() != "localhost" && context.Request.Path.Value.Contains("WeatherForecast"))  
                {  
                    await context.Response.WriteAsync("test");  
                 }  
                else  
                {  
                    await next.Invoke();  
                }  
  
              
  
            });  

More details about how to use middleware, you could refer to this article.