123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824 |
- package iris
- import (
- // std packages
- stdContext "context"
- "io"
- "log"
- "net"
- "net/http"
- "sync"
- "time"
- "github.com/kataras/golog"
- // context for the handlers
- "github.com/kataras/iris/context"
- // core packages, needed to build the application
- "github.com/kataras/iris/core/errors"
- "github.com/kataras/iris/core/host"
- "github.com/kataras/iris/core/maintenance"
- "github.com/kataras/iris/core/netutil"
- "github.com/kataras/iris/core/router"
- // handlerconv conversions
- "github.com/kataras/iris/core/handlerconv"
- // cache conversions
- "github.com/kataras/iris/cache"
- // view
- "github.com/kataras/iris/view"
- // middleware used in Default method
- requestLogger "github.com/kataras/iris/middleware/logger"
- "github.com/kataras/iris/middleware/recover"
- )
- var (
- // Version is the current version number of the Iris Web Framework.
- Version = maintenance.Version
- )
- // HTTP status codes as registered with IANA.
- // See: http://www.iana.org/assignments/http-status-codes/http-status-codes.xhtml
- // Raw Copy from the net/http std package in order to recude the import path of "net/http" for the users.
- //
- // Copied from `net/http` package.
- const (
- StatusContinue = 100 // RFC 7231, 6.2.1
- StatusSwitchingProtocols = 101 // RFC 7231, 6.2.2
- StatusProcessing = 102 // RFC 2518, 10.1
- StatusOK = 200 // RFC 7231, 6.3.1
- StatusCreated = 201 // RFC 7231, 6.3.2
- StatusAccepted = 202 // RFC 7231, 6.3.3
- StatusNonAuthoritativeInfo = 203 // RFC 7231, 6.3.4
- StatusNoContent = 204 // RFC 7231, 6.3.5
- StatusResetContent = 205 // RFC 7231, 6.3.6
- StatusPartialContent = 206 // RFC 7233, 4.1
- StatusMultiStatus = 207 // RFC 4918, 11.1
- StatusAlreadyReported = 208 // RFC 5842, 7.1
- StatusIMUsed = 226 // RFC 3229, 10.4.1
- StatusMultipleChoices = 300 // RFC 7231, 6.4.1
- StatusMovedPermanently = 301 // RFC 7231, 6.4.2
- StatusFound = 302 // RFC 7231, 6.4.3
- StatusSeeOther = 303 // RFC 7231, 6.4.4
- StatusNotModified = 304 // RFC 7232, 4.1
- StatusUseProxy = 305 // RFC 7231, 6.4.5
- _ = 306 // RFC 7231, 6.4.6 (Unused)
- StatusTemporaryRedirect = 307 // RFC 7231, 6.4.7
- StatusPermanentRedirect = 308 // RFC 7538, 3
- StatusBadRequest = 400 // RFC 7231, 6.5.1
- StatusUnauthorized = 401 // RFC 7235, 3.1
- StatusPaymentRequired = 402 // RFC 7231, 6.5.2
- StatusForbidden = 403 // RFC 7231, 6.5.3
- StatusNotFound = 404 // RFC 7231, 6.5.4
- StatusMethodNotAllowed = 405 // RFC 7231, 6.5.5
- StatusNotAcceptable = 406 // RFC 7231, 6.5.6
- StatusProxyAuthRequired = 407 // RFC 7235, 3.2
- StatusRequestTimeout = 408 // RFC 7231, 6.5.7
- StatusConflict = 409 // RFC 7231, 6.5.8
- StatusGone = 410 // RFC 7231, 6.5.9
- StatusLengthRequired = 411 // RFC 7231, 6.5.10
- StatusPreconditionFailed = 412 // RFC 7232, 4.2
- StatusRequestEntityTooLarge = 413 // RFC 7231, 6.5.11
- StatusRequestURITooLong = 414 // RFC 7231, 6.5.12
- StatusUnsupportedMediaType = 415 // RFC 7231, 6.5.13
- StatusRequestedRangeNotSatisfiable = 416 // RFC 7233, 4.4
- StatusExpectationFailed = 417 // RFC 7231, 6.5.14
- StatusTeapot = 418 // RFC 7168, 2.3.3
- StatusUnprocessableEntity = 422 // RFC 4918, 11.2
- StatusLocked = 423 // RFC 4918, 11.3
- StatusFailedDependency = 424 // RFC 4918, 11.4
- StatusUpgradeRequired = 426 // RFC 7231, 6.5.15
- StatusPreconditionRequired = 428 // RFC 6585, 3
- StatusTooManyRequests = 429 // RFC 6585, 4
- StatusRequestHeaderFieldsTooLarge = 431 // RFC 6585, 5
- StatusUnavailableForLegalReasons = 451 // RFC 7725, 3
- StatusInternalServerError = 500 // RFC 7231, 6.6.1
- StatusNotImplemented = 501 // RFC 7231, 6.6.2
- StatusBadGateway = 502 // RFC 7231, 6.6.3
- StatusServiceUnavailable = 503 // RFC 7231, 6.6.4
- StatusGatewayTimeout = 504 // RFC 7231, 6.6.5
- StatusHTTPVersionNotSupported = 505 // RFC 7231, 6.6.6
- StatusVariantAlsoNegotiates = 506 // RFC 2295, 8.1
- StatusInsufficientStorage = 507 // RFC 4918, 11.5
- StatusLoopDetected = 508 // RFC 5842, 7.2
- StatusNotExtended = 510 // RFC 2774, 7
- StatusNetworkAuthenticationRequired = 511 // RFC 6585, 6
- )
- // HTTP Methods copied from `net/http`.
- const (
- MethodGet = "GET"
- MethodPost = "POST"
- MethodPut = "PUT"
- MethodDelete = "DELETE"
- MethodConnect = "CONNECT"
- MethodHead = "HEAD"
- MethodPatch = "PATCH"
- MethodOptions = "OPTIONS"
- MethodTrace = "TRACE"
- )
- // MethodNone is an iris-specific "virtual" method
- // to store the "offline" routes.
- const MethodNone = "NONE"
- // Application is responsible to manage the state of the application.
- // It contains and handles all the necessary parts to create a fast web server.
- type Application struct {
- // routing embedded | exposing APIBuilder's and Router's public API.
- *router.APIBuilder
- *router.Router
- ContextPool *context.Pool
- // config contains the configuration fields
- // all fields defaults to something that is working, developers don't have to set it.
- config *Configuration
- // the golog logger instance, defaults to "Info" level messages (all except "Debug")
- logger *golog.Logger
- // view engine
- view view.View
- // used for build
- once sync.Once
- mu sync.Mutex
- // Hosts contains a list of all servers (Host Supervisors) that this app is running on.
- //
- // Hosts may be empty only if application ran(`app.Run`) with `iris.Raw` option runner,
- // otherwise it contains a single host (`app.Hosts[0]`).
- //
- // Additional Host Supervisors can be added to that list by calling the `app.NewHost` manually.
- //
- // Hosts field is available after `Run` or `NewHost`.
- Hosts []*host.Supervisor
- hostConfigurators []host.Configurator
- }
- // New creates and returns a fresh empty iris *Application instance.
- func New() *Application {
- config := DefaultConfiguration()
- app := &Application{
- config: &config,
- logger: golog.Default,
- APIBuilder: router.NewAPIBuilder(),
- Router: router.NewRouter(),
- }
- app.ContextPool = context.New(func() context.Context {
- return context.NewContext(app)
- })
- return app
- }
- // Default returns a new Application instance which, unlike `New`,
- // recovers on panics and logs the incoming http requests.
- func Default() *Application {
- app := New()
- app.Use(recover.New())
- app.Use(requestLogger.New())
- return app
- }
- // WWW creates and returns a "www." subdomain.
- // The difference from `app.Subdomain("www")` or `app.Party("www.")` is that the `app.WWW()` method
- // wraps the router so all http(s)://mydomain.com will be redirect to http(s)://www.mydomain.com.
- // Other subdomains can be registered using the app: `sub := app.Subdomain("mysubdomain")`,
- // child subdomains can be registered using the www := app.WWW(); www.Subdomain("wwwchildSubdomain").
- func (app *Application) WWW() router.Party {
- return app.SubdomainRedirect(app, app.Subdomain("www"))
- }
- // SubdomainRedirect registers a router wrapper which
- // redirects(StatusMovedPermanently) a (sub)domain to another subdomain or to the root domain as fast as possible,
- // before the router's try to execute route's handler(s).
- //
- // It receives two arguments, they are the from and to/target locations,
- // 'from' can be a wildcard subdomain as well (app.WildcardSubdomain())
- // 'to' is not allowed to be a wildcard for obvious reasons,
- // 'from' can be the root domain(app) when the 'to' is not the root domain and visa-versa.
- //
- // Usage:
- // www := app.Subdomain("www") <- same as app.Party("www.")
- // app.SubdomainRedirect(app, www)
- // This will redirect all http(s)://mydomain.com/%anypath% to http(s)://www.mydomain.com/%anypath%.
- //
- // One or more subdomain redirects can be used to the same app instance.
- //
- // If you need more information about this implementation then you have to navigate through
- // the `core/router#NewSubdomainRedirectWrapper` function instead.
- //
- // Example: https://github.com/kataras/iris/tree/master/_examples/subdomains/redirect
- func (app *Application) SubdomainRedirect(from, to router.Party) router.Party {
- sd := router.NewSubdomainRedirectWrapper(app.ConfigurationReadOnly().GetVHost, from.GetRelPath(), to.GetRelPath())
- app.WrapRouter(sd)
- return to
- }
- // Configure can called when modifications to the framework instance needed.
- // It accepts the framework instance
- // and returns an error which if it's not nil it's printed to the logger.
- // See configuration.go for more.
- //
- // Returns itself in order to be used like `app:= New().Configure(...)`
- func (app *Application) Configure(configurators ...Configurator) *Application {
- for _, cfg := range configurators {
- cfg(app)
- }
- return app
- }
- // ConfigurationReadOnly returns an object which doesn't allow field writing.
- func (app *Application) ConfigurationReadOnly() context.ConfigurationReadOnly {
- return app.config
- }
- // Logger returns the golog logger instance(pointer) that is being used inside the "app".
- //
- // Available levels:
- // - "disable"
- // - "fatal"
- // - "error"
- // - "warn"
- // - "info"
- // - "debug"
- // Usage: app.Logger().SetLevel("error")
- // Defaults to "info" level.
- //
- // Callers can use the application's logger which is
- // the same `golog.Default` logger,
- // to print custom logs too.
- // Usage:
- // app.Logger().Error/Errorf("...")
- // app.Logger().Warn/Warnf("...")
- // app.Logger().Info/Infof("...")
- // app.Logger().Debug/Debugf("...")
- //
- // Setting one or more outputs: app.Logger().SetOutput(io.Writer...)
- // Adding one or more outputs : app.Logger().AddOutput(io.Writer...)
- //
- // Adding custom levels requires import of the `github.com/kataras/golog` package:
- // First we create our level to a golog.Level
- // in order to be used in the Log functions.
- // var SuccessLevel golog.Level = 6
- // Register our level, just three fields.
- // golog.Levels[SuccessLevel] = &golog.LevelMetadata{
- // Name: "success",
- // RawText: "[SUCC]",
- // // ColorfulText (Green Color[SUCC])
- // ColorfulText: "\x1b[32m[SUCC]\x1b[0m",
- // }
- // Usage:
- // app.Logger().SetLevel("success")
- // app.Logger().Logf(SuccessLevel, "a custom leveled log message")
- func (app *Application) Logger() *golog.Logger {
- return app.logger
- }
- var (
- // HTML view engine.
- // Conversion for the view.HTML.
- HTML = view.HTML
- // Django view engine.
- // Conversion for the view.Django.
- Django = view.Django
- // Handlebars view engine.
- // Conversion for the view.Handlebars.
- Handlebars = view.Handlebars
- // Pug view engine.
- // Conversion for the view.Pug.
- Pug = view.Pug
- // Amber view engine.
- // Conversion for the view.Amber.
- Amber = view.Amber
- )
- // NoLayout to disable layout for a particular template file
- // A shortcut for the `view#NoLayout`.
- const NoLayout = view.NoLayout
- // RegisterView should be used to register view engines mapping to a root directory
- // and the template file(s) extension.
- func (app *Application) RegisterView(viewEngine view.Engine) {
- app.view.Register(viewEngine)
- }
- // View executes and writes the result of a template file to the writer.
- //
- // First parameter is the writer to write the parsed template.
- // Second parameter is the relative, to templates directory, template filename, including extension.
- // Third parameter is the layout, can be empty string.
- // Forth parameter is the bindable data to the template, can be nil.
- //
- // Use context.View to render templates to the client instead.
- // Returns an error on failure, otherwise nil.
- func (app *Application) View(writer io.Writer, filename string, layout string, bindingData interface{}) error {
- if app.view.Len() == 0 {
- err := errors.New("view engine is missing, use `RegisterView`")
- app.Logger().Error(err)
- return err
- }
- err := app.view.ExecuteWriter(writer, filename, layout, bindingData)
- if err != nil {
- app.Logger().Error(err)
- }
- return err
- }
- var (
- // LimitRequestBodySize is a middleware which sets a request body size limit
- // for all next handlers in the chain.
- //
- // A shortcut for the `context#LimitRequestBodySize`.
- LimitRequestBodySize = context.LimitRequestBodySize
- // StaticEmbeddedHandler returns a Handler which can serve
- // embedded into executable files.
- //
- //
- // Examples: https://github.com/kataras/iris/tree/master/_examples/file-server
- StaticEmbeddedHandler = router.StaticEmbeddedHandler
- // StripPrefix returns a handler that serves HTTP requests
- // by removing the given prefix from the request URL's Path
- // and invoking the handler h. StripPrefix handles a
- // request for a path that doesn't begin with prefix by
- // replying with an HTTP 404 not found error.
- //
- // Usage:
- // fileserver := Party#StaticHandler("./static_files", false, false)
- // h := iris.StripPrefix("/static", fileserver)
- // app.Get("/static/{f:path}", h)
- // app.Head("/static/{f:path}", h)
- StripPrefix = router.StripPrefix
- // Gzip is a middleware which enables writing
- // using gzip compression, if client supports.
- //
- // A shortcut for the `context#Gzip`.
- Gzip = context.Gzip
- // FromStd converts native http.Handler, http.HandlerFunc & func(w, r, next) to context.Handler.
- //
- // Supported form types:
- // .FromStd(h http.Handler)
- // .FromStd(func(w http.ResponseWriter, r *http.Request))
- // .FromStd(func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc))
- //
- // A shortcut for the `handlerconv#FromStd`.
- FromStd = handlerconv.FromStd
- // Cache is a middleware providing server-side cache functionalities
- // to the next handlers, can be used as: `app.Get("/", iris.Cache, aboutHandler)`.
- // It should be used after Static methods.
- // See `iris#Cache304` for an alternative, faster way.
- //
- // Examples can be found at: https://github.com/kataras/iris/tree/master/_examples/#caching
- Cache = cache.Handler
- // NoCache is a middleware which overrides the Cache-Control, Pragma and Expires headers
- // in order to disable the cache during the browser's back and forward feature.
- //
- // A good use of this middleware is on HTML routes; to refresh the page even on "back" and "forward" browser's arrow buttons.
- //
- // See `iris#StaticCache` for the opposite behavior.
- //
- // A shortcut of the `cache#NoCache`
- NoCache = cache.NoCache
- // StaticCache middleware for caching static files by sending the "Cache-Control" and "Expires" headers to the client.
- // It accepts a single input parameter, the "cacheDur", a time.Duration that it's used to calculate the expiration.
- //
- // If "cacheDur" <=0 then it returns the `NoCache` middleware instaed to disable the caching between browser's "back" and "forward" actions.
- //
- // Usage: `app.Use(iris.StaticCache(24 * time.Hour))` or `app.Use(iris.Staticcache(-1))`.
- // A middleware, which is a simple Handler can be called inside another handler as well, example:
- // cacheMiddleware := iris.StaticCache(...)
- // func(ctx iris.Context){
- // cacheMiddleware(ctx)
- // [...]
- // }
- //
- // A shortcut of the `cache#StaticCache`
- StaticCache = cache.StaticCache
- // Cache304 sends a `StatusNotModified` (304) whenever
- // the "If-Modified-Since" request header (time) is before the
- // time.Now() + expiresEvery (always compared to their UTC values).
- // Use this, which is a shortcut of the, `chache#Cache304` instead of the "github.com/kataras/iris/cache" or iris.Cache
- // for better performance.
- // Clients that are compatible with the http RCF (all browsers are and tools like postman)
- // will handle the caching.
- // The only disadvantage of using that instead of server-side caching
- // is that this method will send a 304 status code instead of 200,
- // So, if you use it side by side with other micro services
- // you have to check for that status code as well for a valid response.
- //
- // Developers are free to extend this method's behavior
- // by watching system directories changes manually and use of the `ctx.WriteWithExpiration`
- // with a "modtime" based on the file modified date,
- // simillary to the `StaticWeb`(which sends status OK(200) and browser disk caching instead of 304).
- //
- // A shortcut of the `cache#Cache304`.
- Cache304 = cache.Cache304
- // CookiePath is a `CookieOption`.
- // Use it to change the cookie's Path field.
- //
- // A shortcut for the `context#CookiePath`.
- CookiePath = context.CookiePath
- // CookieCleanPath is a `CookieOption`.
- // Use it to clear the cookie's Path field, exactly the same as `CookiePath("")`.
- //
- // A shortcut for the `context#CookieCleanPath`.
- CookieCleanPath = context.CookieCleanPath
- // CookieExpires is a `CookieOption`.
- // Use it to change the cookie's Expires and MaxAge fields by passing the lifetime of the cookie.
- //
- // A shortcut for the `context#CookieExpires`.
- CookieExpires = context.CookieExpires
- // CookieHTTPOnly is a `CookieOption`.
- // Use it to set the cookie's HttpOnly field to false or true.
- // HttpOnly field defaults to true for `RemoveCookie` and `SetCookieKV`.
- //
- // A shortcut for the `context#CookieHTTPOnly`.
- CookieHTTPOnly = context.CookieHTTPOnly
- // CookieEncode is a `CookieOption`.
- // Provides encoding functionality when adding a cookie.
- // Accepts a `context#CookieEncoder` and sets the cookie's value to the encoded value.
- // Users of that is the `context#SetCookie` and `context#SetCookieKV`.
- //
- // Example: https://github.com/kataras/iris/tree/master/_examples/cookies/securecookie
- //
- // A shortcut for the `context#CookieEncode`.
- CookieEncode = context.CookieEncode
- // CookieDecode is a `CookieOption`.
- // Provides decoding functionality when retrieving a cookie.
- // Accepts a `context#CookieDecoder` and sets the cookie's value to the decoded value before return by the `GetCookie`.
- // User of that is the `context#GetCookie`.
- //
- // Example: https://github.com/kataras/iris/tree/master/_examples/cookies/securecookie
- //
- // A shortcut for the `context#CookieDecode`.
- CookieDecode = context.CookieDecode
- )
- // SPA accepts an "assetHandler" which can be the result of an
- // app.StaticHandler or app.StaticEmbeddedHandler.
- // Use that when you want to navigate from /index.html to / automatically
- // it's a helper function which just makes some checks based on the `IndexNames` and `AssetValidators`
- // before the assetHandler call.
- //
- // Example: https://github.com/kataras/iris/tree/master/_examples/file-server/single-page-application
- func (app *Application) SPA(assetHandler context.Handler) *router.SPABuilder {
- s := router.NewSPABuilder(assetHandler)
- app.APIBuilder.HandleMany("GET HEAD", "/{f:path}", s.Handler)
- return s
- }
- // ConfigureHost accepts one or more `host#Configuration`, these configurators functions
- // can access the host created by `app.Run`,
- // they're being executed when application is ready to being served to the public.
- //
- // It's an alternative way to interact with a host that is automatically created by
- // `app.Run`.
- //
- // These "configurators" can work side-by-side with the `iris#Addr, iris#Server, iris#TLS, iris#AutoTLS, iris#Listener`
- // final arguments("hostConfigs") too.
- //
- // Note that these application's host "configurators" will be shared with the rest of
- // the hosts that this app will may create (using `app.NewHost`), meaning that
- // `app.NewHost` will execute these "configurators" everytime that is being called as well.
- //
- // These "configurators" should be registered before the `app.Run` or `host.Serve/Listen` functions.
- func (app *Application) ConfigureHost(configurators ...host.Configurator) *Application {
- app.mu.Lock()
- app.hostConfigurators = append(app.hostConfigurators, configurators...)
- app.mu.Unlock()
- return app
- }
- // NewHost accepts a standar *http.Server object,
- // completes the necessary missing parts of that "srv"
- // and returns a new, ready-to-use, host (supervisor).
- func (app *Application) NewHost(srv *http.Server) *host.Supervisor {
- app.mu.Lock()
- defer app.mu.Unlock()
- // set the server's handler to the framework's router
- if srv.Handler == nil {
- srv.Handler = app.Router
- }
- // check if different ErrorLog provided, if not bind it with the framework's logger
- if srv.ErrorLog == nil {
- srv.ErrorLog = log.New(app.logger.Printer.Output, "[HTTP Server] ", 0)
- }
- if srv.Addr == "" {
- srv.Addr = ":8080"
- }
- app.logger.Debugf("Host: addr is %s", srv.Addr)
- // create the new host supervisor
- // bind the constructed server and return it
- su := host.New(srv)
- if app.config.vhost == "" { // vhost now is useful for router subdomain on wildcard subdomains,
- // in order to correct decide what to do on:
- // mydomain.com -> invalid
- // localhost -> invalid
- // sub.mydomain.com -> valid
- // sub.localhost -> valid
- // we need the host (without port if 80 or 443) in order to validate these, so:
- app.config.vhost = netutil.ResolveVHost(srv.Addr)
- }
- app.logger.Debugf("Host: virtual host is %s", app.config.vhost)
- // the below schedules some tasks that will run among the server
- if !app.config.DisableStartupLog {
- // show the available info to exit from app.
- su.RegisterOnServe(host.WriteStartupLogOnServe(app.logger.Printer.Output)) // app.logger.Writer -> Info
- app.logger.Debugf("Host: register startup notifier")
- }
- if !app.config.DisableInterruptHandler {
- // when CTRL+C/CMD+C pressed.
- shutdownTimeout := 5 * time.Second
- host.RegisterOnInterrupt(host.ShutdownOnInterrupt(su, shutdownTimeout))
- app.logger.Debugf("Host: register server shutdown on interrupt(CTRL+C/CMD+C)")
- }
- su.IgnoredErrors = append(su.IgnoredErrors, app.config.IgnoreServerErrors...)
- if len(su.IgnoredErrors) > 0 {
- app.logger.Debugf("Host: server will ignore the following errors: %s", su.IgnoredErrors)
- }
- su.Configure(app.hostConfigurators...)
- app.Hosts = append(app.Hosts, su)
- return su
- }
- // RegisterOnInterrupt registers a global function to call when CTRL+C/CMD+C pressed or a unix kill command received.
- //
- // A shortcut for the `host#RegisterOnInterrupt`.
- var RegisterOnInterrupt = host.RegisterOnInterrupt
- // Shutdown gracefully terminates all the application's server hosts.
- // Returns an error on the first failure, otherwise nil.
- func (app *Application) Shutdown(ctx stdContext.Context) error {
- for i, su := range app.Hosts {
- app.logger.Debugf("Host[%d]: Shutdown now", i)
- if err := su.Shutdown(ctx); err != nil {
- app.logger.Debugf("Host[%d]: Error while trying to shutdown", i)
- return err
- }
- }
- return nil
- }
- // Runner is just an interface which accepts the framework instance
- // and returns an error.
- //
- // It can be used to register a custom runner with `Run` in order
- // to set the framework's server listen action.
- //
- // Currently Runner is being used to declare the built'n server listeners.
- //
- // See `Run` for more.
- type Runner func(*Application) error
- // Listener can be used as an argument for the `Run` method.
- // It can start a server with a custom net.Listener via server's `Serve`.
- //
- // Second argument is optional, it accepts one or more
- // `func(*host.Configurator)` that are being executed
- // on that specific host that this function will create to start the server.
- // Via host configurators you can configure the back-end host supervisor,
- // i.e to add events for shutdown, serve or error.
- // An example of this use case can be found at:
- // https://github.com/kataras/iris/blob/master/_examples/http-listening/notify-on-shutdown/main.go
- // Look at the `ConfigureHost` too.
- //
- // See `Run` for more.
- func Listener(l net.Listener, hostConfigs ...host.Configurator) Runner {
- return func(app *Application) error {
- app.config.vhost = netutil.ResolveVHost(l.Addr().String())
- return app.NewHost(&http.Server{Addr: l.Addr().String()}).
- Configure(hostConfigs...).
- Serve(l)
- }
- }
- // Server can be used as an argument for the `Run` method.
- // It can start a server with a *http.Server.
- //
- // Second argument is optional, it accepts one or more
- // `func(*host.Configurator)` that are being executed
- // on that specific host that this function will create to start the server.
- // Via host configurators you can configure the back-end host supervisor,
- // i.e to add events for shutdown, serve or error.
- // An example of this use case can be found at:
- // https://github.com/kataras/iris/blob/master/_examples/http-listening/notify-on-shutdown/main.go
- // Look at the `ConfigureHost` too.
- //
- // See `Run` for more.
- func Server(srv *http.Server, hostConfigs ...host.Configurator) Runner {
- return func(app *Application) error {
- return app.NewHost(srv).
- Configure(hostConfigs...).
- ListenAndServe()
- }
- }
- // Addr can be used as an argument for the `Run` method.
- // It accepts a host address which is used to build a server
- // and a listener which listens on that host and port.
- //
- // Addr should have the form of [host]:port, i.e localhost:8080 or :8080.
- //
- // Second argument is optional, it accepts one or more
- // `func(*host.Configurator)` that are being executed
- // on that specific host that this function will create to start the server.
- // Via host configurators you can configure the back-end host supervisor,
- // i.e to add events for shutdown, serve or error.
- // An example of this use case can be found at:
- // https://github.com/kataras/iris/blob/master/_examples/http-listening/notify-on-shutdown/main.go
- // Look at the `ConfigureHost` too.
- //
- // See `Run` for more.
- func Addr(addr string, hostConfigs ...host.Configurator) Runner {
- return func(app *Application) error {
- return app.NewHost(&http.Server{Addr: addr}).
- Configure(hostConfigs...).
- ListenAndServe()
- }
- }
- // TLS can be used as an argument for the `Run` method.
- // It will start the Application's secure server.
- //
- // Use it like you used to use the http.ListenAndServeTLS function.
- //
- // Addr should have the form of [host]:port, i.e localhost:443 or :443.
- // CertFile & KeyFile should be filenames with their extensions.
- //
- // Second argument is optional, it accepts one or more
- // `func(*host.Configurator)` that are being executed
- // on that specific host that this function will create to start the server.
- // Via host configurators you can configure the back-end host supervisor,
- // i.e to add events for shutdown, serve or error.
- // An example of this use case can be found at:
- // https://github.com/kataras/iris/blob/master/_examples/http-listening/notify-on-shutdown/main.go
- // Look at the `ConfigureHost` too.
- //
- // See `Run` for more.
- func TLS(addr string, certFile, keyFile string, hostConfigs ...host.Configurator) Runner {
- return func(app *Application) error {
- return app.NewHost(&http.Server{Addr: addr}).
- Configure(hostConfigs...).
- ListenAndServeTLS(certFile, keyFile)
- }
- }
- // AutoTLS can be used as an argument for the `Run` method.
- // It will start the Application's secure server using
- // certifications created on the fly by the "autocert" golang/x package,
- // so localhost may not be working, use it at "production" machine.
- //
- // Addr should have the form of [host]:port, i.e mydomain.com:443.
- //
- // The whitelisted domains are separated by whitespace in "domain" argument,
- // i.e "iris-go.com", can be different than "addr".
- // If empty, all hosts are currently allowed. This is not recommended,
- // as it opens a potential attack where clients connect to a server
- // by IP address and pretend to be asking for an incorrect host name.
- // Manager will attempt to obtain a certificate for that host, incorrectly,
- // eventually reaching the CA's rate limit for certificate requests
- // and making it impossible to obtain actual certificates.
- //
- // For an "e-mail" use a non-public one, letsencrypt needs that for your own security.
- //
- // Note: `AutoTLS` will start a new server for you
- // which will redirect all http versions to their https, including subdomains as well.
- //
- // Last argument is optional, it accepts one or more
- // `func(*host.Configurator)` that are being executed
- // on that specific host that this function will create to start the server.
- // Via host configurators you can configure the back-end host supervisor,
- // i.e to add events for shutdown, serve or error.
- // An example of this use case can be found at:
- // https://github.com/kataras/iris/blob/master/_examples/http-listening/notify-on-shutdown/main.go
- // Look at the `ConfigureHost` too.
- //
- // Usage:
- // app.Run(iris.AutoTLS("iris-go.com:443", "iris-go.com www.iris-go.com", "mail@example.com"))
- //
- // See `Run` and `core/host/Supervisor#ListenAndServeAutoTLS` for more.
- func AutoTLS(
- addr string,
- domain string, email string,
- hostConfigs ...host.Configurator) Runner {
- return func(app *Application) error {
- return app.NewHost(&http.Server{Addr: addr}).
- Configure(hostConfigs...).
- ListenAndServeAutoTLS(domain, email, "letscache")
- }
- }
- // Raw can be used as an argument for the `Run` method.
- // It accepts any (listen) function that returns an error,
- // this function should be block and return an error
- // only when the server exited or a fatal error caused.
- //
- // With this option you're not limited to the servers
- // that iris can run by-default.
- //
- // See `Run` for more.
- func Raw(f func() error) Runner {
- return func(app *Application) error {
- app.logger.Debugf("HTTP Server will start from unknown, external function")
- return f()
- }
- }
- // Build sets up, once, the framework.
- // It builds the default router with its default macros
- // and the template functions that are very-closed to iris.
- func (app *Application) Build() error {
- rp := errors.NewReporter()
- app.once.Do(func() {
- rp.Describe("api builder: %v", app.APIBuilder.GetReport())
- if !app.Router.Downgraded() {
- // router
- // create the request handler, the default routing handler
- routerHandler := router.NewDefaultHandler()
- rp.Describe("router: %v", app.Router.BuildRouter(app.ContextPool, routerHandler, app.APIBuilder))
- // re-build of the router from outside can be done with;
- // app.RefreshRouter()
- }
- if app.view.Len() > 0 {
- app.logger.Debugf("Application: %d registered view engine(s)", app.view.Len())
- // view engine
- // here is where we declare the closed-relative framework functions.
- // Each engine has their defaults, i.e yield,render,render_r,partial, params...
- rv := router.NewRoutePathReverser(app.APIBuilder)
- app.view.AddFunc("urlpath", rv.Path)
- // app.view.AddFunc("url", rv.URL)
- rp.Describe("view: %v", app.view.Load())
- }
- })
- return rp.Return()
- }
- // ErrServerClosed is returned by the Server's Serve, ServeTLS, ListenAndServe,
- // and ListenAndServeTLS methods after a call to Shutdown or Close.
- //
- // A shortcut for the `http#ErrServerClosed`.
- var ErrServerClosed = http.ErrServerClosed
- // Run builds the framework and starts the desired `Runner` with or without configuration edits.
- //
- // Run should be called only once per Application instance, it blocks like http.Server.
- //
- // If more than one server needed to run on the same iris instance
- // then create a new host and run it manually by `go NewHost(*http.Server).Serve/ListenAndServe` etc...
- // or use an already created host:
- // h := NewHost(*http.Server)
- // Run(Raw(h.ListenAndServe), WithCharset("UTF-8"), WithRemoteAddrHeader("CF-Connecting-IP"))
- //
- // The Application can go online with any type of server or iris's host with the help of
- // the following runners:
- // `Listener`, `Server`, `Addr`, `TLS`, `AutoTLS` and `Raw`.
- func (app *Application) Run(serve Runner, withOrWithout ...Configurator) error {
- // first Build because it doesn't need anything from configuration,
- // this gives the user the chance to modify the router inside a configurator as well.
- if err := app.Build(); err != nil {
- return errors.PrintAndReturnErrors(err, app.logger.Errorf)
- }
- app.Configure(withOrWithout...)
- app.logger.Debugf("Application: running using %d host(s)", len(app.Hosts)+1)
- if !app.config.DisableVersionChecker {
- go maintenance.Start()
- }
- // this will block until an error(unless supervisor's DeferFlow called from a Task).
- err := serve(app)
- if err != nil {
- app.Logger().Error(err)
- }
- return err
- }
|