Stormリアルタイムクラウドコンピューティング学習使用基本apiおよび高レベルapi tridentの基本使用を含む

/**
 *   vladdistance                
 * @author zhuyuping
 * 2016 3 5 
 */
public class ImageVladIndexSearchBolt extends BaseBatchBolt{

	Object uuid;
	BatchOutputCollector collector;
	
	private double[][] distances;
	Map<Integer,double[][]> knns; 
	QueryRunner runner;
	byte[] imgdatas=null;
	
	@Override
	public void prepare(Map conf, TopologyContext context,
			BatchOutputCollector collector, Object id) {
		 this.uuid=id;
		 this.collector=collector;
		 this.runner=new QueryRunner(DBHelper.getDateSource());
		 this.distances=new double[16][256];
		 this.knns=AliyunOSSUtils.getKNNDatas();
	}

	@Override
	public void execute(Tuple tuple) {
		//declarer.declare(new Fields("id","key","query","k"));
		Integer key=tuple.getInteger(1);
		//System.out.println("======================"+key);
		if(imgdatas==null)
		imgdatas=tuple.getBinary(4);
		double[] qus=(double[]) tuple.getValue(2);
		//System.out.println("qu is $$$$$$ "+qus.length);
		Integer k=tuple.getInteger(3);
		final BoundedPriorityQueue<IntDoublePair> queue = new BoundedPriorityQueue<IntDoublePair>(
				k, IntDoublePair.SECOND_ITEM_ASCENDING_COMPARATOR);
		double[][] nn=knns.get(key);
		for (int j = 0; j < k; j++) {
			double score = DoubleFVComparison.SUM_SQUARE.compare(qus,nn[j]);
			int index = key;
			IntDoublePair wp = new IntDoublePair(index, score);
			wp = queue.offerItem(wp);
		}
		List<IntDoublePair> pair = queue.toOrderedListDestructive();// double[]
		double[] sort = new double[pair.size()];
		for (int j = 0; j < pair.size(); j++) {
			sort[j] = pair.get(j).second;
			//System.out.println("vlad sort is ********************"+sort[j]);
		}
		
		
		this.distances[key]=sort;
		System.out.println("=================="+this.distances[key][70]);
	}

	@Override
	public void finishBatch() {
		//                 mysql     count     10 
		System.out.println("=================="+this.distances[0][70]);
		try{
		List<VladIndex> vladIndexs=runner.query("select * from vladindex", new ResultSetHandler<List<VladIndex>>() {
			@Override
			public List<VladIndex> handle(ResultSet rs) throws SQLException {
				List<VladIndex> vladindexs=Lists.newArrayList();
				while(rs.next()){
				String id=rs.getString("id");
				String vlad=rs.getString("vlad");
				vladindexs.add(new VladIndex(id,vlad));
				}
				return vladindexs;
			}
		});
//		ByteArrayDataOutput out=ByteStreams.newDataOutput();
//		BinIO.storeDoubles(this.distances,out);
//		byte[] distancesdatas=out.toByteArray();
		AliyunOCSService.put(uuid+"distance", distances);
		for (VladIndex vladIndex : vladIndexs) {
			//System.out.println("!!!!!!!!!!!!!!!!!!!!!!!!"+distancesdatas.length/1024);
			collector.emit(new Values(uuid.toString(),vladIndex.getId(),vladIndex.getDatas(),imgdatas));
		}
		}catch(Exception e){
			e.printStackTrace();
		}
		
	}

	@Override
	public void declareOutputFields(OutputFieldsDeclarer declarer) {
		declarer.declare(new Fields("id","imageid","datas","imagedatas"));
		
	}

}

declarer.declare(new Fields("id"," 


	LinearDRPCTopologyBuilder builder = new LinearDRPCTopologyBuilder("search");
		builder.addBolt(new IndexSearchTaskBolt());
		builder.addBolt(new ImageSearchBolt(), 10).fieldsGrouping(new Fields("type"));
		builder.addBolt(new ImageSurfCompareBolt(), 30).fieldsGrouping(new Fields("imageid"));
		//builder.addBolt(new ImageTopScoreBolt()).globalGrouping();
		builder.addBolt(new ImageTopScoreBolt()).fieldsGrouping(new Fields("id"));
		return builder;
		main    
		LinearDRPCTopologyBuilder builder = construct();
		Config conf = new Config();
		if (args == null || args.length == 0) {
			conf.setMaxTaskParallelism(5);
			LocalDRPC drpc = new LocalDRPC();
			LocalCluster cluster = new LocalCluster();
			cluster.submitTopology("search", conf, builder.createLocalTopology(drpc));
			try {
				for (int i = 0; i < 4; i++) {
					String result=drpc.execute("search", Base64.encodeBase64String(Files.toByteArray(new File("c:/baidu/test3.jpg"))));
					System.out.println(i+" ------------------------------------- "+result);
				}
			} catch (IOException e) {
			}
			cluster.shutdown();
			drpc.shutdown();
		}else {
			conf.put(Config.TOPOLOGY_NAME, "search");
			conf.put(Config.DRPC_PORT, 3772);
			conf.put(Config.DRPC_SERVERS,Lists.newArrayList("10.47.50.235","10.47.49.206"));
			conf.put(Config.NIMBUS_HOST,  "10.47.50.235");
			conf.put(Config.NIMBUS_THRIFT_PORT, 6627);
			conf.setMaxSpoutPending(5000);
			conf.setNumWorkers(10);
			conf.setNumAckers(5);
			conf.put("topology.spout.max.batch.size", 1000 /* x1000 i.e. every tuple has 1000 feature vectors*/);
		    conf.put("topology.trident.batch.emit.interval.millis", 1000);
			conf.put(Config.STORM_CLUSTER_MODE, "distributed");
		    conf.put(Config.NIMBUS_TASK_TIMEOUT_SECS, 10);
			StormSubmitter.submitTopologyWithProgressBar(args[0], conf, builder.createRemoteTopology());
		}
		
	}

each(new Fields("word"), new StringLength(), new Fields("length"))

 private static StormTopology buildTopology() {
	        FixedBatchSpout spout = new FixedBatchSpout(new Fields("sentence"), 3,
	                new Values("the cow jumped over the moon"),
	                new Values("the man went to the store and bought some candy"),
	                new Values("four score and seven years ago"),
	                new Values("how many apples can you eat"));
	        spout.setCycle(true);

	        TridentTopology topology = new TridentTopology();
	        topology.newStream("spout", spout)
	                //no name
	                .each(new Fields("sentence"), new Split(), new Fields("word"))
	                .partitionBy(new Fields("word"))
	                .name("abc")
	                .each(new Fields("word"), new StringLength(), new Fields("length"))
	                .partitionBy(new Fields("length"))
	                .name("def")
	                .aggregate(new Fields("length"), new Count(), new Fields("count"))
	                .partitionBy(new Fields("count"))
	                .name("ghi")
	                .aggregate(new Fields("count"), new Sum(), new Fields("sum"));
	        return topology.build();
	    }
	  public static void main(String[] args) throws Exception  {
		  StormTopology topology = buildTopology();
		  Config conf = new Config();
		  LocalCluster cluster = new LocalCluster();
		  cluster.submitTopology("search", conf, topology);
		  //StormSubmitter.submitTopology("search", conf, topology);
	}