View Javadoc
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,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.syncope.core.rest.cxf;
20  
21  import org.apache.cxf.interceptor.Fault;
22  import org.apache.cxf.message.Exchange;
23  import org.apache.cxf.message.Message;
24  import org.apache.cxf.phase.AbstractPhaseInterceptor;
25  import org.apache.cxf.phase.Phase;
26  import org.apache.syncope.core.provisioning.api.job.JobDelegate;
27  import org.apache.syncope.core.spring.security.SecureRandomUtils;
28  import org.slf4j.MDC;
29  
30  /**
31   * Populate MDC with sensible information, for the current thread.
32   *
33   * MDC is then cleared up by {@link org.apache.syncope.core.rest.cxf.ThreadLocalCleanupOutInterceptor}
34   */
35  public class MDCInInterceptor extends AbstractPhaseInterceptor<Message> {
36  
37      // just same value as org.apache.cxf.ext.logging.event.LogEvent.KEY_EXCHANGE_ID
38      protected static final String KEY_EXCHANGE_ID = "exchangeId";
39  
40      public MDCInInterceptor() {
41          super(Phase.PRE_INVOKE);
42      }
43  
44      @Override
45      public void handleMessage(final Message message) throws Fault {
46          Exchange exchange = message.getExchange();
47  
48          // this ensures we are not duplicating nor conflicting with
49          // org.apache.cxf.ext.logging.AbstractLoggingInterceptor.createExchangeId(Message)
50          String exchangeId = (String) exchange.get(KEY_EXCHANGE_ID);
51          if (exchangeId == null) {
52              exchangeId = SecureRandomUtils.generateRandomUUID().toString();
53              exchange.put(KEY_EXCHANGE_ID, exchangeId);
54          }
55  
56          MDC.put(JobDelegate.OPERATION_ID, exchangeId);
57      }
58  }