天天看点

深入Jetty源码之Connection

当jetty中的connector收到一个客户端的连接时(serversocket或serversocketchannel的accept()方法返回),connector会首先创建一个connectedendpoint用于和连接的底层(socket、channel)打交道(读写数据),在创建的connectedendpoint时会同时使用该endpoint创建相应类型的connection,然后会创建一个task仍给线程池,最终线程池会启动一个线程启动这个task,而在这个task中调用connection中的handle()方法,以处理当前的连接请求,在这个task中,它会持续的调用connection中的handle()方法直到连接关闭或connector停止。在connectorendpoint中,它自己就实现了runnable接口,因而可以将它自己丢给线程池,而在selectchannelendpoint中则交给selectormanager来管理客户端连接过来的channel,并调用connection的handle方法。 

connection的定义如下: 

public interface connection {

    // connection中的核心逻辑,在httpconnection中,它使用httpparser解析请求数据,httpparser采用事件相应机制,可以通过注册httpparser.eventhandler(requesthandler)填充httpconnection中的需要从请求消息中获取的信息,最后在messagecomplete()事件相应方法中调用handlerequest()方法将解析后的request请求交由server实例的handle()方法处理。在jetty中,server是handlerwrapper子类,它存储了所有注册的handler,从而将最终的处理流程传导给所有注册的handler。在所有注册的handler处理完成后,connection中的handle()方法会继续执行,使用httpgenerator、nestedgenerator将缓存的数据刷新到endpoint中。如果当前请求的相应状态是101(switching protocols),则handle方法返回的connection实例是从注册的以"org.eclipse.jetty.io.connection"为key的实例,也正是因为这个相应状态码的存在,这个handle方法的返回值是一个connection。

    connection handle() throws ioexception;

    // 除了handle方法,connection中还有一些提供了一些包含connection状态的方法:

    // 返回connection创建的时间戳。

    long gettimestamp();

    // 当前connection是否处于idle状态,如httpparser、httpgenerator都处于idle状态。

    boolean isidle();

    // 当前connection是否处于suspended状态,用于continuation机制。

    boolean issuspended();

    // 当connection关闭时会调用这个方法。

    void closed();

    // 当连接的idle时间超时后调用该方法,在httpconnection中,该方法会关闭endpoiont。

    void idleexpired();

}

深入Jetty源码之Connection

httpconnection是jetty中对connection的主要实现,它表示http客户端和服务器的一次连接,用于将request、response、endpoint联系在一起。同时httpconnection也是在避免使用pooling的方式下重用request、response、httpparser、httpgenerator、httpfields(requestfields、responsefields)、buffer、httpuri等(因为jetty保证了每一次连接只创建一个httpconnection实例,这是一个可以学习的点,不用pooling方式的重用,以进一步提升性能)。另外,httpconnection还有对connector和server实例的引用,并且用request字段记录了该connection总共处理的请求数(在headercomplete回调函数中自增)。

如果请求包含expect头,并且其值是100-continue,表示客户端希望在请求被正真处理前发送一个响应以表示是否能处理该请求,因而在第一次调用getinputstream时表示服务器已经准备好开始处理请求消息体了,此时在返回servletinputstream之前,服务器要发送100 continue响应消息给客户端(通过调用httpgenerator中的send1xx()方法)。在jetty中,httpinput类继承自servletinputstream,它从httpparser中读取请求消息体数据。

如果请求头包含expect头,并且它的值是102-processing,此时服务器可能会发送102状态码的响应,表示请求正在被处理,之后会发送最终的响应。在jetty中,可以通过response中的senderror()方法,传入102的状态码以发送102状态码的响应(使用httpgenerator中的send1xx()方法)。

handle()方法是httpconnection中的核心方法,在每一个连接到来时,connector会创建一个runnable实例,将该runnable实例扔到线程池中,在该runnable的run()方法实现中不断的调用connection的handle()方法直到当前连接或connector关闭。在该方法的实现中:

它首先设置_handling字段,表示当前正在处理,并且将当前httpconnection实例设置到__currentconnection的threadlocal变量中。

循环处理请求消息直到endpoint关闭或者在more_in_buffer为true(初始值为true)。

在这些过成中如果出现任何httpexception,则使用httpgenerator发送错误响应码给客户端(使用senderror()方法,并关闭endpoint)。

如果httpparser中还有数据未处理或者endpoint中还有输入数据未处理,则循环继续。

如果此时httpparser已经处理完成,httpgenerator已经处理完成,并且endpoint中的输出缓存中已经没有任何数据:1. 如果响应状态码时101 switching protocols,且在request存在org.eclipse.jetty.io.connection的connection实例,则新的connection从request的该attribute中获取,并重置httpparser和httpgenerator;2. 如果request不存在该attribute的connection,httpgenerator非persisent状态或endpoint的inputstream已经关闭,则重置httpparser,关闭endpoint,设置more_in_buffer为false,重置当前httpconnection。

如果httpparser处于idle状态,并且endpoint的inputstream已经关闭,则关闭当前endpoint,并设置more_in_buffer为false。

如果request的async处于启动状态,则设置more_in_buffer为false。

如果endpoint是asyncendpoint,generator已经commit,但是还未complete,则该endpoint schedule一个write操作。

最后,清理_handle字段和__currentconnection的threadlocal字段。

handlerequest()是httpconnection在对请求消息头解析完成后执行的真正处理逻辑方法:

对任何request还没有处理完成,并且server不为null且处于running状态,循环处理。

设置request的handled为false,以及pathinfo字段(如果pathinfo为null,又不是connect请求,则为400 error)。

如果_out字段不为null,reopen it。

如果request处于initial状态,设置request的dispatchertype为request,使用当前的endpoint和request实例配置connector,并调用使用当前httpconnection作为参数调用server的handle()方法;否则,设置request的dispatchertype为async,调用server的handleasync()方法(传入当前httpconnection做为参数)。

对任何非continuationthrowable异常,设置request的handled为true,error为true,对httpexception使用response发送响应状态码给客户端,而对throwable,使用httpgenerator发送400或500状态码给客户端。

如果此时request处于为完成状态,调用asynccontinuation.docomplete()方法;如果100 continue响应没有发送给客户端,则清除该状态,但是如果此时response还没有commit,则设置httpgenerator的persistent为false,表示客户端并没有发送数据过来,我们可以关闭该连接了;如果endpoint关闭了,则调用response的complete方法;如果endpoint没有关闭并且有error,直接关闭endpoint;如果endpoint没有关闭,也没有error,但是httpgenerator没有commit,request也没有被handle,则使用resonse发送404 resource not found响应消息,之后调用response的complete方法;最后设置request的handled为true。

commitresponse()方法,用于控制httpgenerator的执行流程:

在httpgenerator还没有commit之前(即响应状态行和响应消息头还没写入到endpoint中)时,先调用httpgenerator的setresponse()方法设置状态行。

然后调用httpgenerator的completeheader()方法将响应消息头写入到endpoint中。

最后调用httpgenerator的complete方法,不断的将httpgenerator中的缓存写入到endpoint中。

flushresponse()方法只是调用了commitresponse方法。 

httpoutput时jetty中继承自servletoutputstream的类,它使用abstractorgenerator向底层endpoint中写入数据。

output时httpconnection中的内部类,它继承自httpoutput,它在调用close/flush时会先调用commitresponse/flushreponse方法,保证响应消息先写状态行,然后是响应消息头,最后才是响应消息体。该类还实现了sendcontent方法,其参数可以是httpcontent类型或resource类型,该方法是一个util方法,它会自动设置content-type、content-length、last-modified等头,并将httpcontent或resource对应的数据写入到endpoint中。