package com.bys.aws.main.config;

@Slf4j
@Configuration
public class XRayConfig {

    @Value("${xray.name}")
    private String XRAY_APP_NAME;

    static {
        AWSXRayRecorderBuilder builder = AWSXRayRecorderBuilder
                .standard()
                .withSegmentListener(new SLF4JSegmentListener("bys-xray"))
                .withPlugin(new EC2Plugin())
                .withPlugin(new EKSPlugin());

        URL ruleFile = XRayConfig.class.getResource("/awssdk-xray-rules.json");
        log.debug("xrayRuleFile: " + ruleFile.getFile());
        builder.withSamplingStrategy(new CentralizedSamplingStrategy(ruleFile));
        AWSXRay.setGlobalRecorder(builder.build());
    }

    @Bean
    public Filter TracingFilter() {
        log.debug("The segment name for aws xray tracking has been set to {}.", XRAY_APP_NAME);
        return new AWSXRayServletFilter(
                Optional.ofNullable(XRAY_APP_NAME).orElseThrow()
        );
    }
}

package com.bys.aws.main.config;

@Slf4j
@Aspect
@Component
public class XRayInspector extends BaseAbstractXRayInterceptor {

    @Override
    @Pointcut("@within(com.amazonaws.xray.spring.aop.XRayEnabled) && (bean(*Controller) || bean(*Service) || bean(*Client))")
    public void xrayEnabledClasses() {}

    @Override
    protected Map<String, Map<String, Object>> generateMetadata(ProceedingJoinPoint proceedingJoinPoint, Subsegment subsegment) {
        return super.generateMetadata(proceedingJoinPoint, subsegment);
    }
}

package com.bys.aws.main.config;

@Slf4j
@Component
public class AWSSDKConfig {

    Region region = Region.AP_NORTHEAST_2;

    @Bean
    public S3Client getStsClient()
    {
        return S3Client.builder()
                .overrideConfiguration(ClientOverrideConfiguration.builder()
                        .addExecutionInterceptor(new TracingInterceptor())
                        .build())
                .region(region)
                .build();
    }
}

import com.amazonaws.xray.proxies.apache.http.HttpClientBuilder;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.impl.client.CloseableHttpClient;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class XRayFeignClientConfig {

    @Bean
    public CloseableHttpClient feignClient() {
        CloseableHttpClient httpclient = HttpClientBuilder.create().build();
        return httpclient;
    }
}

import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.web.bind.annotation.GetMapping;
import java.util.HashMap;

@FeignClient(name = "sts-api", url = "${feign.apiurl.iam}")
public interface StsFeignClient {

    @GetMapping(value="/v2/sts/id", produces = "application/json", consumes = "application/json")
    public HashMap<String, String> getCallerId();
}

@Slf4j
@Service
@RequiredArgsConstructor
public class S3ServiceV2 {

    private final S3Client s3Client;
    private final StsFeignClient stsFeignClient;

    public List<BucketModelV2> getBuckets() {
        HashMap<String, String> iamResponse = stsFeignClient.getCallerId();
        log.info("userId: " + iamResponse.get("userId"));
        log.info("account: " + iamResponse.get("account"));
        log.info("arn: " + iamResponse.get("arn"));

        List<BucketModelV2> buckets = new ArrayList<>();
        try {
            ListBucketsRequest listBucketsRequest = ListBucketsRequest.builder().build();
            ListBucketsResponse listBucketsResponse = s3Client.listBuckets(listBucketsRequest);
            listBucketsResponse.buckets().stream().forEach(x -> {
                buckets.add(new BucketModelV2(x.name(), x.creationDate()));
            });
        } catch (S3Exception e) {
            log.error(e.awsErrorDetails().errorMessage());
        }
        return buckets;
    }

    //convert bytes to kbs.
    private static long calKb(Long val) {
        return val/1024;
    }
}