Index: conf/jndi.properties =================================================================== --- conf/jndi.properties (revision 0) +++ conf/jndi.properties (revision 0) @@ -0,0 +1,36 @@ +## --------------------------------------------------------------------------- +## Licensed to the Apache Software Foundation (ASF) under one or more +## contributor license agreements. See the NOTICE file distributed with +## this work for additional information regarding copyright ownership. +## The ASF licenses this file to You under the Apache License, Version 2.0 +## (the "License"); you may not use this file except in compliance with +## the License. You may obtain a copy of the License at +## +## http://www.apache.org/licenses/LICENSE-2.0 +## +## Unless required by applicable law or agreed to in writing, software +## distributed under the License is distributed on an "AS IS" BASIS, +## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +## See the License for the specific language governing permissions and +## limitations under the License. +## --------------------------------------------------------------------------- + +# If ActiveMQ is used then uncomment following properties, else substitute it accordingly. +#java.naming.factory.initial = org.apache.activemq.jndi.ActiveMQInitialContextFactory + +# use the following property to provide location of MQ broker. +#java.naming.provider.url = tcp://localhost:61616 + +# use the following property to specify the JNDI name the connection factory +# should appear as. +#connectionFactoryNames = connectionFactory, queueConnectionFactory, topicConnectionFactry + +# register some queues in JNDI using the form +# queue.[jndiName] = [physicalName] +# queue.MyQueue = example.MyQueue + + +# register some topics in JNDI using the form +# topic.[jndiName] = [physicalName] +# topic.MyTopic = example.MyTopic + Index: ivy.xml =================================================================== --- ivy.xml (revision 1095820) +++ ivy.xml (working copy) @@ -43,5 +43,8 @@ --> - + + + + Index: ivy/ivysettings.xml =================================================================== --- ivy/ivysettings.xml (revision 1095820) +++ ivy/ivysettings.xml (working copy) @@ -36,10 +36,17 @@ + + + Index: ivy/libraries.properties =================================================================== --- ivy/libraries.properties (revision 1095820) +++ ivy/libraries.properties (working copy) @@ -18,4 +18,6 @@ pig.version=0.8.0 commons-cli.version=1.0 #hadoop-core.version=0.20.2 Waiting for a secure version of hadoop in maven - +jms.version=1.1 +activemq.version=5.4.2 +javax-mgmt.version=1.1-rev-1 Index: src/test/org/apache/hcatalog/listener/TestHCatListener.java =================================================================== --- src/test/org/apache/hcatalog/listener/TestHCatListener.java (revision 0) +++ src/test/org/apache/hcatalog/listener/TestHCatListener.java (revision 0) @@ -0,0 +1,189 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.hcatalog.listener; + +import java.util.ArrayList; +import java.util.List; +import java.util.concurrent.atomic.AtomicInteger; + +import javax.jms.Connection; +import javax.jms.ConnectionFactory; +import javax.jms.Destination; +import javax.jms.JMSException; +import javax.jms.Message; +import javax.jms.MessageConsumer; +import javax.jms.MessageListener; +import javax.jms.ObjectMessage; +import javax.jms.Session; + +import org.apache.activemq.ActiveMQConnectionFactory; +import org.apache.hadoop.hive.cli.CliSessionState; +import org.apache.hadoop.hive.conf.HiveConf; +import org.apache.hadoop.hive.metastore.HiveMetaStore; +import org.apache.hadoop.hive.metastore.HiveMetaStoreClient; +import org.apache.hadoop.hive.metastore.api.Database; +import org.apache.hadoop.hive.metastore.api.MetaException; +import org.apache.hadoop.hive.metastore.api.NoSuchObjectException; +import org.apache.hadoop.hive.metastore.api.Partition; +import org.apache.hadoop.hive.metastore.api.Table; +import org.apache.hadoop.hive.metastore.api.UnknownTableException; +import org.apache.hadoop.hive.ql.Driver; +import org.apache.hadoop.hive.ql.session.SessionState; +import org.apache.hcatalog.common.HCatConstants; +import org.apache.thrift.TException; + +import junit.framework.TestCase; + +public class TestHCatListener extends TestCase implements MessageListener{ + + private static final String msPort = "20201"; + private HiveConf hiveConf; + private HiveMetaStoreClient msc; + private Driver driver; + private AtomicInteger cntInvocation = new AtomicInteger(0); + + private static class RunMS implements Runnable { + + @Override + public void run() { + HiveMetaStore.main(new String[]{msPort}); + } + + } + + @Override + protected void setUp() throws Exception { + + super.setUp(); + System.setProperty("java.naming.factory.initial", "org.apache.activemq.jndi.ActiveMQInitialContextFactory"); + System.setProperty("java.naming.provider.url", "vm://localhost?broker.persistent=false"); + System.setProperty("hive.metastore.listener",HCatListener.class.getName()); + ConnectionFactory connFac = new ActiveMQConnectionFactory("vm://localhost?broker.persistent=false"); + Connection conn = connFac.createConnection(); + conn.start(); + // We want message to be sent when session commits, thus we run in + // transacted mode. + Session session = conn.createSession(true, Session.SESSION_TRANSACTED); + Destination hcatTopic = session.createTopic(HCatConstants.HCAT_TOPIC); + MessageConsumer consumer1 = session.createConsumer(hcatTopic); + consumer1.setMessageListener(this); + Destination tblTopic = session.createTopic("mydb.mytbl"); + MessageConsumer consumer2 = session.createConsumer(tblTopic); + consumer2.setMessageListener(this); + Thread t = new Thread(new RunMS()); + t.start(); + Thread.sleep(20000); + hiveConf = new HiveConf(this.getClass()); + hiveConf.set("hive.metastore.local", "false"); + hiveConf.setVar(HiveConf.ConfVars.METASTOREURIS, "thrift://localhost:" + msPort); + hiveConf.setIntVar(HiveConf.ConfVars.METASTORETHRIFTRETRIES, 3); + hiveConf.set(HiveConf.ConfVars.PREEXECHOOKS.varname, ""); + hiveConf.set(HiveConf.ConfVars.POSTEXECHOOKS.varname, ""); + hiveConf.set(HiveConf.ConfVars.HIVE_SUPPORT_CONCURRENCY.varname, "false"); + SessionState.start(new CliSessionState(hiveConf)); + msc = new HiveMetaStoreClient(hiveConf, null); + driver = new Driver(hiveConf); + } + + @Override + protected void tearDown() throws Exception { + assertEquals(7, cntInvocation.get()); + super.tearDown(); + } + + public void testAMQListener() throws MetaException, TException, UnknownTableException, NoSuchObjectException{ + driver.run("create database mydb"); + driver.run("use mydb"); + driver.run("create table mytbl (a string) partitioned by (b string)"); + driver.run("alter table mytbl add partition(b='2011')"); + msc.finalizePartition("mydb", "mytbl", "b='2011'"); + driver.run("alter table mytbl drop partition(b='2011')"); + driver.run("drop table mytbl"); + driver.run("drop database mydb"); + } + + @Override + public void onMessage(Message msg) { + cntInvocation.incrementAndGet(); + String event; + try { + event = msg.getStringProperty(HCatConstants.HCAT_EVENT); + + if(event.equals(HCatConstants.HCAT_ADD_DATABASE_EVENT)){ + + assertEquals("topic://"+HCatConstants.HCAT_TOPIC,msg.getJMSDestination().toString()); + assertEquals("mydb", ((Database) ((ObjectMessage)msg).getObject()).getName()); + } + else if(event.equals(HCatConstants.HCAT_ADD_TABLE_EVENT)){ + + assertEquals("topic://"+HCatConstants.HCAT_TOPIC,msg.getJMSDestination().toString()); + Table tbl = (Table)(((ObjectMessage)msg).getObject()); + assertEquals("mytbl", tbl.getTableName()); + assertEquals("mydb", tbl.getDbName()); + assertEquals(1, tbl.getPartitionKeysSize()); + } + else if(event.equals(HCatConstants.HCAT_ADD_PARTITION_EVENT)){ + + assertEquals("topic://mydb.mytbl",msg.getJMSDestination().toString()); + Partition part = (Partition)(((ObjectMessage)msg).getObject()); + assertEquals("mytbl", part.getTableName()); + assertEquals("mydb", part.getDbName()); + List vals = new ArrayList(1); + vals.add("2001"); + assertEquals("topic://mydb.mytbl",msg.getJMSDestination().toString()); + assertEquals(vals,part.getValues()); + } + else if(event.equals(HCatConstants.HCAT_DROP_PARTITION_EVENT)){ + + assertEquals("topic://mydb.mytbl",msg.getJMSDestination().toString()); + Partition part = (Partition)(((ObjectMessage)msg).getObject()); + assertEquals("mytbl", part.getTableName()); + assertEquals("mydb", part.getDbName()); + List vals = new ArrayList(1); + vals.add("2001"); + assertEquals("topic://mydb.mytbl",msg.getJMSDestination().toString()); + assertEquals(vals,part.getValues()); + } + else if(event.equals(HCatConstants.HCAT_DROP_TABLE_EVENT)){ + + assertEquals("topic://"+HCatConstants.HCAT_TOPIC,msg.getJMSDestination().toString()); + Table tbl = (Table)(((ObjectMessage)msg).getObject()); + assertEquals("mytbl", tbl.getTableName()); + assertEquals("mydb", tbl.getDbName()); + assertEquals(1, tbl.getPartitionKeysSize()); + } + else if(event.equals(HCatConstants.HCAT_DROP_DATABASE_EVENT)){ + + assertEquals("topic://"+HCatConstants.HCAT_TOPIC,msg.getJMSDestination().toString()); + assertEquals("mydb", ((Database) ((ObjectMessage)msg).getObject()).getName()); + } + else if(event.equals(HCatConstants.HCAT_FINALIZE_PARTITON)){ + + assertEquals("topic://mydb.mytbl",msg.getJMSDestination().toString()); + assertEquals("b='2011'", ((String)((ObjectMessage)msg).getObject())); + } + else + assert false; + } catch (JMSException e) { + e.printStackTrace(System.err); + assert false; + throw new RuntimeException(e); + } + } +} Index: src/java/org/apache/hcatalog/listener/HCatListener.java =================================================================== --- src/java/org/apache/hcatalog/listener/HCatListener.java (revision 0) +++ src/java/org/apache/hcatalog/listener/HCatListener.java (revision 0) @@ -0,0 +1,217 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.hcatalog.listener; + +import java.io.Serializable; +import java.util.ArrayList; +import java.util.HashMap; + +import javax.jms.Connection; +import javax.jms.ConnectionFactory; +import javax.jms.Destination; +import javax.jms.JMSException; +import javax.jms.MessageProducer; +import javax.jms.ObjectMessage; +import javax.jms.Session; +import javax.naming.Context; +import javax.naming.InitialContext; +import javax.naming.NamingException; + +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; +import org.apache.hadoop.hive.metastore.NoOpListener; +import org.apache.hadoop.hive.metastore.api.Database; +import org.apache.hadoop.hive.metastore.api.MetaException; +import org.apache.hadoop.hive.metastore.api.Order; +import org.apache.hadoop.hive.metastore.api.Partition; +import org.apache.hadoop.hive.metastore.api.StorageDescriptor; +import org.apache.hadoop.hive.metastore.api.Table; +import org.apache.hcatalog.common.HCatConstants; + +/** + * Implementation of {@link org.apache.hadoop.hive.metastore.MetaStoreEventListener} + * It sends message on two type of topics. One has name of form dbName.tblName + * On this topic, two kind of messages are sent: add/drop partition and + * finalize_partition message. + * Second topic has name "HCAT" and messages sent on it are: add/drop database + * and add/drop table. + * All messages also has a property named "HCAT_EVENT" set on them whose value + * can be used to configure message selector on subscriber side. + */ +public class HCatListener extends NoOpListener{ + + private static final Log LOG = LogFactory.getLog(HCatListener.class); + private Session session; + private Connection conn; + + /** + * Create message bus connection and session in constructor. + */ + public HCatListener() { + + try { + Context jndiCntxt = new InitialContext(); + ConnectionFactory connFac = (ConnectionFactory)jndiCntxt.lookup("ConnectionFactory"); + conn = connFac.createConnection(); + conn.start(); + // We want message to be sent when session commits, thus we run in + // transacted mode. + session = conn.createSession(true, Session.SESSION_TRANSACTED); + + } catch (NamingException e) { + LOG.error("JNDI error while setting up Message Bus connection. " + + "Please make sure file named 'jndi.properties' is in " + + "classpath and contains appropriate key-value pairs.",e); + } + catch (JMSException e) { + LOG.error("Failed to initialize connection to message bus",e); + } + catch(Throwable t){ + LOG.error("HCAT Listener failed to load",t); + } + } + + @Override + public void onAddPartition(Partition partition) throws MetaException { + // Subscriber can get notification of newly add partition in a + // particular table by listening on a topic named "dbName.tableName" + // and message selector string as "HCAT_EVENT = HCAT_ADD_PARTITION" + send(partition, partition.getDbName()+"."+partition.getTableName(), + HCatConstants.HCAT_ADD_PARTITION_EVENT); + } + + @Override + public void onDropPartition(Partition partition) throws MetaException { + // Subscriber can get notification of dropped partition in a + // particular table by listening on a topic named "dbName.tableName" + // and message selector string as "HCAT_EVENT = HCAT_DROP_PARTITION" + + // Datanucleus throws NPE when we try to serialize a partition object + // retrieved from metastore. To workaround that we reset following objects + + StorageDescriptor sd = partition.getSd(); + sd.setBucketCols(new ArrayList()); + sd.setSortCols(new ArrayList()); + sd.setParameters(new HashMap()); + sd.getSerdeInfo().setParameters(new HashMap()); + send(partition, partition.getDbName()+"."+partition.getTableName(), + HCatConstants.HCAT_DROP_PARTITION_EVENT); + } + + @Override + public void onCreateDatabase(Database db) throws MetaException { + // Subscriber can get notification about addition of a database in HCAT + // by listening on a topic named "HCAT" and message selector string + // as "HCAT_EVENT = HCAT_ADD_DATABASE" + send(db,HCatConstants.HCAT_TOPIC,HCatConstants.HCAT_ADD_DATABASE_EVENT); + } + + @Override + public void onDropDatabase(Database db) throws MetaException { + // Subscriber can get notification about drop of a database in HCAT + // by listening on a topic named "HCAT" and message selector string + // as "HCAT_EVENT = HCAT_DROP_DATABASE" + send(db,HCatConstants.HCAT_TOPIC,HCatConstants.HCAT_DROP_DATABASE_EVENT); + } + + @Override + public void onCreateTable(Table table) throws MetaException { + // Subscriber can get notification about addition of a table in HCAT + // by listening on a topic named "HCAT" and message selector string + // as "HCAT_EVENT = HCAT_ADD_TABLE" + + send(table,HCatConstants.HCAT_TOPIC, HCatConstants.HCAT_ADD_TABLE_EVENT); + } + + @Override + public void onDropTable(Table table) throws MetaException { + // Subscriber can get notification about drop of a table in HCAT + // by listening on a topic named "HCAT" and message selector string + // as "HCAT_EVENT = HCAT_DROP_TABLE" + + // Datanucleus throws NPE when we try to serialize a table object + // retrieved from metastore. To workaround that we reset following objects + + StorageDescriptor sd = table.getSd(); + sd.setBucketCols(new ArrayList()); + sd.setSortCols(new ArrayList()); + sd.setParameters(new HashMap()); + sd.getSerdeInfo().setParameters(new HashMap()); + + send(table,HCatConstants.HCAT_TOPIC, HCatConstants.HCAT_DROP_TABLE_EVENT); + } + + @Override + public void onFinalizePartition(String dbName, String tblName, + String partName) throws MetaException { + // Subscriber can get notification about partition finalization by + // listening on topic named "dbName.tableName" and message selector string + // as "HCAT_EVENT = HCAT_FINALIZE_PARTITION". Message body in this case + // will contain partition specification. + send(partName,dbName+"."+tblName,HCatConstants.HCAT_FINALIZE_PARTITON); + } + + /** + * @param msgBody is the metastore object. It is sent in full such that + * if subscriber is really interested in details, it can reconstruct it fully. + * In case of finalize_partition message this will be string specification of + * the partition. + * @param topicName is the name on message broker on which message is sent. + * @param event is the value of HCAT_EVENT property in message. It can be + * used to select messages in client side. + */ + private void send(Serializable msgBody, String topicName, String event){ + + if(null == session){ + // If we weren't able to setup the session in the constructor + // we cant send message in any case. + LOG.error("Invalid session. Failed to send message on topic: "+ + topicName + " event: "+event); + return; + } + + try{ + // Topics are created on demand. If it doesn't exist on broker it will + // be created when broker receives this message. + Destination topic = session.createTopic(topicName); + MessageProducer producer = session.createProducer(topic); + ObjectMessage msg = session.createObjectMessage(msgBody); + msg.setStringProperty(HCatConstants.HCAT_EVENT, event); + producer.send(msg); + // Message must be transacted before we return. + session.commit(); + } catch(Exception e){ + // Gobble up the exception. Message delivery is best effort. + LOG.error("Failed to send message on topic: "+topicName + + " event: "+event , e); + } + } + + @Override + protected void finalize() throws Throwable { + // Close the connection before dying. + try { + if(conn != null) { + conn.close(); + } + } catch (Exception ignore) { + LOG.info("Failed to close message bus connection.", ignore); + } + } +} Index: src/java/org/apache/hcatalog/common/HCatConstants.java =================================================================== --- src/java/org/apache/hcatalog/common/HCatConstants.java (revision 1095820) +++ src/java/org/apache/hcatalog/common/HCatConstants.java (working copy) @@ -64,4 +64,16 @@ public static final String HCAT_KEY_OUTPUT_INFO = HCAT_KEY_OUTPUT_BASE + ".info"; public static final String HCAT_KEY_HIVE_CONF = HCAT_KEY_OUTPUT_BASE + ".hive.conf"; public static final String HCAT_KEY_TOKEN_SIGNATURE = HCAT_KEY_OUTPUT_BASE + ".token.sig"; + + public static final String HCAT_TOPIC = "HCAT"; + public static final String HCAT_EVENT = "HCAT_EVENT"; + public static final String HCAT_FINALIZE_PARTITON = "HCAT_FINALIZE_PARTITON"; + + public static final String HCAT_ADD_PARTITION_EVENT = "HCAT_ADD_PARTITION"; + public static final String HCAT_DROP_PARTITION_EVENT = "HCAT_DROP_PARTITION"; + public static final String HCAT_ADD_TABLE_EVENT = "HCAT_ADD_TABLE"; + public static final String HCAT_DROP_TABLE_EVENT = "HCAT_DROP_TABLE"; + public static final String HCAT_ADD_DATABASE_EVENT = "HCAT_ADD_DATABASE"; + public static final String HCAT_DROP_DATABASE_EVENT = "HCAT_DROP_DATABASE"; + }