Classes in this File | Line Coverage | Branch Coverage | Complexity | ||||
KryoSimpleWrapper |
|
| 1.3333333333333333;1.333 |
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.giraph.writable.kryo; | |
19 | ||
20 | import java.io.DataInput; | |
21 | import java.io.DataOutput; | |
22 | import java.io.IOException; | |
23 | ||
24 | import com.esotericsoftware.kryo.io.Input; | |
25 | import com.esotericsoftware.kryo.io.Output; | |
26 | import org.apache.hadoop.io.Writable; | |
27 | ||
28 | /** | |
29 | * Generic wrapper object, making any object writable. | |
30 | * | |
31 | * Usage of this class is similar to KryoWritableWrapper but | |
32 | * unlike KryoWritableWrapper, this class does not | |
33 | * support recursive/nested objects to provide better | |
34 | * performance. | |
35 | * | |
36 | * If the underlying stream is a kryo output stream than the read/write | |
37 | * happens with a kryo object that doesn't track references, providing | |
38 | * significantly better performance. | |
39 | * | |
40 | * @param <T> Object type | |
41 | */ | |
42 | public class KryoSimpleWrapper<T> implements Writable, Boxed<T> { | |
43 | ||
44 | /** Wrapped object */ | |
45 | private T object; | |
46 | ||
47 | /** | |
48 | * Create wrapper given an object. | |
49 | * @param object Object instance | |
50 | */ | |
51 | 0 | public KryoSimpleWrapper(T object) { |
52 | 0 | this.object = object; |
53 | 0 | } |
54 | ||
55 | /** | |
56 | * Creates wrapper initialized with null. | |
57 | */ | |
58 | 0 | public KryoSimpleWrapper() { |
59 | 0 | } |
60 | ||
61 | /** | |
62 | * Unwrap the object value | |
63 | * @return Object value | |
64 | */ | |
65 | public T get() { | |
66 | 0 | return object; |
67 | } | |
68 | ||
69 | /** | |
70 | * Set wrapped object value | |
71 | * @param object New object value | |
72 | */ | |
73 | public void set(T object) { | |
74 | 0 | this.object = object; |
75 | 0 | } |
76 | ||
77 | @Override | |
78 | public void readFields(DataInput in) throws java.io.IOException { | |
79 | 0 | if (in instanceof Input) { |
80 | 0 | Input inp = (Input) in; |
81 | 0 | object = HadoopKryo.readWithKryo(HadoopKryo.getNontrackingKryo(), inp); |
82 | 0 | } else { |
83 | 0 | object = HadoopKryo.readClassAndObj(in); |
84 | } | |
85 | 0 | } |
86 | ||
87 | @Override | |
88 | public void write(DataOutput out) throws IOException { | |
89 | 0 | if (out instanceof Output) { |
90 | 0 | Output outp = (Output) out; |
91 | 0 | HadoopKryo.writeWithKryo(HadoopKryo.getNontrackingKryo(), outp, object); |
92 | 0 | } else { |
93 | 0 | HadoopKryo.writeClassAndObj(out, object); |
94 | } | |
95 | 0 | } |
96 | } |