前回はCloudbitSubscriptionServletクラスに、

doPostメソッド - littleBits社のクラウドから呼び出され、受け取ったデータをDBに格納する(ので人が直接結果を見ることはない)
doGetメソッド - WebブラウザでURL指定時に呼び出され、DB中のデータを表示する(ので人が目視で確認できる)

の枠を用意しておきました。

さて、cloudBitから受け取ったデータはCloudbitEventオブジェクトに格納できるように準備しましたが、DBへのデータの格納、およびDBからのデータの取り出しのために、CloudbitEventDAOクラスを定義します。

public class CloudbitEventDAO {
	protected DataSource dataSource;

	public CloudbitEventDAO(DataSource newDataSource) throws SQLException {
		setDataSource(newDataSource);
	}

	public DataSource getDataSource() {
		return dataSource;
	}
}

コンストラクタでアクセス先のDBのDataSourceオブジェクトを格納しておくようにします。setDataSource()メソッドの中を見てみましょう。

	public void setDataSource(DataSource newDataSource) throws SQLException {
		this.dataSource = newDataSource;
		Connection connection = dataSource.getConnection();
		try {
			PreparedStatement pstmt = connection.prepareCall(
					"set schema NEO_XXXXXXXXXXXXXXXXXXXXXXXXX");
			pstmt.execute();
		} finally {
			if (connection != null) {
				connection.close();
			}
		}
		checkTable();
	}

ここでは、インスタンス変数dataSourceに引数で渡されたオブジェクトを格納しておくだけでなく、前回も触れさせていただいたSAP HANA Cloud Platform(HCP)を使う際に、これから使用しようとしているテーブルがどのスキーマに属しているのかをset schema文で設定しています。その後に、checkTable()メソッドでCloudbitEventオブジェクトの格納先となるテーブルの有無を確認します。

	void checkTable() throws SQLException {
		Connection connection = null;
		try {
			connection = dataSource.getConnection();
			if (!existsTable(connection)) {
				createTable(connection);	// wouldn't be called
			}
		} finally {
			if (connection != null) {
				connection.close();
			}
		}
	}

	protected boolean existsTable(Connection conn) throws SQLException {
		String tableName = "EVENTS";
		DatabaseMetaData meta = conn.getMetaData();
		ResultSet rs = meta.getTables(null, null, tableName, null);
		while (rs.next()) {
			String name = rs.getString("TABLE_NAME");
			if (name.equals(tableName)) {
				return true;
			}
		}
		return false;
	}

	void createTable(Connection conn) throws SQLException {
		PreparedStatement pstmt = conn.prepareStatement(
				"CREATE COLUMN TABLE EVENTS (" +
				"EVENT_ID   varchar(255)," +
				"EVENT_TYPE varchar(255)," +
				"EVENT_TIMESTAMP bigint," +
				"EVENT_USERID integer," +
				"EVENT_BITID  varchar(255)," +
				"PAYLOAD_ABSOLUTE integer," +
				"PAYLOAD_PERCENT integer," +
				"PAYLOAD_LEVEL varchar(255)," +
				"PAYLOAD_DELTA varchar(255)," +
				"PRIMARY KEY (EVENT_ID))");
		pstmt.execute();
		pstmt.close();
	}

existsTable()メソッドで格納先テーブル(EVENTS)の有無を確認し、まだ存在していないようであれば、createTable()メソッドでEVENTSテーブルを新規に作成しています。テーブルの存在確認、およびテーブルの新規作成(createTable()メソッド内のCREATE TABLE文)に関しては、HCP以外の環境では使用するDBに合わせて修正が必要となるでしょう。CREATE TABLE文にCOLUMNが入っているのは、インメモリ＋カラム型テーブルを使用するためです(CREATE TABLEマニュアル)。

さて、データをDBに格納するために、addRow()メソッドを用意します。引数にはデータを格納したCloudbitEventオブジェクトを指定します。

	public void addRow(CloudbitEvent data) throws SQLException {
		CloudbitEvent eventData = (CloudbitEvent) data;
		Connection connection = dataSource.getConnection();
		try {
			PreparedStatement pstmt = connection.prepareCall(
					"INSERT INTO EVENTS " + 
					"(EVENT_ID,EVENT_TYPE,EVENT_TIMESTAMP,EVENT_USERID,EVENT_BITID," +
					"PAYLOAD_ABSOLUTE,PAYLOAD_PERCENT,PAYLOAD_LEVEL,PAYLOAD_DELTA) " +
					"VALUES(?,?,?,?,?,?,?,?,?)");
			pstmt.setString(1, UUID.randomUUID().toString());
			pstmt.setString(2, eventData.getEventType());
			pstmt.setLong(3, eventData.getEventTimestamp());
			pstmt.setInt(4, eventData.getEventUserId());
			pstmt.setString(5, eventData.getEventBitId());
			pstmt.setInt(6,  eventData.getPayloadAbsolute());
			pstmt.setInt(7, eventData.getPayloadPercent());
			pstmt.setString(8,  eventData.getPayloadLevel());
			pstmt.setString(9, eventData.getPayloadDelta());
			pstmt.executeUpdate();
			pstmt.close();
		} finally {
			if (connection != null) {
				connection.close();
			}
		}
	}

一方、データの取り出しには、selectAllRows()を使います(Allとは言いつつも、ここでは直近のものから最新100件までにしてあります)。

	public List<CloudbitEvent> selectAllRows() throws SQLException {
		Connection connection = dataSource.getConnection();
		try {
			PreparedStatement pstmt = connection.prepareStatement(
					"SELECT EVENT_ID,EVENT_TYPE,EVENT_TIMESTAMP,EVENT_USERID,EVENT_BITID," +
					"PAYLOAD_ABSOLUTE,PAYLOAD_PERCENT,PAYLOAD_LEVEL,PAYLOAD_DELTA " +
					"FROM EVENTS ORDER BY EVENT_TIMESTAMP DESC");
			ResultSet rs = pstmt.executeQuery();
			ArrayList<CloudbitEvent> list = new ArrayList<CloudbitEvent>();
			int count = 0;
			while (rs.next() && count++ < 100) {
				CloudbitEvent p = new CloudbitEvent();
				p.setEventId(rs.getString(1));
				p.setEventType(rs.getString(2));
				p.setEventTimestamp(rs.getInt(3));
				p.setEventUserId(rs.getInt(4));
				p.setEventBitId(rs.getString(5));
				p.setPayloadAbsolute(rs.getInt(6));
				p.setPayloadPercent(rs.getInt(7));
				p.setPayloadLevel(rs.getString(8));
				p.setPayloadDelta(rs.getString(9));
				list.add(p);
			}
			pstmt.close();
			return list;
		} finally {
			if (connection != null) {
				connection.close();
			}
		}
	}

ここまで、どちらかというとDBアクセスまわりの力技系の実装を見てきました。ここから、前回に枠組みを作っておいたCloudbitSubscriptionServletクラスに移りましょう。まずは、先ほど実装したCloudbitEventDAOクラスを利用するための準備をします。

	private CloudbitEventDAO cloudbitEventDAO;
      
	public void init() throws ServletException {
		super.init();
		try {
			InitialContext ctx = new InitialContext();
			DataSource dataSource = (DataSource) ctx
					.lookup("java:comp/env/jdbc/DefaultDB");
			cloudbitEventDAO = new CloudbitEventDAO(dataSource);
		} catch (SQLException e) {
			throw new ServletException(e);
		} catch (NamingException e) {
			throw new ServletException(e);
		}
	}

また、web.xmlファイルの中にlookupしているデータソースに関する情報を追加しておきます。

<web-app xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://java.sun.com/xml/ns/javaee" xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_3_0.xsd" id="WebApp_ID" version="3.0">
  <resource-ref>
    <res-ref-name>jdbc/DefaultDB</res-ref-name>
    <res-type>javax.sql.DataSource</res-type>
  </resource-ref>
</web-app>

HCP以外の環境で使用する場合には、DataSourceオブジェクトの取得に関しては、修正が必要になりかもしれません。いずれにせよ、これ以降、メンバ変数cloudbitEventDAOを経由してDBへのアクセスが可能になりました。

次に、doPost()メソッド、およびdoGet()メソッドの中身を見て行きましょう。

まずは、littleBitsクラウドからの呼び出しを受けるdoPost()メソッドです。前回までの時点では何もしない実装のままでしたが、ここでは、実際の処理はdoAdd()メソッドで行うようにしています。

	protected void doPost(HttpServletRequest request, 
			HttpServletResponse response) throws ServletException, IOException {
		try {
			doAdd(request);
		} catch (Exception e) {
			response.getWriter().println(
					"Database operation(POST) failed with reason: " +
					e.getMessage());
			LOGGER.error("Database operation(POST) failed", e);
		}
	}

さて、実際に処理を行うdoAdd()メソッドです。

	private static final String JSON_TYPE = "type";
	private static final String JSON_TIMESTAMP = "timestamp";
	private static final String JSON_USERID = "user_id";
	private static final String JSON_BITID = "bit_id";
	private static final String JSON_PAYLOAD = "payload";
	private static final String JSON_ABSOLUTE = "absolute";
	private static final String JSON_PERCENT = "percent";
	private static final String JSON_LEVEL = "level";
	private static final String JSON_DELTA = "delta";
	
	private void doAdd(HttpServletRequest request) throws ServletException, IOException, SQLException {
		BufferedReader reader = new BufferedReader(request.getReader());
		StringBuffer body = new StringBuffer();
		String line;
		
		while ((line = reader.readLine()) != null) {
			body.append(line);
		}
		
		CloudbitEvent eventData = new CloudbitEvent();
		try {
			JSONObject json = new JSONObject(body.toString());
			if (json.has(JSON_TYPE)) {
				eventData.setEventType(json.getString(JSON_TYPE));
			}
			if (json.has(JSON_TIMESTAMP)) {
				eventData.setEventTimestamp(json.getInt(JSON_TIMESTAMP));
			}
			if (json.has(JSON_USERID)) {
				eventData.setEventUserId(json.getInt(JSON_USERID));
			}
			if (json.has(JSON_BITID)) {
				eventData.setEventBitId(json.getString(JSON_BITID));
			}
			if (json.has(JSON_PAYLOAD)) {
				JSONObject json2 = json.getJSONObject(JSON_PAYLOAD);
				if (json2.has(JSON_ABSOLUTE)) {
					eventData.setPayloadAbsolute(json2.getInt(JSON_ABSOLUTE));
				}
				if (json2.has(JSON_PERCENT)) {
					eventData.setPayloadPercent(json2.getInt(JSON_PERCENT));
				}
				if (json2.has(JSON_LEVEL)) {
					eventData.setPayloadLevel(json2.getString(JSON_LEVEL));
				}
				if (json2.has(JSON_DELTA)) {
					eventData.setPayloadDelta(json2.getString(JSON_DELTA));
				}
			}
		} catch (Exception e) {
			new IOException("JSON parse exception: " + e.getMessage());
		}
		cloudbitEventDAO.addRow(eventData);
	}

ソースコード自体は長いものの、内部的にはlittleBitsクラウドから通知されたJSON形式のデータを解析しながらCloudbitEventオブジェクトに設定していき、最後に先ほど紹介したCloudbitEventDAOクラスのaddRow()メソッドでDBにINSERTするという単純なものです。

なお、JSON形式のデータの解析には、json.orgで公開されているパーザを使用しています。今回の範囲であれば、JSONArray, JSONException, JSONObject, JSONString, JSONTokenerの5クラスがあれば用が足ります。

さて、受け取ったデータの表示の方に移りましょう。

doGet()メソッドで最後の</body></html>を出力する直前に、以下の部分を追加します。

		try {
			appendTableData(request, response);
		} catch (Exception e) {
			writer.println(
					"Database operation(GET) failed with reason: " +
							e.getMessage());
		}

実際に出力を行うのは、appendTableData()メソッドです。

	void appendTableData(HttpServletRequest request, HttpServletResponse response) 
			throws SQLException, IOException {
		PrintWriter writer = response.getWriter();
		
		List<CloudbitEvent> resultList = cloudbitEventDAO.selectAllRows();
		writer.println(
				"<table><tbody><tr><th colspan=\"6\">" +
				(resultList.isEmpty() ? "No" : resultList.size()) +
				" Record(s) in HCP</th></tr>");
		writer.println("<tr><th>Timestamp</th>" +
				"<th>Label</th><th>Absolute</th><th>Percent</th><th>Level</th><th>Delta</th></tr>");
		IXSSEncoder xssEncoder = XSSEncoder.getInstance();
		for (CloudbitEvent p : resultList) {
			Date date = new Date(p.getEventTimestamp());
			String eventBitId = p.getEventBitId();
			// Timestamp
			writer.println("<tr><td>" +
					xssEncoder.encodeHTML(date.toString()) +
					"</td>");
			// Label
			String label = "(" + eventBitId + ")";
			writer.println("<td>" + xssEncoder.encodeHTML(label) + "</td>");
			// Absolute
			writer.println("<td>" +
					p.getPayloadAbsolute() +
					"</td>");
			// Percent
			writer.println("<td>" +
					p.getPayloadPercent() +
					"</td>");
			// Level
			writer.println("<td>" + 
					xssEncoder.encodeHTML(p.getPayloadLevel()) +
					"</td>");
			// Delta
			writer.println("<td>" + 
					xssEncoder.encodeHTML(p.getPayloadDelta()) + 
					"</td></tr>");
		}
		writer.println("</tbody></table>");
	}

DBから取得したデータをテーブル形式で表示するだけの単純な内容ですが、ここでは、Stringデータを出力する際には、HCPのSDKが提供するIXSSEncoderを使って、不正なデータが原因となってXSS(クロスサイトスクリプティング)が発生しないようにフィルタリングしています。

最後に動作確認です。現時点ではcloudBitから送られてきたデータをDBに格納してはいないものの、前回と同様、ブラウザ上でdoGet()メソッドの出力画面が見られれば(別の言い方をすれば、DBアクセスに問題がなければ)動作検証できた、ということにしておきましょう。

例によって、ソースコードに関しては、github上でcloudbit-javaプロジェクトを構築しました。さらに今回使った内容に関しては、3-CloudBit_Postブランチにて、動作・参照可能なソースコードを公開させて頂いております。

それでは続きはまたにしましょう！