001: /**
002: * Licensed to the Apache Software Foundation (ASF) under one
003: * or more contributor license agreements. See the NOTICE file
004: * distributed with this work for additional information
005: * regarding copyright ownership. The ASF licenses this file
006: * to you under the Apache License, Version 2.0 (the
007: * "License"); you may not use this file except in compliance
008: * with the License. You may obtain a copy of the License at
009: *
010: * http://www.apache.org/licenses/LICENSE-2.0
011: *
012: * Unless required by applicable law or agreed to in writing,
013: * software distributed under the License is distributed on an
014: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015: * KIND, either express or implied. See the License for the
016: * specific language governing permissions and limitations
017: * under the License.
018: */package org.apache.cxf.ws.policy;
019:
020: import java.util.ArrayList;
021: import java.util.Collection;
022: import java.util.Collections;
023: import java.util.List;
024:
025: import org.w3c.dom.Element;
026:
027: import org.apache.cxf.Bus;
028: import org.apache.cxf.endpoint.Client;
029: import org.apache.cxf.endpoint.Endpoint;
030: import org.apache.cxf.endpoint.Server;
031: import org.apache.cxf.feature.AbstractFeature;
032: import org.apache.cxf.service.model.ServiceInfo;
033: import org.apache.neethi.Policy;
034:
035: /**
036: * Configures a Server, Client, Bus with the specified policies. If a series of
037: * Policy <code>Element</code>s are supplied, these will be loaded into a Policy
038: * class using the <code>PolicyBuilder</code> extension on the bus. If the
039: * PolicyEngine has not been started, this feature will start it.
040: *
041: * @see PolicyBuilder
042: * @see AbstractFeature
043: */
044: public class WSPolicyFeature extends AbstractFeature {
045: private Collection<Policy> policies;
046: private Collection<Element> policyElements;
047:
048: public WSPolicyFeature() {
049: super ();
050: }
051:
052: public WSPolicyFeature(Policy... ps) {
053: super ();
054: policies = new ArrayList<Policy>();
055: Collections.addAll(policies, ps);
056: }
057:
058: @Override
059: public void initialize(Client client, Bus bus) {
060: Endpoint endpoint = client.getEndpoint();
061:
062: intializeEndpoint(endpoint, bus);
063: }
064:
065: @Override
066: public void initialize(Server server, Bus bus) {
067: Endpoint endpoint = server.getEndpoint();
068:
069: intializeEndpoint(endpoint, bus);
070: }
071:
072: private void intializeEndpoint(Endpoint endpoint, Bus bus) {
073: Collection<Policy> loadedPolicies = null;
074: if (policyElements != null) {
075: loadedPolicies = new ArrayList<Policy>();
076: PolicyBuilder builder = bus
077: .getExtension(PolicyBuilder.class);
078:
079: for (Element e : policyElements) {
080: loadedPolicies.add(builder.getPolicy(e));
081: }
082: }
083:
084: ensurePolicyEngineActivated(bus);
085:
086: List<ServiceInfo> sis = endpoint.getService().getServiceInfos();
087: for (ServiceInfo si : sis) {
088: if (policies != null) {
089: for (Policy p : policies) {
090: si.addExtensor(p);
091: }
092: }
093:
094: if (loadedPolicies != null) {
095: for (Policy p : loadedPolicies) {
096: si.addExtensor(p);
097: }
098: }
099: }
100: }
101:
102: private void ensurePolicyEngineActivated(Bus bus) {
103: PolicyEngine pe = bus.getExtension(PolicyEngine.class);
104:
105: // Create a PolicyEngine and enable it if there isn't one
106: if (pe == null) {
107: PolicyEngineImpl pei = new PolicyEngineImpl();
108: pei.setBus(bus);
109: bus.setExtension(pei, PolicyEngine.class);
110: pe = pei;
111: }
112:
113: synchronized (pe) {
114: if (!pe.isEnabled()) {
115: pe.setEnabled(true);
116: }
117: }
118: }
119:
120: public Collection<Policy> getPolicies() {
121: if (policies == null) {
122: policies = new ArrayList<Policy>();
123: }
124: return policies;
125: }
126:
127: public void setPolicies(Collection<Policy> policies) {
128: this .policies = policies;
129: }
130:
131: public Collection<Element> getPolicyElements() {
132: if (policyElements == null) {
133: policyElements = new ArrayList<Element>();
134: }
135: return policyElements;
136: }
137:
138: public void setPolicyElements(Collection<Element> policyElements) {
139: this.policyElements = policyElements;
140: }
141: }
|