要在Android Studio中监控APK的网络流量，可以使用Android Profiler工具。下面是一个使用Android Profiler监控网络流量的代码示例：

1. 首先，确保你的项目中已经包含了以下依赖项：

android {
    // ...
    buildTypes {
        debug {
            // ...
            debuggable true
            // ...
        }
    }
}

dependencies {
    // ...
    debugImplementation 'androidx.lifecycle:lifecycle-runtime:2.4.0'
    debugImplementation 'androidx.lifecycle:lifecycle-common-java8:2.4.0'
    debugImplementation 'androidx.lifecycle:lifecycle-service:2.4.0'
    // ...
}

2. 在你的应用程序的入口Activity中，添加以下代码段以启用网络流量监控：

import android.os.Bundle;
import androidx.appcompat.app.AppCompatActivity;
import androidx.lifecycle.Lifecycle;
import androidx.lifecycle.LifecycleObserver;
import androidx.lifecycle.OnLifecycleEvent;

public class MainActivity extends AppCompatActivity {

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        // 注册网络流量监控
        getLifecycle().addObserver(new NetworkTrafficObserver());
    }

    private static class NetworkTrafficObserver implements LifecycleObserver {

        @OnLifecycleEvent(Lifecycle.Event.ON_CREATE)
        void onCreate() {
            // 在应用程序启动时开始监控
            NetworkTrafficMonitor.getInstance().startMonitoring();
        }

        @OnLifecycleEvent(Lifecycle.Event.ON_DESTROY)
        void onDestroy() {
            // 在应用程序销毁时停止监控
            NetworkTrafficMonitor.getInstance().stopMonitoring();
        }
    }
}

3. 创建一个NetworkTrafficMonitor类，用于实际执行网络流量监控的操作：

import android.app.Application;
import android.net.TrafficStats;
import android.os.Handler;
import android.os.Looper;
import android.util.Log;

public class NetworkTrafficMonitor {

    private static final String TAG = "NetworkTrafficMonitor";

    private static NetworkTrafficMonitor instance;

    private final Handler handler = new Handler(Looper.getMainLooper());
    private final Runnable trafficStatsRunnable = new Runnable() {
        @Override
        public void run() {
            long totalRxBytes = TrafficStats.getTotalRxBytes();
            long totalTxBytes = TrafficStats.getTotalTxBytes();

            // 处理接收到的总字节数和发送的总字节数
            // 这里只是简单打印出来，你可以根据自己的需求进行进一步处理

            Log.d(TAG, "Total received bytes: " + totalRxBytes);
            Log.d(TAG, "Total transmitted bytes: " + totalTxBytes);

            // 每秒更新一次网络流量信息
            handler.postDelayed(this, 1000);
        }
    };

    private NetworkTrafficMonitor() {}

    public static NetworkTrafficMonitor getInstance() {
        if (instance == null) {
            instance = new NetworkTrafficMonitor();
        }
        return instance;
    }

    public void startMonitoring() {
        handler.post(trafficStatsRunnable);
    }

    public void stopMonitoring() {
        handler.removeCallbacks(trafficStatsRunnable);
    }
}

这样，当应用程序启动时，将开始监控APK的网络流量，并且每秒更新一次网络流量信息。你可以根据自己的需求进一步处理接收到的总字节数和发送的总字节数。

请注意，由于TrafficStats类只会返回从设备启动以来的总字节数，因此在每次应用程序启动时启动网络流量监控可能会导致一些误差。如果你需要在应用程序的特定时间段内监控网络流量，请根据自己的需求进行修改。