wa-lang.org/wazero@v1.0.2/imports/proxywasm/internal/abi_callback_lifecycle.go (about)

     1  // Copyright 2020-2021 Tetrate
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  // http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  package internal
    16  
    17  import "time"
    18  
    19  //export proxy_on_context_create
    20  func proxyOnContextCreate(contextID uint32, pluginContextID uint32) {
    21  	if recordTiming {
    22  		defer logTiming("proxyOnContextCreate", time.Now())
    23  	}
    24  	if pluginContextID == 0 {
    25  		currentState.createPluginContext(contextID)
    26  	} else if currentState.createHttpContext(contextID, pluginContextID) {
    27  	} else if currentState.createTcpContext(contextID, pluginContextID) {
    28  	} else {
    29  		panic("invalid context id on proxy_on_context_create")
    30  	}
    31  }
    32  
    33  //export proxy_on_log
    34  func proxyOnLog(contextID uint32) {
    35  	if recordTiming {
    36  		defer logTiming("proxyOnLog", time.Now())
    37  	}
    38  	if ctx, ok := currentState.tcpContexts[contextID]; ok {
    39  		currentState.setActiveContextID(contextID)
    40  		ctx.OnStreamDone()
    41  	} else if ctx, ok := currentState.httpContexts[contextID]; ok {
    42  		currentState.setActiveContextID(contextID)
    43  		ctx.OnHttpStreamDone()
    44  	}
    45  }
    46  
    47  //export proxy_on_done
    48  func proxyOnDone(contextID uint32) bool {
    49  	if recordTiming {
    50  		defer logTiming("proxyOnDone", time.Now())
    51  	}
    52  	if ctx, ok := currentState.pluginContexts[contextID]; ok {
    53  		currentState.setActiveContextID(contextID)
    54  		return ctx.context.OnPluginDone()
    55  	}
    56  	return true
    57  }
    58  
    59  //export proxy_on_delete
    60  func proxyOnDelete(contextID uint32) {
    61  	if recordTiming {
    62  		defer logTiming("proxyOnDelete", time.Now())
    63  	}
    64  	delete(currentState.contextIDToRootID, contextID)
    65  	if _, ok := currentState.tcpContexts[contextID]; ok {
    66  		delete(currentState.tcpContexts, contextID)
    67  	} else if _, ok = currentState.httpContexts[contextID]; ok {
    68  		delete(currentState.httpContexts, contextID)
    69  	} else if _, ok = currentState.pluginContexts[contextID]; ok {
    70  		delete(currentState.pluginContexts, contextID)
    71  	} else {
    72  		panic("invalid context on proxy_on_delete")
    73  	}
    74  }