jdk/test/java/nio/channels/Pipe/SelectPipe.java
author duke
Sat, 01 Dec 2007 00:00:00 +0000
changeset 2 90ce3da70b43
child 5506 202f599c92aa
permissions -rw-r--r--
Initial load
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     1
/*
90ce3da70b43 Initial load
duke
parents:
diff changeset
     2
 * Copyright 2002 Sun Microsystems, Inc.  All Rights Reserved.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     4
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
90ce3da70b43 Initial load
duke
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
90ce3da70b43 Initial load
duke
parents:
diff changeset
     7
 * published by the Free Software Foundation.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     8
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
90ce3da70b43 Initial load
duke
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
90ce3da70b43 Initial load
duke
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
90ce3da70b43 Initial load
duke
parents:
diff changeset
    13
 * accompanied this code).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    14
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
90ce3da70b43 Initial load
duke
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    18
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    19
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    20
 * CA 95054 USA or visit www.sun.com if you need additional information or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    21
 * have any questions.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    22
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    23
90ce3da70b43 Initial load
duke
parents:
diff changeset
    24
/* @test
90ce3da70b43 Initial load
duke
parents:
diff changeset
    25
 * @summary Test selection of ready pipe
90ce3da70b43 Initial load
duke
parents:
diff changeset
    26
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    27
90ce3da70b43 Initial load
duke
parents:
diff changeset
    28
import java.io.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    29
import java.nio.ByteBuffer;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    30
import java.nio.channels.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    31
import java.nio.channels.spi.SelectorProvider;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    32
import java.util.Random;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    33
90ce3da70b43 Initial load
duke
parents:
diff changeset
    34
/**
90ce3da70b43 Initial load
duke
parents:
diff changeset
    35
 * Testing PipeChannel
90ce3da70b43 Initial load
duke
parents:
diff changeset
    36
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    37
public class SelectPipe {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    38
90ce3da70b43 Initial load
duke
parents:
diff changeset
    39
    private static Random generator = new Random();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    40
90ce3da70b43 Initial load
duke
parents:
diff changeset
    41
    public static void main(String[] args) throws Exception {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    42
90ce3da70b43 Initial load
duke
parents:
diff changeset
    43
        SelectorProvider sp = SelectorProvider.provider();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    44
        Selector selector = Selector.open();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    45
        Pipe p = sp.openPipe();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    46
        Pipe.SinkChannel sink = p.sink();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    47
        Pipe.SourceChannel source = p.source();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    48
90ce3da70b43 Initial load
duke
parents:
diff changeset
    49
        source.configureBlocking(false);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    50
        sink.configureBlocking(false);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    51
90ce3da70b43 Initial load
duke
parents:
diff changeset
    52
        SelectionKey readkey = source.register(selector, SelectionKey.OP_READ);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    53
        SelectionKey writekey = sink.register(selector, SelectionKey.OP_WRITE);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    54
90ce3da70b43 Initial load
duke
parents:
diff changeset
    55
        ByteBuffer outgoingdata = ByteBuffer.allocateDirect(10);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    56
        byte[] someBytes = new byte[10];
90ce3da70b43 Initial load
duke
parents:
diff changeset
    57
        generator.nextBytes(someBytes);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    58
        outgoingdata.put(someBytes);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    59
        outgoingdata.flip();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    60
90ce3da70b43 Initial load
duke
parents:
diff changeset
    61
        int totalWritten = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    62
        while (totalWritten < 10) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    63
            int written = sink.write(outgoingdata);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    64
            if (written < 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    65
                throw new Exception("Write failed");
90ce3da70b43 Initial load
duke
parents:
diff changeset
    66
            totalWritten += written;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    67
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    68
90ce3da70b43 Initial load
duke
parents:
diff changeset
    69
        if (selector.select(1000) == 0) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    70
            throw new Exception("test failed");
90ce3da70b43 Initial load
duke
parents:
diff changeset
    71
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    72
90ce3da70b43 Initial load
duke
parents:
diff changeset
    73
        ByteBuffer incomingdata = ByteBuffer.allocateDirect(10);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    74
        int totalRead = 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    75
        do {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    76
            int bytesRead = source.read(incomingdata);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    77
            if (bytesRead > 0)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    78
                totalRead += bytesRead;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    79
        } while(totalRead < 10);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    80
90ce3da70b43 Initial load
duke
parents:
diff changeset
    81
        for(int i=0; i<10; i++)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    82
            if (outgoingdata.get(i) != incomingdata.get(i))
90ce3da70b43 Initial load
duke
parents:
diff changeset
    83
                throw new Exception("Pipe failed");
90ce3da70b43 Initial load
duke
parents:
diff changeset
    84
        sink.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    85
        source.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    86
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    87
}