用Servlet写简单的ApiGateway (二) --结合mybatis实现服务的注册和请求转发

最新推荐文章于 2024-07-19 14:44:32 发布

moonncat

最新推荐文章于 2024-07-19 14:44:32 发布

阅读量572

点赞数

分类专栏： JAVA 文章标签： Servlet java Mybatis

本文链接：https://blog.csdn.net/moonncat/article/details/81159812

版权

JAVA 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

一, 环境准备

1, 下载一个简单的restful服务, 本文用了spring官网的demo, 传送门: https://spring.io/guides/gs/rest-service/

2, 上个章节用到的api-gateway项目, 传送门: 用Servlet写简单的ApiGateway (一)

3. 安装mysql, 并创建好所需的测试数据

二, 配置mybatis

具体步骤传送门: Mybatis组件在项目中的基础用法

三, 主程序

1, 项目文件结构

主程序文件App.java

package com.demo.apigateway;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.http.HttpEntity;
import org.apache.http.HttpException;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClients;

import com.demo.data.RouteInfoDataAccess;

public class App extends HttpServlet {

	private static final long serialVersionUID = 1L;
	private HttpClient client;
	private RouteInfoDataAccess routeData;
	private String serviceHost="http://localhost:8090";

	@Override
	public void init() throws ServletException {
		client = HttpClients.createDefault();
		routeData = new RouteInfoDataAccess();
	}

	@Override
	public void destroy() {
		// 什么也不做
	}

	@Override
	public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		PrintWriter out = response.getWriter();

		String clientUrl =request.getRequestURI();

		String serviceUrl = routeData.getRouteInfo(clientUrl);
		if (serviceUrl == null) {
			out.println(String.format("404: %s", clientUrl));
		} else {
			String url= mapUrl(serviceUrl);
			HttpGet httpGet = new HttpGet(url);
			HttpResponse httpResponse = client.execute(httpGet);

			String content = readContent(httpResponse);
			response.setContentType("application/json;charset=utf-8");
			out.println(content);
		}
	}

	@Override
	public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		PrintWriter out = response.getWriter();

		String clientUrl =request.getRequestURI();

		String serviceUrl = routeData.getRouteInfo(clientUrl);
		if (serviceUrl == null) {
			out.println(String.format("404: %s", clientUrl));
		}else {
			String url= mapUrl(serviceUrl);
			HttpPost httpPost = new HttpPost(url);
			HttpEntity entity=new StringEntity(readContent(request));
			
			httpPost.setEntity(entity);
			HttpResponse httpResponse = client.execute(httpPost);

			String content=readContent(httpResponse);
			response.setContentType("application/json;charset=utf-8");
			out.println(content);
		}
	}

	@Override
	public void doPut(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

	}

	private String mapUrl(String clientUrl) {
		return String.format("%s%s",serviceHost, clientUrl);
	}
	private String readContent(HttpResponse httpResponse) throws UnsupportedOperationException, IOException {
		InputStream inStream = httpResponse.getEntity().getContent();
		BufferedReader reader = new BufferedReader(new InputStreamReader(inStream, "utf-8")); // utf-8 encoded
		StringBuilder strber = new StringBuilder();
		String line = null;
		while ((line = reader.readLine()) != null)
			strber.append(line);

		inStream.close();
		return strber.toString();
	}

	private String readContent(HttpServletRequest request) throws UnsupportedOperationException, IOException {
		BufferedReader reader = request.getReader();
		StringBuilder strber = new StringBuilder();
		String line = null;
		while ((line = reader.readLine()) != null)
			strber.append(line);

		return strber.toString();
	}
}

为了简化代码, 我把公用部分封装成方法mapUrl, readContent

四, 启动服务

1, 用maven把restful的服务打包, 然后运行:

maven install打包成jar