Google Chrome Blog: A Speedier Google Chrome for all users

Google Chrome Blog: A Speedier Google Chrome for all users

Groovy Http POST/GET Class

I was working on a project in Groovy and needed to do POST as well as read the response headers.

I was unable to find a suitable class or method for this in either Groovy or Java, so I wrote this class. It opens a socket to the requested host, and writes to the socket using getOutputStream() and PrintWriter(). It reads the response from the server using DataInputStream() and getInputStream().

/**
 * A Simple HTTP POST/GET Helper Class for Groovy 
 *
 * @author   Tony Landis
 * @copyright  2007 Tony Landis
 * @website  http://www.tonylandis.com
 * @license  BSD License (http://www.opensource.org/licenses/bsd-license.php)
 * @example  h = new GroovyHTTP('http://www.google.com/search')   
 *    h.setMethod('GET') 
 *    h.setParam('q', 'groovy')
 *    h.open()
 *    h.write()
 *    h.read() 
 *    println h.getHeader('Server')  
 *    println h.getContent() 
 *    h.close()
 */
class GroovyHTTP {
 public method='POST'
 public uri
 public host
 public path
 public port
 public params=null
 public socket=null
 public writer=null
 public reader=null
 public writedata
 public headers = []
 public content

 // set the url and create new URI object
 def GroovyHTTP(url) {
  uri = new URI(url)
  host = uri.getHost()
  path = uri.getRawPath() 
  port = uri.getPort() 
  def tpar = uri.getQuery()
  if(tpar != null && tpar != '') {
   tpar.tokenize('&').each{
    def pp = it.tokenize('=');
    this.setParam(pp[0],pp[1]);
   }
  } 
  if(port == null || port < 0) port = 80
  if(path == null || path.length() == 0) path = "/"   
 }

 // sets the method (GET or POST)
 def setMethod(setmethod) {
  method = setmethod
 }

 // push params into this request
 def setParam(var,value) {
  if(params != null) 
   params += '&'
  else
   params=''
  params += var +'='+URLEncoder.encode(value) 
 }

 // clear params
 def clearParams() {
  params = null
 }

 // open a new socket
 def open() { 
  socket = new Socket(host, port)
 }

 // write data to the socket
 def write() { 
  def contentLen = 0
  if(params!=null) contentLen = params.length() 
  def writedata = '';
  
  if(this.method == 'GET')  
   writedata += "GET " + path +'?'+ params + " HTTP/1.0\r\n"
  else
   writedata += "POST " + path + " HTTP/1.0\r\n"
  
  writedata +=
   "Host: " + host + "\r\n" + 
   "Content-Type: application/x-www-form-urlencoded\r\n" + 
   "Content-Length: " + contentLen + "\r\n\r\n" + 
   params + "\r\n"
   "Connection: close\r\n\r\n"  
      
  writer = new PrintWriter(socket.getOutputStream(), true)
  writer.write(writedata)
  writer.flush() 
 }

 // read response from the server
 def read() { 
  reader = new DataInputStream(socket.getInputStream())
  def c = null
  while (null != ((c = reader.readLine()))) {   
   if(c=='') break 
   headers.add(c)   
  } 
 } 

 // get header value by name
 def getHeader(name) {
  def pattern = name + ': ' 
  def result
  headers.each{ 
   if(it ==~ /${pattern}.*/) { 
    result = it.replace(pattern,'').trim()
    return 2
   }   
  }
  return result
 }

 // get the response content
 def getContent() { 
  def row 
  content = ''
  while (null != ((row = reader.readLine()))) content += row + "\r\n"  
  return content = content.trim();
 }

 // close the socket
 def close() {
  reader.close() 
  writer.close()
  socket.close()
 }
}