博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
golang martini 源码阅读笔记之martini核心
阅读量:6979 次
发布时间:2019-06-27

本文共 6351 字,大约阅读时间需要 21 分钟。

继上一篇关于inject注入的笔记,理解了martini的关键核心之一:依赖注入。注入回调函数,由运行时进行主动调用执行。这一篇主要是注解martini的骨架martini.go的实现,下面先从一个简单的martini使用实例,即创建一个最简单的http服务器开始。

server.go

//martini使用简单实例 package main import "github.com/go-martini/martini" func main() {   m := martini.Classic() //获取一个martini实例   m.Get("/", func() string { // 用户自定义路由规则     return "Hello world!"   })   m.Run() // 运行服务器 }

 

 martini.go

package martiniimport (	"log"	"net/http"	"os"	"reflect"	"github.com/codegangsta/inject")// Martini represents the top level web application. inject.Injector methods can be invoked to map services on a global level.type Martini struct {	inject.Injector // 注入工具,利用反射实现函数回调	handlers []Handler // 存储处理http请求的所有中间件	action   Handler // 路由匹配以及路由处理,在所有中间件都处理完之后执行	logger   *log.Logger // 日志工具}// 基础骨架:具备基本的注入与反射调用功能// New creates a bare bones Martini instance. Use this method if you want to have full control over the middleware that is used.func New() *Martini {	m := &Martini{Injector: inject.New(), action: func() {}, logger: log.New(os.Stdout, "[martini] ", 0)}	m.Map(m.logger)	m.Map(defaultReturnHandler())	return m}// Handlers sets the entire middleware stack with the given Handlers. This will clear any current middleware handlers.// Will panic if any of the handlers is not a callable function// 设置所有的中间件func (m *Martini) Handlers(handlers ...Handler) {	m.handlers = make([]Handler, 0)	for _, handler := range handlers {		m.Use(handler)	}}// Action sets the handler that will be called after all the middleware has been invoked. This is set to martini.Router in a martini.Classic().// 设置真正的路由处理器,所有中间件执行完之后才会执行func (m *Martini) Action(handler Handler) {	validateHandler(handler)	m.action = handler}// Logger sets the loggerfunc (m *Martini) Logger(logger *log.Logger) {	m.logger = logger	m.Map(m.logger)}// Use adds a middleware Handler to the stack. Will panic if the handler is not a callable func. Middleware Handlers are invoked in the order that they are added.// 添加一个中间件处理器,每一个http请求都会先执行, 按照添加的顺序依次执行func (m *Martini) Use(handler Handler) {	validateHandler(handler)	m.handlers = append(m.handlers, handler)}// http接口,每一次http请求 用户级别处理的入口// ServeHTTP is the HTTP Entry point for a Martini instance. Useful if you want to control your own HTTP server.func (m *Martini) ServeHTTP(res http.ResponseWriter, req *http.Request) {	println("call....")	m.createContext(res, req).run() // 每一个请求创建一个上下文,保存一些必要的信息,之后开始处理请求}// Run the http server on a given host and port.// http 服务器启动func (m *Martini) RunOnAddr(addr string) {	// TODO: Should probably be implemented using a new instance of http.Server in place of	// calling http.ListenAndServer directly, so that it could be stored in the martini struct for later use.	// This would also allow to improve testing when a custom host and port are passed.	logger := m.Injector.Get(reflect.TypeOf(m.logger)).Interface().(*log.Logger)	logger.Printf("listening on %s (%s)\n", addr, Env)	logger.Fatalln(http.ListenAndServe(addr, m)) // m是整个框架控制的开始}// Run the http server. Listening on os.GetEnv("PORT") or 3000 by default.func (m *Martini) Run() {	port := os.Getenv("PORT")	if len(port) == 0 {		port = "3000"	}	host := os.Getenv("HOST")	m.RunOnAddr(host + ":" + port)}//创建请求上下文,与大部分的web框架一样,使用上下文的方式存储处理请求过程中的相关数据func (m *Martini) createContext(res http.ResponseWriter, req *http.Request) *context {	// NewResponseWriter 对res进行了封装修饰,添加了一些其他功能,比如过滤器之类的	c := &context{inject.New(), m.handlers, m.action, NewResponseWriter(res), 0}	c.SetParent(m)	c.MapTo(c, (*Context)(nil)) //Context 为接口类型, c 是实现了Context接口的具体类型结构体								//这里就是做一种接口与实现的映射, 使用具体实现的时候只能使用映射类型已经实现的接口	c.MapTo(c.rw, (*http.ResponseWriter)(nil))  // rw 同样是映射到interface{}类型	c.Map(req) // req 是一种具体类型,这里只是想存储这个值	return c}//经典的搭配,整合了路由以及martini核心功能// ClassicMartini represents a Martini with some reasonable defaults. Embeds the router functions for convenience.type ClassicMartini struct {	*Martini //	Router // 匿名变量类型,需要一个实现了所有的接口的对象,这样ClassicMartini实例可以无缝调用Router的接口,比如m.Get(pattern, handler)}// Classic creates a classic Martini with some basic default middleware - martini.Logger, martini.Recovery and martini.Static.// Classic also maps martini.Routes as a service.func Classic() *ClassicMartini {	r := NewRouter() // 基础路由器,用于存储用户自定义路由规则以及处理器	m := New() // martini基础框架	m.Use(Logger()) // 添加日志处理中间件	m.Use(Recovery()) // 服务器异常错误时进行处理	m.Use(Static("public")) // 静态文件处理 	m.MapTo(r, (*Routes)(nil)) // nil 表示这里只需要一个类型	m.Action(r.Handle) //所有router中间件执行完才执行的处理,相当于是正式的路由匹配处理,中间件做一些web常规必要的处理	return &ClassicMartini{m, r}}// Handler can be any callable function. Martini attempts to inject services into the handler's argument list.// Martini will panic if an argument could not be fullfilled via dependency injection.// 定义Handler类型为一个泛型type Handler interface{}//检查Handler是否为函数类型func validateHandler(handler Handler) {	if reflect.TypeOf(handler).Kind() != reflect.Func {		panic("martini handler must be a callable func")	}}// Context represents a request context. Services can be mapped on the request level from this interface.type Context interface {	// 包含了另一个接口类型的所有接口,Context的实例必须实现所有的接口,或者包含一个匿名的具体事例实现该所有接口	inject.Injector 	// 中间件的顺序执行过程中按顺序执行时,使用Next接口不断的更新索引指向下一个中间件	Next() 	// Written returns whether or not the response for this context has been written.	// 返回是否 http请求已经处理完并发送应答的标识	Written() bool }// http请求处理的上下文实例type context struct {	// 包含一个接口类型,初始化的时候需要一个具体实现的实例, 一个依赖注入的事例,注入具体函数,运行时按顺序回调各个函数	inject.Injector 	// handler数组,处理http请求时按顺序一个一个执行	handlers []Handler 	// 其实就是最后一个handler	action   Handler 	// 对http.ResponseWriter的进一步封装,加入更多功能,比如过滤器,Before After等处理	rw       ResponseWriter 	// 表示当前第n个hanlder的索引	index    int }// 取出当前第n个处理器,如果索引值到达最大值,则返回action函数,即开始路由匹配逻辑func (c *context) handler() Handler {	if c.index < len(c.handlers) {		return c.handlers[c.index]	}	if c.index == len(c.handlers) {		return c.action	}	panic("invalid index for context handler")}// 更新指向下一个处理器,之后继续执行剩余处理器对请求的处理func (c *context) Next() {	c.index += 1	c.run()}// 判断是否已发送应答,若已发送,则不需要再进行处理func (c *context) Written() bool {	return c.rw.Written()}// 获取当前的处理器,并处理http请求,执行一连串的处理func (c *context) run() {	for c.index <= len(c.handlers) {		_, err := c.Invoke(c.handler()) // Invoke 对初六进行回调,返回结构存储在context		if err != nil {			panic(err)		}		c.index += 1 // for循环先执行调用处理,再更新索引,因此与 Next中更新索引不冲突		if c.Written() {			return		}	}}

 如理解有错误,欢迎在评论指出,不胜感激!

转载于:https://www.cnblogs.com/bicowang/p/5251267.html

你可能感兴趣的文章
阿里镜像
查看>>
python 实现脚本上传远程服务并执行脚本
查看>>
Direct2D (39) : 使用 IDWriteTextLayout.Draw() 方法绘制文本
查看>>
突然想起曾经一面试官问我 | 和 || 的区别
查看>>
Struts框架之ActionForm(2)——ActionForm的生命周期
查看>>
GNS3中实现IP访问控制列表配置实验
查看>>
关于iwebshop里excel表格导出开发整理
查看>>
DNS软件bind使用(二)
查看>>
用J-Link调试S3C6410(2) --- 运行Leds程序
查看>>
jpeg格式文件剖析(一)
查看>>
TinyX对触摸屏支持的实现
查看>>
as3与java的socket通信实例1
查看>>
Spring学习笔记(11)——bean之间的关系
查看>>
Java学习笔记(35)——Java集合07之TreeMap
查看>>
甲骨文推Oracle WebLogic应用服务器12c
查看>>
WEB服务器、应用程序服务器、HTTP服务器区别
查看>>
工厂方法
查看>>
IPSEC ××× 综合应用
查看>>
Linux下安装及管理应用程序
查看>>
day13-权限管理su/sudo及限制root用户登录
查看>>