WritableSerialization.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.io.serializer;
  19. import java.io.DataInputStream;
  20. import java.io.DataOutputStream;
  21. import java.io.IOException;
  22. import java.io.InputStream;
  23. import java.io.OutputStream;
  24. import org.apache.hadoop.conf.Configuration;
  25. import org.apache.hadoop.conf.Configured;
  26. import org.apache.hadoop.io.Writable;
  27. import org.apache.hadoop.util.ReflectionUtils;
  28. /**
  29.  * A {@link Serialization} for {@link Writable}s that delegates to
  30.  * {@link Writable#write(java.io.DataOutput)} and
  31.  * {@link Writable#readFields(java.io.DataInput)}.
  32.  */
  33. public class WritableSerialization extends Configured 
  34.   implements Serialization<Writable> {
  35.   
  36.   static class WritableDeserializer extends Configured 
  37.     implements Deserializer<Writable> {
  38.     private Class<?> writableClass;
  39.     private DataInputStream dataIn;
  40.     
  41.     public WritableDeserializer(Configuration conf, Class<?> c) {
  42.       setConf(conf);
  43.       this.writableClass = c;
  44.     }
  45.     
  46.     public void open(InputStream in) {
  47.       if (in instanceof DataInputStream) {
  48.         dataIn = (DataInputStream) in;
  49.       } else {
  50.         dataIn = new DataInputStream(in);
  51.       }
  52.     }
  53.     
  54.     public Writable deserialize(Writable w) throws IOException {
  55.       Writable writable;
  56.       if (w == null) {
  57.         writable 
  58.           = (Writable) ReflectionUtils.newInstance(writableClass, getConf());
  59.       } else {
  60.         writable = w;
  61.       }
  62.       writable.readFields(dataIn);
  63.       return writable;
  64.     }
  65.     public void close() throws IOException {
  66.       dataIn.close();
  67.     }
  68.     
  69.   }
  70.   
  71.   static class WritableSerializer implements Serializer<Writable> {
  72.     private DataOutputStream dataOut;
  73.     
  74.     public void open(OutputStream out) {
  75.       if (out instanceof DataOutputStream) {
  76.         dataOut = (DataOutputStream) out;
  77.       } else {
  78.         dataOut = new DataOutputStream(out);
  79.       }
  80.     }
  81.     public void serialize(Writable w) throws IOException {
  82.       w.write(dataOut);
  83.     }
  84.     public void close() throws IOException {
  85.       dataOut.close();
  86.     }
  87.   }
  88.   public boolean accept(Class<?> c) {
  89.     return Writable.class.isAssignableFrom(c);
  90.   }
  91.   public Deserializer<Writable> getDeserializer(Class<Writable> c) {
  92.     return new WritableDeserializer(getConf(), c);
  93.   }
  94.   public Serializer<Writable> getSerializer(Class<Writable> c) {
  95.     return new WritableSerializer();
  96.   }
  97. }