blob: fd0487ab2fdb36c2caecdf2d39839d8b64156eb1 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.servicecomb.pack.omega.context;
/**
* OmegaContext holds the globalTxId and localTxId which are used to build the invocation map
*/
public class OmegaContext {
public static final String GLOBAL_TX_ID_KEY = "X-Pack-Global-Transaction-Id";
public static final String LOCAL_TX_ID_KEY = "X-Pack-Local-Transaction-Id";
private final ThreadLocal<String> globalTxId = new InheritableThreadLocal<>();
private final ThreadLocal<String> localTxId = new InheritableThreadLocal<>();
private final IdGenerator<String> idGenerator;
private final AlphaMetas alphaMetas;
public OmegaContext(IdGenerator<String> idGenerator) {
this(idGenerator, AlphaMetas.builder().akkaEnabled(false).build());
}
public OmegaContext(IdGenerator<String> idGenerator, AlphaMetas alphaMetas) {
this.idGenerator = idGenerator;
this.alphaMetas = alphaMetas;
}
public String newGlobalTxId() {
String id = idGenerator.nextId();
globalTxId.set(id);
return id;
}
public void setGlobalTxId(String txId) {
globalTxId.set(txId);
}
public String globalTxId() {
return globalTxId.get();
}
public String newLocalTxId() {
String id = idGenerator.nextId();
localTxId.set(id);
return id;
}
public void setLocalTxId(String localTxId) {
this.localTxId.set(localTxId);
}
public String localTxId() {
return localTxId.get();
}
public AlphaMetas getAlphaMetas() {
return alphaMetas;
}
public TransactionContext getTransactionContext() {
return new TransactionContext(globalTxId(), localTxId());
}
public void clear() {
globalTxId.remove();
localTxId.remove();
}
@Override
public String toString() {
return "OmegaContext{" +
"globalTxId=" + globalTxId.get() +
", localTxId=" + localTxId.get() +
", " + alphaMetas +
'}';
}
public void verify(){
if(this.globalTxId == null){
throw new RuntimeException("OmegaContext globalTxId is empty, Please check if you setup the pack transport handler rightly");
}
}
}